source.dussan.org Git - archiva.git/blob

   1 package org.apache.maven.archiva.database.constraints;
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements.  See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership.  The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License.  You may obtain a copy of the License at
  11  *
  12  *  http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied.  See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import java.util.Calendar;
  23 import java.util.Date;
  24 import java.util.List;
  25
  26 import org.apache.maven.archiva.database.AbstractArchivaDatabaseTestCase;
  27 import org.apache.maven.archiva.database.ArchivaDAO;
  28 import org.apache.maven.archiva.database.ArtifactDAO;
  29 import org.apache.maven.archiva.model.ArchivaArtifact;
  30
  31 /**
  32  * ArtifactsByRepositoryConstraintTest
  33  *
  34  * @author <a href="mailto:oching@apache.org">Maria Odea Ching</a>
  35  * @version
  36  */
  37 public class ArtifactsByRepositoryConstraintTest
  38     extends AbstractArchivaDatabaseTestCase
  39 {
  40     private ArtifactDAO artifactDao;
  41
  42     public void setUp()
  43         throws Exception
  44     {
  45         super.setUp();
  46
  47         ArchivaDAO dao = (ArchivaDAO) lookup( ArchivaDAO.ROLE, "jdo" );
  48         artifactDao = dao.getArtifactDAO();
  49     }
  50
  51     private ArchivaArtifact createArtifact( String groupId, String artifactId, String version )
  52     {
  53         ArchivaArtifact artifact = artifactDao.createArtifact( groupId, artifactId, version, null, "jar" );
  54         artifact.getModel().setLastModified( new Date() );
  55         artifact.getModel().setRepositoryId( "test-repo" );
  56
  57         return artifact;
  58     }
  59
  60     public void testQueryAllArtifactsInRepo()
  61         throws Exception
  62     {
  63         Date whenGathered = Calendar.getInstance().getTime();
  64         whenGathered.setTime( 123456789 );
  65
  66         ArchivaArtifact artifact = createArtifact( "org.apache.archiva", "artifact-one", "1.0" );
  67         artifact.getModel().setWhenGathered( whenGathered );
  68         artifactDao.saveArtifact( artifact );
  69
  70         artifact = createArtifact( "org.apache.archiva", "artifact-one", "1.0.1" );
  71         artifact.getModel().setWhenGathered( whenGathered );
  72         artifactDao.saveArtifact( artifact );
  73
  74         artifact = createArtifact( "org.apache.archiva", "artifact-two", "1.0.2" );
  75         artifact.getModel().setWhenGathered( whenGathered );
  76         artifactDao.saveArtifact( artifact );
  77
  78         artifact = createArtifact( "org.apache.archiva", "artifact-one", "2.0" );
  79         artifact.getModel().setRepositoryId( "different-repo" );
  80         artifact.getModel().setWhenGathered( whenGathered );
  81         artifactDao.saveArtifact( artifact );
  82
  83         assertConstraint( "Artifacts By Repository", 3, new ArtifactsByRepositoryConstraint( "test-repo" ) );
  84     }
  85
  86     public void testQueryArtifactsInRepoWithWhenGathered()
  87         throws Exception
  88     {
  89         Date whenGathered = Calendar.getInstance().getTime();
  90
  91         ArchivaArtifact artifact = createArtifact( "org.apache.archiva", "artifact-one", "1.0" );
  92         artifact.getModel().setWhenGathered( whenGathered );
  93         artifactDao.saveArtifact( artifact );
  94
  95         artifact = createArtifact( "org.apache.archiva", "artifact-one", "1.0.1" );
  96         artifact.getModel().setWhenGathered( whenGathered );
  97         artifactDao.saveArtifact( artifact );
  98
  99         artifact = createArtifact( "org.apache.archiva", "artifact-one", "1.0.2" );
 100         artifact.getModel().setWhenGathered( whenGathered );
 101         artifactDao.saveArtifact( artifact );
 102
 103         artifact = createArtifact( "org.apache.archiva", "artifact-one", "2.0" );
 104         artifact.getModel().setRepositoryId( "different-repo" );
 105         artifact.getModel().setWhenGathered( whenGathered );
 106         artifactDao.saveArtifact( artifact );
 107
 108         artifact = createArtifact( "org.apache.archiva", "artifact-two", "1.1-SNAPSHOT" );
 109         artifact.getModel().setWhenGathered( whenGathered );
 110         artifactDao.saveArtifact( artifact );
 111
 112         artifact = createArtifact( "org.apache.archiva", "artifact-three", "2.0-beta-1" );
 113         artifact.getModel().setWhenGathered( whenGathered );
 114         artifactDao.saveArtifact( artifact );
 115
 116         assertConstraint( "Artifacts By Repository and When Gathered", 5,
 117                           new ArtifactsByRepositoryConstraint( "test-repo" ) );
 118     }
 119
 120     private void assertConstraint( String msg, int count, ArtifactsByRepositoryConstraint constraint )
 121         throws Exception
 122     {
 123         List results = artifactDao.queryArtifacts( constraint );
 124         assertNotNull( msg + ": Not Null", results );
 125         assertEquals( msg + ": Results.size", count, results.size() );
 126     }
 127 }