import static com.google.common.base.Preconditions.checkState;
import com.google.common.annotations.Beta;
-import com.google.common.collect.ClassToInstanceMap;
-import com.google.common.collect.ImmutableClassToInstanceMap;
import com.google.common.collect.ImmutableMap;
-import com.google.common.collect.ImmutableSet;
-import com.google.common.collect.MutableClassToInstanceMap;
import com.google.common.util.concurrent.ListeningExecutorService;
import com.google.common.util.concurrent.MoreExecutors;
+import java.util.Set;
import javassist.ClassPool;
import org.opendaylight.controller.md.sal.binding.api.DataBroker;
import org.opendaylight.controller.md.sal.binding.api.MountPointService;
import org.opendaylight.controller.md.sal.binding.api.NotificationService;
import org.opendaylight.controller.md.sal.binding.compat.HeliumNotificationProviderServiceAdapter;
import org.opendaylight.controller.md.sal.binding.compat.HeliumRpcProviderRegistry;
-import org.opendaylight.controller.md.sal.binding.compat.HydrogenDataBrokerAdapter;
-import org.opendaylight.controller.md.sal.binding.compat.HydrogenMountProvisionServiceAdapter;
import org.opendaylight.controller.md.sal.binding.impl.BindingDOMDataBrokerAdapter;
import org.opendaylight.controller.md.sal.binding.impl.BindingDOMMountPointServiceAdapter;
import org.opendaylight.controller.md.sal.binding.impl.BindingDOMNotificationPublishServiceAdapter;
import org.opendaylight.controller.md.sal.dom.store.impl.InMemoryDOMDataStore;
import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
-import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
-import org.opendaylight.controller.sal.binding.api.mount.MountProviderService;
import org.opendaylight.controller.sal.binding.impl.RootBindingAwareBroker;
-import org.opendaylight.controller.sal.core.api.Broker.ProviderSession;
-import org.opendaylight.controller.sal.core.api.BrokerService;
import org.opendaylight.controller.sal.core.spi.data.DOMStore;
-import org.opendaylight.controller.sal.dom.broker.BrokerImpl;
-import org.opendaylight.yangtools.binding.data.codec.gen.impl.DataObjectSerializerGenerator;
-import org.opendaylight.yangtools.binding.data.codec.gen.impl.StreamWriterGenerator;
-import org.opendaylight.yangtools.binding.data.codec.impl.BindingNormalizedNodeCodecRegistry;
-import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy;
-import org.opendaylight.yangtools.sal.binding.generator.impl.ModuleInfoBackedContext;
-import org.opendaylight.yangtools.sal.binding.generator.util.JavassistUtils;
+import org.opendaylight.mdsal.binding.dom.codec.gen.impl.DataObjectSerializerGenerator;
+import org.opendaylight.mdsal.binding.dom.codec.gen.impl.StreamWriterGenerator;
+import org.opendaylight.mdsal.binding.dom.codec.impl.BindingNormalizedNodeCodecRegistry;
+import org.opendaylight.mdsal.binding.generator.impl.GeneratedClassLoadingStrategy;
+import org.opendaylight.mdsal.binding.generator.impl.ModuleInfoBackedContext;
+import org.opendaylight.mdsal.binding.generator.util.JavassistUtils;
import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
@Beta
public class BindingTestContext implements AutoCloseable {
-
-
private BindingToNormalizedNodeCodec codec;
private RootBindingAwareBroker baBrokerImpl;
private HeliumNotificationProviderServiceAdapter baNotifyImpl;
- private BrokerImpl biBrokerImpl;
-
private final ListeningExecutorService executor;
private final ClassPool classPool;
private ImmutableMap<LogicalDatastoreType, DOMStore> newDatastores;
- @Deprecated
- private DataProviderService baData;
-
private DOMDataBroker newDOMDataBroker;
private final MockSchemaService mockSchemaService = new MockSchemaService();
private BindingDOMRpcProviderServiceAdapter baProviderRpc;
private DOMRpcRouter domRouter;
+ private org.opendaylight.mdsal.dom.broker.DOMRpcRouter delegateDomRouter;
private NotificationPublishService publishService;
private DOMNotificationService domListenService;
-
+ private Set<YangModuleInfo> schemaModuleInfos;
public DOMDataBroker getDomAsyncDataBroker() {
- return newDOMDataBroker;
+ return this.newDOMDataBroker;
}
public BindingToNormalizedNodeCodec getCodec() {
- return codec;
+ return this.codec;
}
- protected BindingTestContext(final ListeningExecutorService executor, final ClassPool classPool, final boolean startWithSchema) {
+ protected BindingTestContext(final ListeningExecutorService executor, final ClassPool classPool,
+ final boolean startWithSchema) {
this.executor = executor;
this.classPool = classPool;
this.startWithSchema = startWithSchema;
}
public void startNewDataBroker() {
- checkState(executor != null, "Executor needs to be set");
- checkState(newDOMDataBroker != null, "DOM Data Broker must be set");
- dataBroker = new BindingDOMDataBrokerAdapter(newDOMDataBroker, codec);
+ checkState(this.executor != null, "Executor needs to be set");
+ checkState(this.newDOMDataBroker != null, "DOM Data Broker must be set");
+ this.dataBroker = new BindingDOMDataBrokerAdapter(this.newDOMDataBroker, this.codec);
}
public void startNewDomDataBroker() {
- checkState(executor != null, "Executor needs to be set");
- final InMemoryDOMDataStore operStore = new InMemoryDOMDataStore("OPER", MoreExecutors.sameThreadExecutor());
- final InMemoryDOMDataStore configStore = new InMemoryDOMDataStore("CFG", MoreExecutors.sameThreadExecutor());
- newDatastores = ImmutableMap.<LogicalDatastoreType, DOMStore>builder()
+ checkState(this.executor != null, "Executor needs to be set");
+ final InMemoryDOMDataStore operStore = new InMemoryDOMDataStore("OPER",
+ MoreExecutors.newDirectExecutorService());
+ final InMemoryDOMDataStore configStore = new InMemoryDOMDataStore("CFG",
+ MoreExecutors.newDirectExecutorService());
+ this.newDatastores = ImmutableMap.<LogicalDatastoreType, DOMStore>builder()
.put(LogicalDatastoreType.OPERATIONAL, operStore)
.put(LogicalDatastoreType.CONFIGURATION, configStore)
.build();
- newDOMDataBroker = new SerializedDOMDataBroker(newDatastores, executor);
+ this.newDOMDataBroker = new SerializedDOMDataBroker(this.newDatastores, this.executor);
- mockSchemaService.registerSchemaContextListener(configStore);
- mockSchemaService.registerSchemaContextListener(operStore);
+ this.mockSchemaService.registerSchemaContextListener(configStore);
+ this.mockSchemaService.registerSchemaContextListener(operStore);
}
public void startBindingDataBroker() {
}
public void startBindingBroker() {
- checkState(executor != null, "Executor needs to be set");
- checkState(baData != null, "Binding Data Broker must be started");
- checkState(baNotifyImpl != null, "Notification Service must be started");
-
- baConsumerRpc = new BindingDOMRpcServiceAdapter(getDomRpcInvoker(), codec);
- baProviderRpc = new BindingDOMRpcProviderServiceAdapter(getDomRpcRegistry(), codec);
-
- baBrokerImpl = new RootBindingAwareBroker("test");
-
- final MountPointService mountService = new BindingDOMMountPointServiceAdapter(biMountImpl, codec);
- baBrokerImpl.setMountService(mountService);
- baBrokerImpl.setLegacyMountManager(new HydrogenMountProvisionServiceAdapter(mountService));
- baBrokerImpl.setRpcBroker(new HeliumRpcProviderRegistry(baConsumerRpc, baProviderRpc));
- baBrokerImpl.setLegacyDataBroker(baData);
- baBrokerImpl.setNotificationBroker(baNotifyImpl);
- baBrokerImpl.start();
- }
+ checkState(this.executor != null, "Executor needs to be set");
+ checkState(this.baNotifyImpl != null, "Notification Service must be started");
- public void startForwarding() {
+ this.baConsumerRpc = new BindingDOMRpcServiceAdapter(getDomRpcInvoker(), this.codec);
+ this.baProviderRpc = new BindingDOMRpcProviderServiceAdapter(getDomRpcRegistry(), this.codec);
- }
-
- private ProviderSession createMockContext() {
-
- final ClassToInstanceMap<BrokerService> domBrokerServices = ImmutableClassToInstanceMap
- .<BrokerService> builder()
- //
- .put(DOMRpcRouter.class, biBrokerImpl.getRouter()) //
- .put(DOMMountPointService.class, biMountImpl)
- .build();
-
- return new ProviderSession() {
+ this.baBrokerImpl = new RootBindingAwareBroker("test");
- @Override
- public <T extends BrokerService> T getService(final Class<T> service) {
- return domBrokerServices.getInstance(service);
- }
+ final MountPointService mountService = new BindingDOMMountPointServiceAdapter(this.biMountImpl, this.codec);
+ this.baBrokerImpl.setMountService(mountService);
+ this.baBrokerImpl.setRpcBroker(new HeliumRpcProviderRegistry(this.baConsumerRpc, this.baProviderRpc));
+ this.baBrokerImpl.setNotificationBroker(this.baNotifyImpl);
+ this.baBrokerImpl.start();
+ }
- @Override
- public boolean isClosed() {
- return false;
- }
+ public void startForwarding() {
- @Override
- public void close() {
- }
- };
}
public void startBindingToDomMappingService() {
- checkState(classPool != null, "ClassPool needs to be present");
+ checkState(this.classPool != null, "ClassPool needs to be present");
- final DataObjectSerializerGenerator generator = StreamWriterGenerator.create(JavassistUtils.forClassPool(classPool));
+ final DataObjectSerializerGenerator generator = StreamWriterGenerator.create(
+ JavassistUtils.forClassPool(this.classPool));
final BindingNormalizedNodeCodecRegistry codecRegistry = new BindingNormalizedNodeCodecRegistry(generator);
final GeneratedClassLoadingStrategy loading = GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy();
- codec = new BindingToNormalizedNodeCodec(loading, codecRegistry);
- mockSchemaService.registerSchemaContextListener(codec);
+ this.codec = new BindingToNormalizedNodeCodec(loading, codecRegistry);
+ this.mockSchemaService.registerSchemaContextListener(this.codec);
}
- private void updateYangSchema(final ImmutableSet<YangModuleInfo> moduleInfos) {
- mockSchemaService.changeSchema(getContext(moduleInfos));
+ private void updateYangSchema(final Set<YangModuleInfo> moduleInfos) {
+ this.mockSchemaService.changeSchema(getContext(moduleInfos));
}
- private SchemaContext getContext(final ImmutableSet<YangModuleInfo> moduleInfos) {
+ private static SchemaContext getContext(final Set<YangModuleInfo> moduleInfos) {
final ModuleInfoBackedContext ctx = ModuleInfoBackedContext.create();
ctx.addModuleInfos(moduleInfos);
return ctx.tryToCreateSchemaContext().get();
startDomMountPoint();
startBindingToDomMappingService();
startNewDataBroker();
- startNewBindingDataBroker();
startBindingNotificationBroker();
startBindingBroker();
startForwarding();
- if (startWithSchema) {
+
+ if (schemaModuleInfos != null) {
+ updateYangSchema(schemaModuleInfos);
+ } else if (this.startWithSchema) {
loadYangSchemaFromClasspath();
}
}
- public void startNewBindingDataBroker() {
- final HydrogenDataBrokerAdapter forwarded = new HydrogenDataBrokerAdapter(dataBroker);
- baData = forwarded;
- }
-
private void startDomMountPoint() {
- biMountImpl = new DOMMountPointServiceImpl();
+ this.biMountImpl = new DOMMountPointServiceImpl();
}
private void startDomBroker() {
- checkState(executor != null);
-
- domRouter = new DOMRpcRouter();
- mockSchemaService.registerSchemaContextListener(domRouter);
-
- final ClassToInstanceMap<BrokerService> services = MutableClassToInstanceMap.create();
- services.put(DOMRpcService.class, domRouter);
-
- biBrokerImpl = new BrokerImpl(domRouter,services);
+ checkState(this.executor != null);
+ delegateDomRouter = org.opendaylight.mdsal.dom.broker.DOMRpcRouter.newInstance(mockSchemaService);
+ this.domRouter = new DOMRpcRouter(delegateDomRouter.getRpcService(), delegateDomRouter.getRpcProviderService());
}
public void startBindingNotificationBroker() {
- checkState(executor != null);
+ checkState(this.executor != null);
final DOMNotificationRouter router = DOMNotificationRouter.create(16);
- domPublishService = router;
- domListenService = router;
- publishService = new BindingDOMNotificationPublishServiceAdapter(codec, domPublishService);
- listenService = new BindingDOMNotificationServiceAdapter(codec, domListenService);
- baNotifyImpl = new HeliumNotificationProviderServiceAdapter(publishService,listenService);
+ this.domPublishService = router;
+ this.domListenService = router;
+ this.publishService = new BindingDOMNotificationPublishServiceAdapter(this.codec, this.domPublishService);
+ this.listenService = new BindingDOMNotificationServiceAdapter(this.codec, this.domListenService);
+ this.baNotifyImpl = new HeliumNotificationProviderServiceAdapter(this.publishService,this.listenService);
}
public void loadYangSchemaFromClasspath() {
- final ImmutableSet<YangModuleInfo> moduleInfos = BindingReflections.loadModuleInfos();
- updateYangSchema(moduleInfos);
- }
-
- @Deprecated
- public DataProviderService getBindingDataBroker() {
- return baData;
+ updateYangSchema(BindingReflections.loadModuleInfos());
}
public RpcProviderRegistry getBindingRpcRegistry() {
- return baBrokerImpl.getRoot();
+ return this.baBrokerImpl.getRoot();
}
public DOMRpcProviderService getDomRpcRegistry() {
- return domRouter;
+ return this.domRouter;
}
public DOMRpcService getDomRpcInvoker() {
- return domRouter;
+ return this.domRouter;
+ }
+
+ public org.opendaylight.mdsal.dom.broker.DOMRpcRouter getDelegateDomRouter() {
+ return delegateDomRouter;
}
@Override
}
- public MountProviderService getBindingMountProviderService() {
- return baBrokerImpl.getLegacyMount();
+ public MountPointService getBindingMountPointService() {
+ return this.baBrokerImpl.getMountService();
}
public DOMMountPointService getDomMountProviderService() {
- return biMountImpl;
+ return this.biMountImpl;
}
public DataBroker getDataBroker() {
- return dataBroker;
+ return this.dataBroker;
}
-
+ public void setSchemaModuleInfos(Set<YangModuleInfo> moduleInfos) {
+ this.schemaModuleInfos = moduleInfos;
+ }
}