opendaylight/md-sal/sal-binding-broker/src/test/java/org/opendaylight/controller/sal/binding/test/util/BindingBrokerTestFactory.java

   1 package org.opendaylight.controller.sal.binding.test.util;
   2
   3 import java.util.concurrent.ExecutorService;
   4
   5 import javassist.ClassPool;
   6
   7 import org.opendaylight.controller.sal.core.api.data.DataStore;
   8 import org.opendaylight.controller.sal.dom.broker.impl.DataStoreStatsWrapper;
   9 import org.opendaylight.controller.sal.dom.broker.impl.HashMapDataStore;
  10 import org.opendaylight.controller.sal.dom.broker.impl.SchemaAwareDataStoreAdapter;
  11
  12 import com.google.common.base.Preconditions;
  13 import com.google.common.util.concurrent.ListeningExecutorService;
  14 import com.google.common.util.concurrent.MoreExecutors;
  15
  16 public class BindingBrokerTestFactory {
  17
  18     private static final ClassPool CLASS_POOL = new ClassPool();
  19     private boolean startWithParsedSchema = true;
  20     private ExecutorService executor;
  21     private ClassPool classPool;
  22
  23
  24     public boolean isStartWithParsedSchema() {
  25         return startWithParsedSchema;
  26     }
  27
  28     public void setStartWithParsedSchema(boolean startWithParsedSchema) {
  29         this.startWithParsedSchema = startWithParsedSchema;
  30     }
  31
  32     public ExecutorService getExecutor() {
  33         return executor;
  34     }
  35
  36     public void setExecutor(ExecutorService executor) {
  37         this.executor = executor;
  38     }
  39
  40
  41     public BindingTestContext getTestContext() {
  42         Preconditions.checkState(executor != null, "Executor is not set.");
  43         ListeningExecutorService listenableExecutor = MoreExecutors.listeningDecorator(executor);
  44         return new BindingTestContext(listenableExecutor, getClassPool(),startWithParsedSchema);
  45     }
  46
  47     public ClassPool getClassPool() {
  48         if(classPool == null) {
  49             return CLASS_POOL;
  50         }
  51
  52         return classPool;
  53     }
  54
  55     public void setClassPool(ClassPool classPool) {
  56         this.classPool = classPool;
  57     }
  58
  59 }