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

   1 package org.apache.archiva.metadata.repository.cassandra;
   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.archiva.metadata.model.MetadataFacetFactory;
  23 import org.apache.archiva.metadata.repository.AbstractMetadataRepositoryTest;
  24 import org.apache.commons.io.FileUtils;
  25 import org.junit.After;
  26 import org.junit.Before;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29
  30 import javax.inject.Inject;
  31 import javax.inject.Named;
  32 import java.io.File;
  33 import java.util.Arrays;
  34 import java.util.Collections;
  35 import java.util.List;
  36 import java.util.Map;
  37
  38 /**
  39  * @author Olivier Lamy
  40  */
  41 public class CassandraMetadataRepositoryTest
  42     extends AbstractMetadataRepositoryTest
  43 {
  44     private Logger logger = LoggerFactory.getLogger( getClass() );
  45
  46     @Inject
  47     @Named(value = "archivaEntityManagerFactory#cassandra")
  48     CassandraArchivaManager cassandraArchivaManager;
  49
  50     CassandraMetadataRepository cmr;
  51
  52     @Before
  53     public void setUp()
  54         throws Exception
  55     {
  56         super.setUp();
  57
  58         File directory = new File( "target/test-repositories" );
  59         if ( directory.exists() )
  60         {
  61             FileUtils.deleteDirectory( directory );
  62         }
  63
  64         Map<String, MetadataFacetFactory> factories = createTestMetadataFacetFactories();
  65
  66         this.cmr = new CassandraMetadataRepository( factories, null, cassandraArchivaManager );
  67         this.repository = this.cmr;
  68
  69         clearReposAndNamespace();
  70     }
  71
  72
  73     @After
  74     public void shutdown()
  75         throws Exception
  76     {
  77         clearReposAndNamespace();
  78         super.tearDown();
  79     }
  80
  81     protected void clearReposAndNamespace()
  82         throws Exception
  83     {
  84         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
  85                                                        "project" );
  86         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
  87                                                        "namespace" );
  88         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
  89                                                        "repository" );
  90
  91         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
  92                                                        "projectversionmetadatamodel" );
  93
  94         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
  95                                                        "artifactmetadatamodel" );
  96
  97         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
  98                                                        "metadatafacetmodel" );
  99
 100         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
 101                                                        "metadatafacetmodel" );
 102
 103         cassandraArchivaManager.getCluster().truncate( cassandraArchivaManager.getKeyspace().getKeyspaceName(),
 104                                                        "mailinglist" );
 105
 106     }
 107
 108 }