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

   1 package org.apache.maven.archiva.consumers.database;
   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 org.codehaus.plexus.cache.Cache;
  23 import org.easymock.MockControl;
  24 import org.apache.maven.archiva.database.ProjectModelDAO;
  25 import org.apache.maven.archiva.model.ArchivaArtifact;
  26 import org.apache.maven.archiva.model.ArchivaProjectModel;
  27
  28 /**
  29  * Test for DatabaseCleanupRemoveProjectConsumer
  30  *
  31  */
  32 public class DatabaseCleanupRemoveProjectConsumerTest
  33         extends AbstractDatabaseCleanupTest
  34 {
  35     private MockControl projectModelDAOControl;
  36
  37     private ProjectModelDAO projectModelDAOMock;
  38
  39     private DatabaseCleanupRemoveProjectConsumer dbCleanupRemoveProjectConsumer;
  40
  41     private Cache effectiveProjectCache;
  42
  43     public void setUp()
  44         throws Exception
  45     {
  46         super.setUp();
  47
  48         dbCleanupRemoveProjectConsumer = new DatabaseCleanupRemoveProjectConsumer();
  49
  50         projectModelDAOControl = MockControl.createControl( ProjectModelDAO.class );
  51
  52         projectModelDAOMock = (ProjectModelDAO) projectModelDAOControl.getMock();
  53
  54         effectiveProjectCache = (Cache) lookup( Cache.class, "effective-project-cache" );
  55
  56         dbCleanupRemoveProjectConsumer.setProjectModelDAO( projectModelDAOMock );
  57
  58         dbCleanupRemoveProjectConsumer.setRepositoryFactory( repositoryFactory );
  59
  60         dbCleanupRemoveProjectConsumer.setEffectiveProjectCache( effectiveProjectCache );
  61     }
  62
  63     public void testIfArtifactWasNotDeleted()
  64         throws Exception
  65     {
  66         ArchivaArtifact artifact = createArtifact( TEST_GROUP_ID, "do-not-cleanup-artifact-test", TEST_VERSION, "pom" );
  67
  68         projectModelDAOControl.replay();
  69
  70         dbCleanupRemoveProjectConsumer.processArchivaArtifact( artifact );
  71
  72         projectModelDAOControl.verify();
  73     }
  74
  75     public void testIfArtifactWasDeleted()
  76         throws Exception
  77     {
  78         ArchivaArtifact artifact = createArtifact( TEST_GROUP_ID, TEST_ARTIFACT_ID, TEST_VERSION, "pom" );
  79
  80         ArchivaProjectModel projectModel = createProjectModel( TEST_GROUP_ID, TEST_ARTIFACT_ID, TEST_VERSION );
  81
  82         //this should return a value
  83         projectModelDAOControl.expectAndReturn(
  84             projectModelDAOMock.getProjectModel( TEST_GROUP_ID, TEST_ARTIFACT_ID, TEST_VERSION ),
  85             (ArchivaProjectModel) projectModel );
  86
  87         projectModelDAOMock.deleteProjectModel( projectModel );
  88
  89         projectModelDAOControl.replay();
  90
  91         dbCleanupRemoveProjectConsumer.processArchivaArtifact( artifact );
  92
  93         projectModelDAOControl.verify();
  94     }
  95
  96     public void testIfArtifactWasNotAPom()
  97         throws Exception
  98         {
  99         ArchivaArtifact artifact = createArtifact( TEST_GROUP_ID, "do-not-cleanup-artifact-test", TEST_VERSION, "jar" );
 100
 101         projectModelDAOControl.replay();
 102
 103         dbCleanupRemoveProjectConsumer.processArchivaArtifact( artifact );
 104
 105         projectModelDAOControl.verify();
 106         }
 107
 108     public void tearDown()
 109         throws Exception
 110     {
 111         super.tearDown();
 112     }
 113
 114 }