pylucene 3.5.0-3

[pylucene.git] / lucene-java-3.5.0 / lucene / contrib / benchmark / src / test / org / apache / lucene / benchmark / BenchmarkTestCase.java

diff --git a/lucene-java-3.5.0/lucene/contrib/benchmark/src/test/org/apache/lucene/benchmark/BenchmarkTestCase.java b/lucene-java-3.5.0/lucene/contrib/benchmark/src/test/org/apache/lucene/benchmark/BenchmarkTestCase.java
new file mode 100644 (file)
index 0000000..2af8f93
--- /dev/null
+++ b/lucene-java-3.5.0/lucene/contrib/benchmark/src/test/org/apache/lucene/benchmark/BenchmarkTestCase.java
@@ -0,0 +1,116 @@
+package org.apache.lucene.benchmark;
+
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.io.StringReader;
+
+import org.apache.lucene.benchmark.byTask.Benchmark;
+import org.apache.lucene.util.LuceneTestCase;
+import org.apache.lucene.util._TestUtil;
+import org.junit.AfterClass;
+import org.junit.BeforeClass;
+
+/** Base class for all Benchmark unit tests. */
+public abstract class BenchmarkTestCase extends LuceneTestCase {
+  private static File WORKDIR;
+  
+  @BeforeClass
+  public static void beforeClassBenchmarkTestCase() throws Exception {
+    WORKDIR = _TestUtil.getTempDir("benchmark");
+    WORKDIR.delete();
+    WORKDIR.mkdirs();
+  }
+  
+  @AfterClass
+  public static void afterClassBenchmarkTestCase() throws Exception {
+    WORKDIR = null;
+  }
+  
+  public File getWorkDir() {
+    return WORKDIR;
+  }
+  
+  /** Copy a resource into the workdir */
+  public void copyToWorkDir(String resourceName) throws IOException {
+    InputStream resource = getClass().getResourceAsStream(resourceName);
+    OutputStream dest = new FileOutputStream(new File(getWorkDir(), resourceName));
+    byte[] buffer = new byte[8192];
+    int len;
+    
+    while ((len = resource.read(buffer)) > 0) {
+        dest.write(buffer, 0, len);
+    }
+
+    resource.close();
+    dest.close();
+  }
+  
+  /** Return a path, suitable for a .alg config file, for a resource in the workdir */
+  public String getWorkDirResourcePath(String resourceName) {
+    return new File(getWorkDir(), resourceName).getAbsolutePath().replace("\\", "/");
+  }
+  
+  /** Return a path, suitable for a .alg config file, for the workdir */
+  public String getWorkDirPath() {
+    return getWorkDir().getAbsolutePath().replace("\\", "/");
+  }
+  
+  // create the benchmark and execute it. 
+  public Benchmark execBenchmark(String[] algLines) throws Exception {
+    String algText = algLinesToText(algLines);
+    logTstLogic(algText);
+    Benchmark benchmark = new Benchmark(new StringReader(algText));
+    benchmark.execute();
+    return benchmark;
+  }
+  
+  // properties in effect in all tests here
+  final String propLines [] = {
+    "work.dir=" + getWorkDirPath(),
+    "directory=RAMDirectory",
+    "print.props=false",
+  };
+  
+  static final String NEW_LINE = System.getProperty("line.separator");
+  
+  // catenate alg lines to make the alg text
+  private String algLinesToText(String[] algLines) {
+    String indent = "  ";
+    StringBuilder sb = new StringBuilder();
+    for (int i = 0; i < propLines.length; i++) {
+      sb.append(indent).append(propLines[i]).append(NEW_LINE);
+    }
+    for (int i = 0; i < algLines.length; i++) {
+      sb.append(indent).append(algLines[i]).append(NEW_LINE);
+    }
+    return sb.toString();
+  }
+
+  private static void logTstLogic (String txt) {
+    if (!VERBOSE) 
+      return;
+    System.out.println("Test logic of:");
+    System.out.println(txt);
+  }
+
+}