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.easymock.MockControl;
  23 import org.apache.maven.archiva.database.ProjectModelDAO;
  24 import org.apache.maven.archiva.model.ArchivaArtifact;
  25 import org.apache.maven.archiva.model.ArchivaProjectModel;
  26
  27 /**
  28  * Test for DatabaseCleanupRemoveProjectConsumer
  29  *
  30  * @author <a href="mailto:oching@apache.org">Maria Odea Ching</a>
  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     public void setUp()
  42         throws Exception
  43     {
  44         super.setUp();
  45
  46         dbCleanupRemoveProjectConsumer = new DatabaseCleanupRemoveProjectConsumer();
  47
  48         projectModelDAOControl = MockControl.createControl( ProjectModelDAO.class );
  49
  50         projectModelDAOMock = (ProjectModelDAO) projectModelDAOControl.getMock();
  51
  52         dbCleanupRemoveProjectConsumer.setProjectModelDAO( projectModelDAOMock );
  53
  54         dbCleanupRemoveProjectConsumer.setRepositoryFactory( repositoryFactory );
  55     }
  56
  57     public void testIfArtifactWasNotDeleted()
  58         throws Exception
  59     {
  60         ArchivaArtifact artifact = createArtifact( TEST_GROUP_ID, "do-not-cleanup-artifact-test", TEST_VERSION, "pom" );
  61
  62         projectModelDAOControl.replay();
  63
  64         dbCleanupRemoveProjectConsumer.processArchivaArtifact( artifact );
  65
  66         projectModelDAOControl.verify();
  67     }
  68
  69     public void testIfArtifactWasDeleted()
  70         throws Exception
  71     {
  72         ArchivaArtifact artifact = createArtifact( TEST_GROUP_ID, TEST_ARTIFACT_ID, TEST_VERSION, "pom" );
  73
  74         ArchivaProjectModel projectModel = createProjectModel( TEST_GROUP_ID, TEST_ARTIFACT_ID, TEST_VERSION );
  75
  76         //this should return a value
  77         projectModelDAOControl.expectAndReturn(
  78             projectModelDAOMock.getProjectModel( TEST_GROUP_ID, TEST_ARTIFACT_ID, TEST_VERSION ),
  79             (ArchivaProjectModel) projectModel );
  80
  81         projectModelDAOMock.deleteProjectModel( projectModel );
  82
  83         projectModelDAOControl.replay();
  84
  85         dbCleanupRemoveProjectConsumer.processArchivaArtifact( artifact );
  86
  87         projectModelDAOControl.verify();
  88     }
  89
  90     public void testIfArtifactWasNotAPom()
  91         throws Exception
  92         {
  93         ArchivaArtifact artifact = createArtifact( TEST_GROUP_ID, "do-not-cleanup-artifact-test", TEST_VERSION, "jar" );
  94
  95         projectModelDAOControl.replay();
  96
  97         dbCleanupRemoveProjectConsumer.processArchivaArtifact( artifact );
  98
  99         projectModelDAOControl.verify();
 100         }
 101
 102     public void tearDown()
 103         throws Exception
 104     {
 105         super.tearDown();
 106     }
 107
 108 }