Java SolrTestCaseJ4类代码示例

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
@Override
@Before
public void setUp() throws Exception {
  super.setUp();

  System.setProperty("solr.solr.home", SOLR_HOME.getAbsolutePath());
  System.setProperty("configSetBase", SolrTestCaseJ4.getFile("solrj/solr/configsets").getAbsolutePath());
  System.out.println("Solr home: " + SOLR_HOME.getAbsolutePath());

  //The index is always stored within a temporary directory
  tempDir = createTempDir();
  
  File dataDir = new File(tempDir,"data1");
  File dataDir2 = new File(tempDir,"data2");
  System.setProperty("dataDir1", dataDir.getAbsolutePath());
  System.setProperty("dataDir2", dataDir2.getAbsolutePath());
  System.setProperty("tempDir", tempDir.getAbsolutePath());
  System.setProperty("tests.shardhandler.randomSeed", Long.toString(random().nextLong()));
  cores = CoreContainer.createAndLoad(SOLR_HOME.getAbsolutePath(), getSolrXml());
  //cores.setPersistent(false);
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
public void setUp() throws Exception {
  File home = new File(dataDir,
          getClass().getName() + "-" + System.currentTimeMillis());
  homeDir = new File(home, name);
  dataDir = new File(homeDir + "/collection1", "data");
  confDir = new File(homeDir + "/collection1", "conf");

  homeDir.mkdirs();
  dataDir.mkdirs();
  confDir.mkdirs();

  FileUtils.copyFile(SolrTestCaseJ4.getFile(getSolrXmlFile()), new File(homeDir, "solr.xml"));

  File f = new File(confDir, "solrconfig.xml");
  FileUtils.copyFile(SolrTestCaseJ4.getFile(getSolrConfigFile()), f);
  f = new File(confDir, "schema.xml");
  FileUtils.copyFile(SolrTestCaseJ4.getFile(getSchemaFile()), f);

}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
private SolrDocumentList getDocs(int start, int rows) {
  SolrDocumentList docs = new SolrDocumentList();
  docs.setNumFound(docsData.size());
  docs.setStart(start);

  int endIndex = start + rows;
  int end = docsData.size() < endIndex ? docsData.size() : endIndex;
  for (int i = start; i < end; i++) {
    SolrDocument doc = new SolrDocument();
    SolrTestCaseJ4.Doc testDoc = docsData.get(i);
    doc.addField("id", testDoc.id);
    doc.addField("description", testDoc.getValues("description"));
    docs.add(doc);
  }
  return docs;
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
public static void teardownClass(MiniDFSCluster dfsCluster) throws Exception {
  SolrTestCaseJ4.resetFactory();
  System.clearProperty("solr.lock.type");
  System.clearProperty("test.build.data");
  System.clearProperty("test.cache.data");
  System.clearProperty("solr.hdfs.home");
  System.clearProperty("solr.hdfs.blockcache.global");
  if (dfsCluster != null) {
    timers.remove(dfsCluster);
    dfsCluster.shutdown();
  }
  
  // TODO: we HACK around HADOOP-9643
  if (savedLocale != null) {
    Locale.setDefault(savedLocale);
  }
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
@BeforeClass
public static void beforeTest() throws Exception {
  File homeDir = createTempDir();

  File collDir = new File(homeDir, "collection1");
  File dataDir = new File(collDir, "data");
  File confDir = new File(collDir, "conf");

  homeDir.mkdirs();
  collDir.mkdirs();
  dataDir.mkdirs();
  confDir.mkdirs();

  FileUtils.copyFile(new File(SolrTestCaseJ4.TEST_HOME(), "solr.xml"), new File(homeDir, "solr.xml"));

  String src_dir = TEST_HOME() + "/collection1/conf";
  FileUtils.copyFile(new File(src_dir, "schema-binaryfield.xml"), 
                     new File(confDir, "schema.xml"));
  FileUtils.copyFile(new File(src_dir, "solrconfig-basic.xml"), 
                     new File(confDir, "solrconfig.xml"));
  FileUtils.copyFile(new File(src_dir, "solrconfig.snippet.randomindexconfig.xml"), 
                     new File(confDir, "solrconfig.snippet.randomindexconfig.xml"));

  createJetty(homeDir.getAbsolutePath(), null, null);
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
public void testPropertySub() throws IOException {

    System.setProperty("coreRootDirectory", "myCoreRoot" + File.separator);
    System.setProperty("hostPort", "8888");
    System.setProperty("shareSchema", "false");
    System.setProperty("socketTimeout", "220");
    System.setProperty("connTimeout", "200");

    File testSrcRoot = new File(SolrTestCaseJ4.TEST_HOME());
    FileUtils.copyFile(new File(testSrcRoot, "solr-50-all.xml"), new File(solrHome, "solr.xml"));

    ConfigSolr cfg = ConfigSolr.fromSolrHome(loader, solrHome.getAbsolutePath());
    assertEquals("core root dir", "myCoreRoot" + File.separator, cfg.getCoreRootDirectory());
    assertEquals("zk host port", "8888", cfg.getZkHostPort());
    assertEquals("schema cache", false, cfg.hasSchemaCache());
  }
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
private void makeCores(File home, boolean oldStyle) throws Exception {
  File testSrcRoot = new File(SolrTestCaseJ4.TEST_HOME());
  String srcSolrXml = "solr-stress-new.xml";

  if (oldStyle) {
    srcSolrXml = "solr-stress-old.xml";
  }
  FileUtils.copyFile(new File(testSrcRoot, srcSolrXml), new File(home, "solr.xml"));

  // create directories in groups of 100 until you have enough.
  for (int idx = 0; idx < numCores; ++idx) {
    String coreName = String.format(Locale.ROOT, "%05d_core", idx);
    makeCore(new File(home, coreName), testSrcRoot, coreName);
    coreCounts.put(coreName, 0L);
    coreNames.add(coreName);
  }
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
@Override
public void setUp() throws Exception {
  saveProp = System.getProperty("solr.directoryFactory");
  System.setProperty("solr.directoryFactory", "solr.StandardDirectoryFactory");
  super.setUp();

  SolrCore.log.info("CORES=" + cores + " : " + cores.getCoreNames());
  cores.setPersistent(false);
  
  // setup datadirs
  System.setProperty( "solr.core0.data.dir", SolrTestCaseJ4.dataDir.getCanonicalPath() ); 
  
  dataDir2 = new File(TEMP_DIR, getClass().getName() + "-"
      + System.currentTimeMillis());
  dataDir2.mkdirs();
  
  System.setProperty( "solr.core1.data.dir", this.dataDir2.getCanonicalPath() ); 
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
public void setUp() throws Exception {
  File home = new File(LuceneTestCase.TEMP_DIR,
          getClass().getName() + "-" + System.currentTimeMillis());


  homeDir = new File(home, name);
  dataDir = new File(homeDir + "/collection1", "data");
  confDir = new File(homeDir + "/collection1", "conf");

  homeDir.mkdirs();
  dataDir.mkdirs();
  confDir.mkdirs();

  File f = new File(confDir, "solrconfig.xml");
  FileUtils.copyFile(SolrTestCaseJ4.getFile(getSolrConfigFile()), f);
  f = new File(confDir, "schema.xml");
  FileUtils.copyFile(SolrTestCaseJ4.getFile(getSchemaFile()), f);

}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
public void setupSentry() throws Exception {
  sentrySite = File.createTempFile("sentry-site", "xml");
  File authProviderDir = SolrTestCaseJ4.getFile("sentry-handlers/sentry");
  sentrySite.deleteOnExit();

  // need to write sentry-site at execution time because we don't know
  // the location of sentry.solr.provider.resource beforehand
  StringBuilder sentrySiteData = new StringBuilder();
  sentrySiteData.append("<configuration>\n");
  addPropertyToSentry(sentrySiteData, "sentry.provider",
    "org.apache.sentry.provider.file.LocalGroupResourceAuthorizationProvider");
  addPropertyToSentry(sentrySiteData, "sentry.solr.provider.resource",
     new File(authProviderDir.toString(), "test-authz-provider.ini").toURI().toURL().toString());
  sentrySiteData.append("</configuration>\n");
  FileUtils.writeStringToFile(sentrySite,sentrySiteData.toString(), Charsets.UTF_8.toString());
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
protected SolrServer createServer( String name )
{
  try {
    // setup the server...
    String url = buildUrl(port, context) + "/" + name;
    HttpSolrServer s = new HttpSolrServer( url );
    s.setConnectionTimeout(SolrTestCaseJ4.DEFAULT_CONNECTION_TIMEOUT);
    s.setDefaultMaxConnectionsPerHost(100);
    s.setMaxTotalConnections(100);
    return s;
  }
  catch( Exception ex ) {
    throw new RuntimeException( ex );
  }
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
@BeforeClass
public static void beforeTest() throws Exception {
  solrHomeDirectory = createTempDir();
  setupJettyTestHome(solrHomeDirectory, "collection1");
  String top = SolrTestCaseJ4.TEST_HOME() + "/collection1/conf";
  FileUtils.copyFile(new File(top, "solrconfig-headers.xml"), new File(solrHomeDirectory + "/collection1/conf", "solrconfig.xml"));
  createJetty(solrHomeDirectory.getAbsolutePath(), null, null);
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
private static void setupCore(String coreName, boolean blivet) throws IOException {
  File instDir = new File(solrHomeDirectory, coreName);
  File subHome = new File(instDir, "conf");
  assertTrue("Failed to make subdirectory ", subHome.mkdirs());

  // Be sure we pick up sysvars when we create this
  String srcDir = SolrTestCaseJ4.TEST_HOME() + "/collection1/conf";
  FileUtils.copyFile(new File(srcDir, "schema-tiny.xml"), new File(subHome, "schema_ren.xml"));
  FileUtils.copyFile(new File(srcDir, "solrconfig-minimal.xml"), new File(subHome, "solrconfig_ren.xml"));

  FileUtils.copyFile(new File(srcDir, "solrconfig.snippet.randomindexconfig.xml"),
      new File(subHome, "solrconfig.snippet.randomindexconfig.xml"));
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
private static JettySolrRunner createJetty(SolrInstance instance) throws Exception {
  FileUtils.copyFile(new File(SolrTestCaseJ4.TEST_HOME(), "solr.xml"), new File(instance.getHomeDir(), "solr.xml"));
  JettySolrRunner jetty = new JettySolrRunner(instance.getHomeDir(), "/solr", 0);
  jetty.setDataDir(instance.getDataDir());
  jetty.start();
  return jetty;
}
 

import org.apache.solr.SolrTestCaseJ4; //导入依赖的package包/类
private static JettySolrRunner createJetty(TestReplicationHandler.SolrInstance instance) throws Exception {
  FileUtils.copyFile(new File(SolrTestCaseJ4.TEST_HOME(), "solr.xml"), new File(instance.getHomeDir(), "solr.xml"));
  JettySolrRunner jetty = new JettySolrRunner(instance.getHomeDir(), "/solr", 0);
  jetty.setDataDir(instance.getDataDir());
  jetty.start();
  return jetty;
}