searchloadtest.java

Lucene in Action 中文版代码下载

package lia.extsearch.perf;

import com.clarkware.junitperf.LoadTest;
import com.clarkware.junitperf.TimedTest;
import junit.extensions.TestSetup;
import junit.framework.Test;
import junit.framework.TestSuite;

/**
 * @author <b>Mike Clark</b>
 * @author Clarkware Consulting, Inc.
 */
public class SearchLoadTest {

  public static Test suite() {

    int maxTimeInMillis = 100;
    int concurrentUsers = 10;

    //Test test = new SearchTest("testSearchByTimestamp");
    Test test = new SearchTest("testSearchByDay");

    TestSuite suite = new TestSuite();
    suite.addTest(test); // runs first to warm up the cache
    Test timedTest = new TimedTest(test, maxTimeInMillis);
    LoadTest loadTest = new LoadTest(timedTest, concurrentUsers);
    suite.addTest(loadTest);

    return suite;

    //
    // Use this to ensure that the index is
    // created before the performance test.
    //
    //return makeTestFixture(suite);
  }

  public static Test makeTestFixture(Test test) {

    TestSetup oneTimer = new TestSetup(test) {

      public void setUp() throws Exception {
        IndexBuilder builder = new IndexBuilder();
        builder.buildIndex(1000);
      }
    };

    return oneTimer;
  }

  public static void main(String args[]) {
    junit.textui.TestRunner.run(suite());
  }
}