binding/mdsal-binding-dom-adapter/src/test/java/org/opendaylight/mdsal/binding/dom/adapter/test/util/BindingTestContext.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.mdsal.binding.dom.adapter.test.util;
   9
  10 import static com.google.common.base.Preconditions.checkState;
  11
  12 import com.google.common.annotations.Beta;
  13 import com.google.common.collect.ImmutableMap;
  14 import com.google.common.collect.ImmutableSet;
  15 import com.google.common.util.concurrent.ListeningExecutorService;
  16 import com.google.common.util.concurrent.MoreExecutors;
  17 import java.util.Set;
  18 import org.opendaylight.mdsal.binding.api.DataBroker;
  19 import org.opendaylight.mdsal.binding.api.MountPointService;
  20 import org.opendaylight.mdsal.binding.api.NotificationPublishService;
  21 import org.opendaylight.mdsal.binding.api.NotificationService;
  22 import org.opendaylight.mdsal.binding.api.RpcConsumerRegistry;
  23 import org.opendaylight.mdsal.binding.api.RpcProviderService;
  24 import org.opendaylight.mdsal.binding.dom.adapter.AdapterContext;
  25 import org.opendaylight.mdsal.binding.dom.adapter.BindingDOMDataBrokerAdapter;
  26 import org.opendaylight.mdsal.binding.dom.adapter.BindingDOMMountPointServiceAdapter;
  27 import org.opendaylight.mdsal.binding.dom.adapter.BindingDOMNotificationPublishServiceAdapter;
  28 import org.opendaylight.mdsal.binding.dom.adapter.BindingDOMNotificationServiceAdapter;
  29 import org.opendaylight.mdsal.binding.dom.adapter.BindingDOMRpcProviderServiceAdapter;
  30 import org.opendaylight.mdsal.binding.dom.adapter.BindingDOMRpcServiceAdapter;
  31 import org.opendaylight.mdsal.binding.runtime.spi.BindingRuntimeHelpers;
  32 import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
  33 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  34 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
  35 import org.opendaylight.mdsal.dom.api.DOMMountPointService;
  36 import org.opendaylight.mdsal.dom.api.DOMNotificationPublishService;
  37 import org.opendaylight.mdsal.dom.api.DOMNotificationService;
  38 import org.opendaylight.mdsal.dom.api.DOMRpcProviderService;
  39 import org.opendaylight.mdsal.dom.api.DOMRpcService;
  40 import org.opendaylight.mdsal.dom.broker.DOMMountPointServiceImpl;
  41 import org.opendaylight.mdsal.dom.broker.DOMNotificationRouter;
  42 import org.opendaylight.mdsal.dom.broker.DOMRpcRouter;
  43 import org.opendaylight.mdsal.dom.broker.SerializedDOMDataBroker;
  44 import org.opendaylight.mdsal.dom.spi.store.DOMStore;
  45 import org.opendaylight.mdsal.dom.store.inmemory.InMemoryDOMDataStore;
  46 import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
  47 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  48
  49 @Beta
  50 public class BindingTestContext implements AutoCloseable {
  51     private final ListeningExecutorService executor;
  52
  53     private final boolean startWithSchema;
  54
  55     private DOMMountPointService biMountImpl;
  56
  57     private ImmutableMap<LogicalDatastoreType, DOMStore> newDatastores;
  58
  59     private DOMDataBroker newDOMDataBroker;
  60
  61     private final MockSchemaService mockSchemaService = new MockSchemaService();
  62
  63     private DataBroker dataBroker;
  64
  65     private RpcConsumerRegistry baConsumerRpc;
  66
  67     private BindingDOMRpcProviderServiceAdapter baProviderRpc;
  68     private DOMRpcRouter domRouter;
  69
  70     private NotificationPublishService publishService;
  71
  72     private NotificationService listenService;
  73
  74     private DOMNotificationPublishService domPublishService;
  75
  76     private DOMNotificationService domListenService;
  77
  78     private Set<YangModuleInfo> schemaModuleInfos;
  79
  80     public DOMDataBroker getDomAsyncDataBroker() {
  81         return newDOMDataBroker;
  82     }
  83
  84     public AdapterContext getCodec() {
  85         return mockSchemaService;
  86     }
  87
  88     protected BindingTestContext(final ListeningExecutorService executor, final boolean startWithSchema) {
  89         this.executor = executor;
  90         this.startWithSchema = startWithSchema;
  91     }
  92
  93     public void startDomDataBroker() {
  94     }
  95
  96     public void startNewDataBroker() {
  97         checkState(executor != null, "Executor needs to be set");
  98         checkState(newDOMDataBroker != null, "DOM Data Broker must be set");
  99         dataBroker = new BindingDOMDataBrokerAdapter(mockSchemaService, newDOMDataBroker);
 100     }
 101
 102     public void startNewDomDataBroker() {
 103         checkState(executor != null, "Executor needs to be set");
 104         final InMemoryDOMDataStore operStore = new InMemoryDOMDataStore("OPER",
 105             MoreExecutors.newDirectExecutorService());
 106         final InMemoryDOMDataStore configStore = new InMemoryDOMDataStore("CFG",
 107             MoreExecutors.newDirectExecutorService());
 108         newDatastores = ImmutableMap.<LogicalDatastoreType, DOMStore>builder()
 109                 .put(LogicalDatastoreType.OPERATIONAL, operStore)
 110                 .put(LogicalDatastoreType.CONFIGURATION, configStore)
 111                 .build();
 112
 113         newDOMDataBroker = new SerializedDOMDataBroker(newDatastores, executor);
 114
 115         mockSchemaService.registerSchemaContextListener(configStore);
 116         mockSchemaService.registerSchemaContextListener(operStore);
 117     }
 118
 119     public void startBindingDataBroker() {
 120
 121     }
 122
 123     public void startBindingBroker() {
 124         checkState(executor != null, "Executor needs to be set");
 125
 126         baConsumerRpc = new BindingDOMRpcServiceAdapter(mockSchemaService, getDomRpcInvoker());
 127         baProviderRpc = new BindingDOMRpcProviderServiceAdapter(mockSchemaService, getDomRpcRegistry());
 128         final MountPointService mountService = new BindingDOMMountPointServiceAdapter(mockSchemaService, biMountImpl);
 129     }
 130
 131     public void startForwarding() {
 132
 133     }
 134
 135     public void startBindingToDomMappingService() {
 136         // No-op, really
 137     }
 138
 139     private void updateYangSchema(final Set<YangModuleInfo> moduleInfos) {
 140         mockSchemaService.changeSchema(BindingRuntimeHelpers.createRuntimeContext(moduleInfos));
 141     }
 142
 143     public EffectiveModelContext getContext() {
 144         return mockSchemaService.getEffectiveModelContext();
 145     }
 146
 147     public void start() {
 148         startNewDomDataBroker();
 149
 150         startDomBroker();
 151         startDomMountPoint();
 152         startBindingToDomMappingService();
 153         startNewDataBroker();
 154         startBindingNotificationBroker();
 155         startBindingBroker();
 156
 157         startForwarding();
 158
 159         if (schemaModuleInfos != null) {
 160             updateYangSchema(schemaModuleInfos);
 161         } else if (startWithSchema) {
 162             loadYangSchemaFromClasspath();
 163         }
 164     }
 165
 166     private void startDomMountPoint() {
 167         biMountImpl = new DOMMountPointServiceImpl();
 168     }
 169
 170     private void startDomBroker() {
 171         checkState(executor != null);
 172         domRouter = new DOMRpcRouter();
 173         mockSchemaService.registerSchemaContextListener(domRouter);
 174     }
 175
 176     public void startBindingNotificationBroker() {
 177         checkState(executor != null);
 178         final DOMNotificationRouter router = DOMNotificationRouter.create(16);
 179         domPublishService = router;
 180         domListenService = router;
 181         publishService = new BindingDOMNotificationPublishServiceAdapter(mockSchemaService, domPublishService);
 182         listenService = new BindingDOMNotificationServiceAdapter(mockSchemaService, domListenService);
 183
 184     }
 185
 186     public void loadYangSchemaFromClasspath() {
 187         final ImmutableSet<YangModuleInfo> moduleInfos = BindingReflections.loadModuleInfos();
 188         updateYangSchema(moduleInfos);
 189     }
 190
 191     public DOMRpcProviderService getDomRpcRegistry() {
 192         return domRouter.getRpcProviderService();
 193     }
 194
 195     public DOMRpcService getDomRpcInvoker() {
 196         return domRouter.getRpcService();
 197     }
 198
 199     public RpcProviderService getBindingRpcProviderRegistry() {
 200         return baProviderRpc;
 201     }
 202
 203     public RpcConsumerRegistry getBindingRpcConsumerRegistry() {
 204         return baConsumerRpc;
 205     }
 206
 207     @Override
 208     public void close() throws Exception {
 209
 210     }
 211
 212     public DOMMountPointService getDomMountProviderService() {
 213         return biMountImpl;
 214     }
 215
 216     public DataBroker getDataBroker() {
 217         return dataBroker;
 218     }
 219
 220     public void setSchemaModuleInfos(final Set<YangModuleInfo> moduleInfos) {
 221         schemaModuleInfos = moduleInfos;
 222     }
 223 }