lucene-java-3.4.0/lucene/contrib/benchmark/src/test/org/apache/lucene/benchmark/byTask/tasks/PerfTaskTest.java

   1 package org.apache.lucene.benchmark.byTask.tasks;
   2
   3 /**
   4  * Licensed to the Apache Software Foundation (ASF) under one or more
   5  * contributor license agreements.  See the NOTICE file distributed with
   6  * this work for additional information regarding copyright ownership.
   7  * The ASF licenses this file to You under the Apache License, Version 2.0
   8  * (the "License"); you may not use this file except in compliance with
   9  * the License.  You may obtain a copy of the License at
  10  *
  11  *     http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  */
  19
  20 import java.util.Properties;
  21
  22 import org.apache.lucene.benchmark.BenchmarkTestCase;
  23 import org.apache.lucene.benchmark.byTask.PerfRunData;
  24 import org.apache.lucene.benchmark.byTask.utils.Config;
  25
  26 /** Tests the functionality of the abstract {@link PerfTask}. */
  27 public class PerfTaskTest extends BenchmarkTestCase {
  28
  29   private static final class MyPerfTask extends PerfTask {
  30
  31     public MyPerfTask(PerfRunData runData) {
  32       super(runData);
  33     }
  34
  35     @Override
  36     public int doLogic() throws Exception {
  37       return 0;
  38     }
  39
  40     public int getLogStep() { return logStep; }
  41
  42   }
  43
  44   private PerfRunData createPerfRunData(boolean setLogStep, int logStepVal,
  45       boolean setTaskLogStep, int taskLogStepVal) throws Exception {
  46     Properties props = new Properties();
  47     if (setLogStep) {
  48       props.setProperty("log.step", Integer.toString(logStepVal));
  49     }
  50     if (setTaskLogStep) {
  51       props.setProperty("log.step.MyPerf", Integer.toString(taskLogStepVal));
  52     }
  53     props.setProperty("directory", "RAMDirectory"); // no accidental FS dir.
  54     Config config = new Config(props);
  55     return new PerfRunData(config);
  56   }
  57
  58   private void doLogStepTest(boolean setLogStep, int logStepVal,
  59       boolean setTaskLogStep, int taskLogStepVal, int expLogStepValue) throws Exception {
  60     PerfRunData runData = createPerfRunData(setLogStep, logStepVal, setTaskLogStep, taskLogStepVal);
  61     MyPerfTask mpt = new MyPerfTask(runData);
  62     assertEquals(expLogStepValue, mpt.getLogStep());
  63   }
  64
  65   public void testLogStep() throws Exception {
  66     doLogStepTest(false, -1, false, -1, PerfTask.DEFAULT_LOG_STEP);
  67     doLogStepTest(true, -1, false, -1, Integer.MAX_VALUE);
  68     doLogStepTest(true, 100, false, -1, 100);
  69     doLogStepTest(false, -1, true, -1, Integer.MAX_VALUE);
  70     doLogStepTest(false, -1, true, 100, 100);
  71   }
  72
  73 }