southbound/southbound-impl/src/test/java/org/opendaylight/ovsdb/southbound/reconciliation/configuration/BridgeConfigReconciliationTaskTest.java

   1 /*
   2  * Copyright © 2016, 2017 NEC Corporation 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.ovsdb.southbound.reconciliation.configuration;
   9
  10 import static org.junit.Assert.assertTrue;
  11 import static org.mockito.ArgumentMatchers.any;
  12 import static org.mockito.Mockito.doNothing;
  13 import static org.mockito.Mockito.mock;
  14 import static org.mockito.Mockito.spy;
  15 import static org.mockito.Mockito.verify;
  16 import static org.mockito.Mockito.when;
  17
  18 import com.google.common.base.Optional;
  19 import com.google.common.util.concurrent.CheckedFuture;
  20 import com.google.common.util.concurrent.Futures;
  21 import java.util.ArrayList;
  22 import java.util.Arrays;
  23 import java.util.Collections;
  24 import java.util.HashMap;
  25 import java.util.List;
  26 import java.util.Map;
  27 import org.junit.Before;
  28 import org.junit.Test;
  29 import org.junit.runner.RunWith;
  30 import org.mockito.Mock;
  31 import org.mockito.Mockito;
  32 import org.mockito.junit.MockitoJUnitRunner;
  33 import org.mockito.stubbing.Answer;
  34 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  35 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  36 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  37 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  38 import org.opendaylight.ovsdb.southbound.InstanceIdentifierCodec;
  39 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
  40 import org.opendaylight.ovsdb.southbound.OvsdbConnectionManager;
  41 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
  42 import org.opendaylight.ovsdb.southbound.SouthboundProvider;
  43 import org.opendaylight.ovsdb.southbound.reconciliation.ReconciliationManager;
  44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeName;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeProtocolBase;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeProtocolOpenflow10;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeRef;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntry;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntryKey;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ProtocolEntry;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ProtocolEntryKey;
  54 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  55 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  56 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  57 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  58 import org.opendaylight.yangtools.yang.binding.DataObject;
  59 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  60 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
  61
  62 @RunWith(MockitoJUnitRunner.class)
  63 public class BridgeConfigReconciliationTaskTest {
  64     private static final String BR_INT = "br-int";
  65     private static final String NODE_ID = "ovsdb://uuid/6ff3d0cf-4102-429d-b41c-f8027a0fd7f4";
  66     private BridgeConfigReconciliationTask configurationReconciliationTask;
  67     @Mock private OvsdbConnectionManager ovsdbConnectionManager;
  68     @Mock private OvsdbConnectionInstance ovsdbConnectionInstance;
  69     @Mock private DataBroker db;
  70     @Mock private ReconciliationManager reconciliationManager;
  71     @Mock private Topology topology;
  72     @Mock private InstanceIdentifier<Node> iid;
  73     @Mock private SouthboundProvider provider;
  74
  75     @Before
  76     public void setUp() throws Exception {
  77         NodeKey nodeKey = new NodeKey(new NodeId(new Uri(NODE_ID)));
  78         List<Node> bridgeNodes = new ArrayList<>();
  79
  80         iid = SouthboundMapper.createInstanceIdentifier(nodeKey.getNodeId());
  81         when(topology.getNode()).thenReturn(bridgeNodes);
  82         SouthboundProvider.setBridgesReconciliationInclusionList(Arrays.asList(BR_INT));
  83         Node brIntNode = createBridgeNode(NODE_ID + "/bridge/" + BR_INT);
  84         Optional<Node> nodeOptional = Optional.of(brIntNode);
  85         CheckedFuture<Optional<Node>, ReadFailedException> readNodeFuture =
  86                 Futures.immediateCheckedFuture(nodeOptional);
  87         when(reconciliationManager.getDb()).thenReturn(db);
  88         ReadOnlyTransaction tx = mock(ReadOnlyTransaction.class);
  89         Mockito.when(db.newReadOnlyTransaction()).thenReturn(tx);
  90         Mockito.when(tx.read(any(LogicalDatastoreType.class),any(InstanceIdentifier.class)))
  91                 .thenReturn(readNodeFuture);
  92         bridgeNodes.add(brIntNode);
  93
  94         configurationReconciliationTask =
  95                 new BridgeConfigReconciliationTask(reconciliationManager, ovsdbConnectionManager, iid,
  96                         ovsdbConnectionInstance, mock(InstanceIdentifierCodec.class));
  97     }
  98
  99     @Test
 100     public void testReconcileConfiguration() throws Exception {
 101         BridgeConfigReconciliationTask underTest = spy(configurationReconciliationTask);
 102         doNothing().when(underTest).reconcileBridgeConfigurations(any(Map.class));
 103         assertTrue(underTest.reconcileConfiguration(ovsdbConnectionManager));
 104         Map<InstanceIdentifier<?>, DataObject> changes = new HashMap<>();
 105         for (Node bridgeNode : topology.getNode()) {
 106             changes.putAll(createExpectedConfigurationChanges(bridgeNode));
 107         }
 108         verify(underTest).reconcileBridgeConfigurations(changes);
 109     }
 110
 111     private Node createBridgeNode(final String bridgeName) {
 112         Node bridgeNode = mock(Node.class);
 113         String nodeString = bridgeName;
 114         when(bridgeNode.getNodeId()).thenReturn(new NodeId(new Uri(nodeString)));
 115         OvsdbBridgeAugmentation ovsdbBridgeAugmentation = mock(OvsdbBridgeAugmentation.class);
 116         OvsdbNodeRef ovsdbNodeRef = mock(OvsdbNodeRef.class);
 117
 118         when((InstanceIdentifier<Node>)ovsdbNodeRef.getValue()).thenReturn(iid);
 119         OvsdbBridgeName ovsdbBridgeName = new OvsdbBridgeName(bridgeName);
 120         when(bridgeNode.augmentation(OvsdbBridgeAugmentation.class)).thenReturn(ovsdbBridgeAugmentation);
 121         ProtocolEntry protocolEntry = mock(ProtocolEntry.class);
 122         ProtocolEntryKey protocolEntryKey = mock(ProtocolEntryKey.class);
 123         Mockito.when(protocolEntry.getProtocol()).thenAnswer(
 124                 (Answer<Class<? extends OvsdbBridgeProtocolBase>>) invocation -> OvsdbBridgeProtocolOpenflow10.class);
 125         when(protocolEntry.key()).thenReturn(protocolEntryKey);
 126         when(ovsdbBridgeAugmentation.getProtocolEntry()).thenReturn(Collections.singletonList(protocolEntry));
 127
 128         ControllerEntry controllerEntry = mock(ControllerEntry.class);
 129         ControllerEntryKey controllerEntryKey = mock(ControllerEntryKey.class);
 130         when(controllerEntry.key()).thenReturn(controllerEntryKey);
 131         when(ovsdbBridgeAugmentation.getControllerEntry()).thenReturn(Collections.singletonList(controllerEntry));
 132
 133         when(ovsdbBridgeAugmentation.getManagedBy()).thenReturn(ovsdbNodeRef);
 134
 135         return bridgeNode;
 136     }
 137
 138     private Map<InstanceIdentifier<?>, DataObject> createExpectedConfigurationChanges(final Node bridgeNode) {
 139         OvsdbBridgeAugmentation ovsdbBridge = bridgeNode.augmentation(OvsdbBridgeAugmentation.class);
 140
 141         Map<InstanceIdentifier<?>, DataObject> changes = new HashMap<>();
 142         final InstanceIdentifier<Node> bridgeNodeIid =
 143                 SouthboundMapper.createInstanceIdentifier(bridgeNode.getNodeId());
 144         final InstanceIdentifier<OvsdbBridgeAugmentation> ovsdbBridgeIid =
 145                 bridgeNodeIid.builder().augmentation(OvsdbBridgeAugmentation.class).build();
 146         changes.put(bridgeNodeIid, bridgeNode);
 147         changes.put(ovsdbBridgeIid, ovsdbBridge);
 148         for (ProtocolEntry protocolEntry : ovsdbBridge.getProtocolEntry()) {
 149             KeyedInstanceIdentifier<ProtocolEntry, ProtocolEntryKey> protocolIid =
 150                     ovsdbBridgeIid.child(ProtocolEntry.class, protocolEntry.key());
 151             changes.put(protocolIid, protocolEntry);
 152         }
 153         for (ControllerEntry controller : ovsdbBridge.getControllerEntry()) {
 154             KeyedInstanceIdentifier<ControllerEntry, ControllerEntryKey> controllerIid =
 155                     ovsdbBridgeIid.child(ControllerEntry.class, controller.key());
 156             changes.put(controllerIid, controller);
 157         }
 158         return changes;
 159     }
 160 }