/**
* Shuts down the test harness, and makes the best attempt possible to delete dataDir, unless the
* system property "solr.test.leavedatadir" is set.
*/
@Override
public void tearDown() throws Exception {
log.info("####TEARDOWN_START " + getTestName());
if (factoryProp == null) {
System.clearProperty("solr.directoryFactory");
}
if (h != null) {
h.close();
}
String skip = System.getProperty("solr.test.leavedatadir");
if (null != skip && 0 != skip.trim().length()) {
System.err.println(
"NOTE: per solr.test.leavedatadir, dataDir will not be removed: "
+ dataDir.getAbsolutePath());
} else {
if (!recurseDelete(dataDir)) {
System.err.println(
"!!!! WARNING: best effort to remove " + dataDir.getAbsolutePath() + " FAILED !!!!!");
}
}
resetExceptionIgnores();
super.tearDown();
}
public static boolean recurseDelete(File f) {
if (f.isDirectory()) {
for (File sub : f.listFiles()) {
if (!recurseDelete(sub)) {
System.err.println(
"!!!! WARNING: best effort to remove " + sub.getAbsolutePath() + " FAILED !!!!!");
return false;
}
}
}
return f.delete();
}
/**
* Initializes things your test might need
*
* <ul>
* <li>Creates a dataDir in the "java.io.tmpdir"
* <li>initializes the TestHarness h using this data directory, and getSchemaPath()
* <li>initializes the LocalRequestFactory lrf using sensible defaults.
* </ul>
*/
@Override
public void setUp() throws Exception {
super.setUp();
log.info("####SETUP_START " + getTestName());
ignoreException("ignore_exception");
factoryProp = System.getProperty("solr.directoryFactory");
if (factoryProp == null) {
System.setProperty("solr.directoryFactory", "solr.RAMDirectoryFactory");
}
dataDir = new File(TEMP_DIR, getClass().getName() + "-" + System.currentTimeMillis());
dataDir.mkdirs();
String configFile = getSolrConfigFile();
System.setProperty("solr.solr.home", getSolrHome());
if (configFile != null) {
solrConfig = TestHarness.createConfig(getSolrConfigFile());
h = new TestHarness(dataDir.getAbsolutePath(), solrConfig, getSchemaFile());
lrf = h.getRequestFactory("standard", 0, 20, CommonParams.VERSION, "2.2");
}
log.info("####SETUP_END " + getTestName());
}
public void init() throws ServletException {
final String CONFIG_PATH = getServletContext().getRealPath("/") + "config.ini";
final String APPLICATION_NAME = getServletContext().getServletContextName();
BasicConfigurator.configure();
FileInputStream config_file = null;
File config = new File(CONFIG_PATH);
if (config.canRead()) {
log.debug("Parsing " + CONFIG_PATH);
try {
config_file = new FileInputStream(CONFIG_PATH);
} catch (java.io.FileNotFoundException e) {
throw new ServletException(e);
}
Properties config_prop = new Properties();
try {
config_prop.load(config_file);
} catch (java.io.IOException e) {
throw new ServletException(e);
}
this.config = config_prop;
log.debug("Parsing finished");
} else {
log.fatal("Error, cannot read " + CONFIG_PATH);
}
log.debug("maximumRecords : " + this.config.getProperty("default_maximumRecords"));
log.debug("srw_header_file: " + this.config.getProperty("srw_header_file"));
log.debug("srw_diag: " + this.config.getProperty("srw_diag_file"));
// log.debug("server_list: " + this.config.getProperty("server_list"));
try {
FileReader fis =
new FileReader(
getServletContext().getRealPath("/") + this.config.getProperty("srw_header_file"));
BufferedReader in1 = new BufferedReader(fis);
fis =
new FileReader(
getServletContext().getRealPath("/") + this.config.getProperty("srw_footer_file"));
BufferedReader in2 = new BufferedReader(fis);
fis =
new FileReader(
getServletContext().getRealPath("/") + this.config.getProperty("srw_diag_file"));
BufferedReader in3 = new BufferedReader(fis);
try {
String line = null;
while ((line = in1.readLine()) != null) {
this.SRW_HEADER = this.SRW_HEADER + line;
}
while ((line = in2.readLine()) != null) {
this.SRW_FOOTER = this.SRW_FOOTER + line;
}
while ((line = in3.readLine()) != null) {
this.SRW_DIAG = this.SRW_DIAG + line;
}
} finally {
in1.close();
in2.close();
in3.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}