infra/it/it-test/src/test/java/io/fd/honeycomb/data/impl/AbstractInfraTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.data.impl;
  18
  19 import static org.mockito.Mockito.when;
  20
  21 import com.google.common.collect.HashMultimap;
  22 import com.google.common.collect.Multimap;
  23 import com.google.common.util.concurrent.Futures;
  24 import java.util.Arrays;
  25 import java.util.Map;
  26 import javassist.ClassPool;
  27 import org.junit.Before;
  28 import org.mockito.Mock;
  29 import org.mockito.MockitoAnnotations;
  30 import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodec;
  31 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
  32 import org.opendaylight.mdsal.binding.dom.codec.gen.impl.DataObjectSerializerGenerator;
  33 import org.opendaylight.mdsal.binding.dom.codec.gen.impl.StreamWriterGenerator;
  34 import org.opendaylight.mdsal.binding.dom.codec.impl.BindingNormalizedNodeCodecRegistry;
  35 import org.opendaylight.mdsal.binding.generator.impl.ModuleInfoBackedContext;
  36 import org.opendaylight.mdsal.binding.generator.util.BindingRuntimeContext;
  37 import org.opendaylight.mdsal.binding.generator.util.JavassistUtils;
  38 import org.opendaylight.yangtools.yang.binding.DataObject;
  39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  40 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  41 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  42 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerNode;
  43 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  44 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  45
  46 /**
  47  * Base IT test infrastructure.
  48  */
  49 abstract class AbstractInfraTest {
  50
  51     protected BindingNormalizedNodeSerializer serializer;
  52     protected SchemaContext schemaContext;
  53
  54     @Mock
  55     protected org.opendaylight.controller.md.sal.binding.api.DataBroker contextBroker;
  56     @Mock
  57     private org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction ctxTx;
  58
  59     static BindingToNormalizedNodeCodec getSerializer(final ModuleInfoBackedContext moduleInfoBackedContext,
  60                                                       final SchemaContext schemaContext) {
  61         final DataObjectSerializerGenerator serializerGenerator =
  62             StreamWriterGenerator.create(JavassistUtils.forClassPool(ClassPool.getDefault()));
  63         final BindingNormalizedNodeCodecRegistry codecRegistry = new BindingNormalizedNodeCodecRegistry(serializerGenerator);
  64         final BindingRuntimeContext ctx =
  65                 BindingRuntimeContext.create(moduleInfoBackedContext, schemaContext);
  66         codecRegistry.onBindingRuntimeContextUpdated(ctx);
  67         return new BindingToNormalizedNodeCodec(moduleInfoBackedContext, codecRegistry);
  68     }
  69
  70     static ModuleInfoBackedContext getSchemaContext() {
  71         final ModuleInfoBackedContext moduleInfoBackedContext = ModuleInfoBackedContext.create();
  72         moduleInfoBackedContext.addModuleInfos(Arrays.asList(
  73             org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.hc.test.rev150105.$YangModuleInfoImpl.getInstance(),
  74             org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.hc.aug.test.rev170109.$YangModuleInfoImpl.getInstance(),
  75             org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.hc.subtree.test.rev180116.$YangModuleInfoImpl.getInstance()
  76
  77         ));
  78         return moduleInfoBackedContext;
  79     }
  80
  81     @Before
  82     public void setUp() throws Exception {
  83         MockitoAnnotations.initMocks(this);
  84         when(contextBroker.newReadWriteTransaction()).thenReturn(ctxTx);
  85         when(ctxTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
  86
  87         initSerializer();
  88         postSetup();
  89     }
  90
  91     abstract void postSetup();
  92
  93     private void initSerializer() {
  94         final ModuleInfoBackedContext moduleInfoBackedContext = getSchemaContext();
  95         schemaContext = moduleInfoBackedContext.tryToCreateSchemaContext().get();
  96         serializer = getSerializer(moduleInfoBackedContext, schemaContext);
  97     }
  98
  99     protected Multimap<InstanceIdentifier<? extends DataObject>, ? extends DataObject> toBinding(
 100             final NormalizedNode<?, ?> read) {
 101         Multimap<InstanceIdentifier<? extends DataObject>, DataObject> baNodes = HashMultimap.create();
 102
 103         for (DataContainerChild<?, ?> o : ((DataContainerNode<?>) read).getValue()) {
 104             final YangInstanceIdentifier yid = YangInstanceIdentifier.of(o.getNodeType());
 105             final Map.Entry<InstanceIdentifier<?>, DataObject> baEntry = serializer.fromNormalizedNode(yid, o);
 106             baNodes.put(baEntry.getKey(), baEntry.getValue());
 107         }
 108         return baNodes;
 109     }
 110 }