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

   1 package org.apache.maven.archiva.indexer.search;
   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.commons.io.FileUtils;
  23 import org.apache.maven.archiva.indexer.filecontent.FileContentRecord;
  24 import org.apache.maven.archiva.model.ArchivaArtifact;
  25
  26 import java.io.File;
  27 import java.io.IOException;
  28 import java.util.HashMap;
  29 import java.util.Map;
  30
  31 import junit.framework.AssertionFailedError;
  32
  33 /**
  34  * FileContentIndexPopulator
  35  *
  36  * @version $Id$
  37  */
  38 public class FileContentIndexPopulator
  39     implements IndexPopulator
  40 {
  41     public Map<String, ArchivaArtifact> getObjectMap()
  42     {
  43         return null;
  44     }
  45
  46     public Map<String, FileContentRecord> populate( File basedir )
  47     {
  48         Map<String, FileContentRecord> map = new HashMap<String, FileContentRecord>();
  49
  50         File repoDir = new File( basedir, "src/test/managed-repository" );
  51
  52         String prefix = "org/apache/maven/archiva/record/";
  53
  54         map.put( "parent-pom-1", createFileContentRecord( repoDir, prefix + "parent-pom/1/parent-pom-1.pom" ) );
  55         map.put( "child-pom-1.0-SNAPSHOT", createFileContentRecord( repoDir, prefix
  56             + "test-child-pom/1.0-SNAPSHOT/test-child-pom-1.0-20060728.121314-1.pom" ) );
  57         map.put( "test-archetype-1.0", createFileContentRecord( repoDir, prefix
  58             + "test-archetype/1.0/test-archetype-1.0.pom" ) );
  59         map.put( "test-jar-and-pom-1.0-alpha-1", createFileContentRecord( repoDir, prefix
  60             + "test-jar-and-pom/1.0-alpha-1/test-jar-and-pom-1.0-alpha-1.pom" ) );
  61         map.put( "test-plugin-1.0", createFileContentRecord( repoDir, prefix + "test-plugin/1.0/test-plugin-1.0.pom" ) );
  62         map.put( "test-pom-1.0", createFileContentRecord( repoDir, prefix + "test-pom/1.0/test-pom-1.0.pom" ) );
  63         map.put( "test-skin-1.0", createFileContentRecord( repoDir, prefix + "test-skin/1.0/test-skin-1.0.pom" ) );
  64
  65         return map;
  66     }
  67
  68     private FileContentRecord createFileContentRecord( File repoDir, String path )
  69     {
  70         File pathToFile = new File( repoDir, path );
  71
  72         if ( !pathToFile.exists() )
  73         {
  74             throw new AssertionFailedError( "Can't find test file: " + pathToFile.getAbsolutePath() );
  75         }
  76
  77         FileContentRecord record = new FileContentRecord();
  78         record.setRepositoryId( "test-repo" );
  79         record.setFilename( path );
  80
  81         try
  82         {
  83             record.setContents( FileUtils.readFileToString( pathToFile, null ) );
  84         }
  85         catch ( IOException e )
  86         {
  87             e.printStackTrace();
  88             throw new AssertionFailedError( "Can't load test file contents: " + pathToFile.getAbsolutePath() );
  89         }
  90
  91         return record;
  92     }
  93 }