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

   1 package org.apache.maven.archiva.scheduled.executors;
   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.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  23 import org.apache.maven.archiva.database.ArchivaDAO;
  24 import org.apache.maven.archiva.database.ArtifactDAO;
  25 import org.apache.maven.archiva.database.constraints.ArtifactsProcessedConstraint;
  26 import org.apache.maven.archiva.model.ArchivaArtifact;
  27 import org.apache.maven.archiva.scheduled.tasks.DatabaseTask;
  28 import org.codehaus.plexus.jdo.DefaultConfigurableJdoFactory;
  29 import org.codehaus.plexus.jdo.JdoFactory;
  30 import org.codehaus.plexus.spring.PlexusInSpringTestCase;
  31 import org.codehaus.plexus.taskqueue.execution.TaskExecutor;
  32 import org.jpox.SchemaTool;
  33
  34 import java.io.File;
  35 import java.net.URL;
  36 import java.util.Date;
  37 import java.util.Iterator;
  38 import java.util.List;
  39 import java.util.Map;
  40 import java.util.Properties;
  41
  42 import javax.jdo.PersistenceManager;
  43 import javax.jdo.PersistenceManagerFactory;
  44
  45 /**
  46  * ArchivaDatabaseUpdateTaskExecutorTest
  47  *
  48  * @version $Id:$
  49  */
  50 public class ArchivaDatabaseUpdateTaskExecutorTest
  51     extends PlexusInSpringTestCase
  52 {
  53     private TaskExecutor taskExecutor;
  54
  55     protected ArchivaDAO dao;
  56
  57     protected void setUp()
  58         throws Exception
  59     {
  60         super.setUp();
  61
  62         DefaultConfigurableJdoFactory jdoFactory = (DefaultConfigurableJdoFactory) lookup( JdoFactory.ROLE, "archiva" );
  63         assertEquals( DefaultConfigurableJdoFactory.class.getName(), jdoFactory.getClass().getName() );
  64
  65         jdoFactory.setPersistenceManagerFactoryClass( "org.jpox.PersistenceManagerFactoryImpl" );
  66
  67         /* derby version
  68          File derbyDbDir = new File( "target/plexus-home/testdb" );
  69          if ( derbyDbDir.exists() )
  70          {
  71          FileUtils.deleteDirectory( derbyDbDir );
  72          }
  73
  74          jdoFactory.setDriverName( System.getProperty( "jdo.test.driver", "org.apache.derby.jdbc.EmbeddedDriver" ) );
  75          jdoFactory.setUrl( System.getProperty( "jdo.test.url", "jdbc:derby:" + derbyDbDir.getAbsolutePath() + ";create=true" ) );
  76          */
  77
  78         jdoFactory.setDriverName( System.getProperty( "jdo.test.driver", "org.hsqldb.jdbcDriver" ) );
  79         jdoFactory.setUrl( System.getProperty( "jdo.test.url", "jdbc:hsqldb:mem:" + getName() ) );
  80
  81         jdoFactory.setUserName( System.getProperty( "jdo.test.user", "sa" ) );
  82
  83         jdoFactory.setPassword( System.getProperty( "jdo.test.pass", "" ) );
  84
  85         jdoFactory.setProperty( "org.jpox.transactionIsolation", "READ_COMMITTED" );
  86
  87         jdoFactory.setProperty( "org.jpox.poid.transactionIsolation", "READ_COMMITTED" );
  88
  89         jdoFactory.setProperty( "org.jpox.autoCreateSchema", "true" );
  90
  91         jdoFactory.setProperty( "javax.jdo.option.RetainValues", "true" );
  92
  93         jdoFactory.setProperty( "javax.jdo.option.RestoreValues", "true" );
  94
  95         // jdoFactory.setProperty( "org.jpox.autoCreateColumns", "true" );
  96
  97         jdoFactory.setProperty( "org.jpox.validateTables", "true" );
  98
  99         jdoFactory.setProperty( "org.jpox.validateColumns", "true" );
 100
 101         jdoFactory.setProperty( "org.jpox.validateConstraints", "true" );
 102
 103         Properties properties = jdoFactory.getProperties();
 104
 105         for ( Iterator it = properties.entrySet().iterator(); it.hasNext(); )
 106         {
 107             Map.Entry entry = (Map.Entry) it.next();
 108
 109             System.setProperty( (String) entry.getKey(), (String) entry.getValue() );
 110         }
 111
 112         URL jdoFileUrls[] = new URL[] { getClass().getResource( "/org/apache/maven/archiva/model/package.jdo" ) };
 113
 114         if ( ( jdoFileUrls == null ) || ( jdoFileUrls[0] == null ) )
 115         {
 116             fail( "Unable to process test " + getName() + " - missing package.jdo." );
 117         }
 118
 119         File propsFile = null; // intentional
 120         boolean verbose = true;
 121
 122         SchemaTool.deleteSchemaTables( jdoFileUrls, new URL[] {}, propsFile, verbose );
 123         SchemaTool.createSchemaTables( jdoFileUrls, new URL[] {}, propsFile, verbose, null );
 124
 125         PersistenceManagerFactory pmf = jdoFactory.getPersistenceManagerFactory();
 126
 127         assertNotNull( pmf );
 128
 129         PersistenceManager pm = pmf.getPersistenceManager();
 130
 131         pm.close();
 132
 133         this.dao = (ArchivaDAO) lookup( ArchivaDAO.class.getName(), "jdo" );
 134
 135         taskExecutor = (TaskExecutor) lookup( TaskExecutor.class, "test-database-update" );
 136     }
 137
 138     public void testExecutor()
 139         throws Exception
 140     {
 141         File repoDir = new File( getBasedir(), "src/test/repositories/default-repository" );
 142
 143         assertTrue( "Default Test Repository should exist.", repoDir.exists() && repoDir.isDirectory() );
 144
 145         ManagedRepositoryConfiguration repo = createRepository( "testRepo", "Test Repository", repoDir );
 146         assertNotNull( repo );
 147
 148         ArtifactDAO adao = dao.getArtifactDAO();
 149
 150         ArchivaArtifact sqlArtifact = adao.createArtifact( "javax.sql", "jdbc", "2.0", "", "jar", repo.getId() );
 151         sqlArtifact.getModel().setLastModified( new Date() );
 152         sqlArtifact.getModel().setSize( 1234 );
 153         sqlArtifact.getModel().setOrigin( "testcase" );
 154         sqlArtifact.getModel().setWhenProcessed( null );
 155
 156         adao.saveArtifact( sqlArtifact );
 157
 158         ArchivaArtifact artifact = adao.getArtifact( "javax.sql", "jdbc", "2.0", null, "jar", repo.getId() );
 159
 160         assertNotNull( artifact );
 161
 162         // Test for artifact existance.
 163         List artifactList = adao.queryArtifacts( null );
 164         assertNotNull( "Artifact list should not be null.", artifactList );
 165         assertEquals( "Artifact list size", 1, artifactList.size() );
 166
 167         // Test for unprocessed artifacts.
 168         List unprocessedResultList = adao.queryArtifacts( new ArtifactsProcessedConstraint( false ) );
 169         assertNotNull( "Unprocessed Results should not be null.", unprocessedResultList );
 170         assertEquals( "Incorrect number of unprocessed artifacts detected.", 1, unprocessedResultList.size() );
 171
 172         // Execute the database task.
 173         DatabaseTask dataTask = new DatabaseTask();
 174
 175         dataTask.setName( "testDataTask" );
 176
 177         taskExecutor.executeTask( dataTask );
 178
 179         // Test for artifact existance.
 180         artifactList = adao.queryArtifacts( null );
 181         assertNotNull( "Artifact list should not be null.", artifactList );
 182         assertEquals( "Artifact list size", 1, artifactList.size() );
 183
 184         // Test for processed artifacts.
 185         List processedResultList = adao.queryArtifacts( new ArtifactsProcessedConstraint( true ) );
 186         assertNotNull( "Processed Results should not be null.", processedResultList );
 187         assertEquals( "Incorrect number of processed artifacts detected.", 1, processedResultList.size() );
 188     }
 189
 190     protected ManagedRepositoryConfiguration createRepository( String id, String name, File location )
 191     {
 192         ManagedRepositoryConfiguration repo = new ManagedRepositoryConfiguration();
 193         repo.setId( id );
 194         repo.setName( name );
 195         repo.setLocation( location.getAbsolutePath() );
 196         return repo;
 197     }
 198 }