Add JUnit testing of TenantNetworkManagerImpl class

Change-Id: I305ddb0dbdea8b214ed7dcd041ac9f2d585bb9db
Signed-off-by: Alexis de Talhouët <adetalhouet@inocybe.com>

diff --git a/openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/impl/TenantNetworkManagerImplTest.java b/openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/impl/TenantNetworkManagerImplTest.java
new file mode 100644 (file)
index 0000000..12d6cb1
--- /dev/null
+++ b/openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/impl/TenantNetworkManagerImplTest.java
@@ -0,0 +1,226 @@
+/*
+ * Copyright (c) 2015 Inocybe and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.ovsdb.openstack.netvirt.impl;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyString;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Matchers.same;
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
+import org.opendaylight.neutron.spi.INeutronNetworkCRUD;
+import org.opendaylight.neutron.spi.INeutronPortCRUD;
+import org.opendaylight.neutron.spi.NeutronNetwork;
+import org.opendaylight.neutron.spi.NeutronPort;
+import org.opendaylight.ovsdb.lib.notation.Column;
+import org.opendaylight.ovsdb.lib.notation.Row;
+import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
+import org.opendaylight.ovsdb.openstack.netvirt.api.Constants;
+import org.opendaylight.ovsdb.openstack.netvirt.api.NetworkingProvider;
+import org.opendaylight.ovsdb.openstack.netvirt.api.NetworkingProviderManager;
+import org.opendaylight.ovsdb.openstack.netvirt.api.VlanConfigurationCache;
+import org.opendaylight.ovsdb.plugin.api.OvsdbConfigurationService;
+import org.opendaylight.ovsdb.plugin.api.Status;
+import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
+import org.opendaylight.ovsdb.schema.openvswitch.Interface;
+import org.opendaylight.ovsdb.schema.openvswitch.Port;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
+
+/**
+ * Unit test for class TenantNetworkManagerImpl
+ */
+@RunWith(MockitoJUnitRunner.class)
+public class TenantNetworkManagerImplTest {
+
+    @Mock private VlanConfigurationCache vlanConfigurationCache;
+    @Mock private INeutronPortCRUD neutronCache;
+    @Mock private OvsdbConfigurationService ovsdbConfigurationService;
+    @Mock private NetworkingProviderManager networkingProviderManager;
+
+    @InjectMocks private TenantNetworkManagerImpl tenantNetworkManagerImpl;
+    @InjectMocks private INeutronPortCRUD neutronPortCache = mock(INeutronPortCRUD.class);
+    @InjectMocks private INeutronNetworkCRUD neutronNetworkCache = mock(INeutronNetworkCRUD.class);
+
+    private static final String NETWORK_ID= "networkId";
+
+    /**
+     * Test method {@link TenantNetworkManagerImpl#getInternalVlan(Node, String)}
+     */
+    @Test
+    public void testGetInternalVlan() {
+        // configure vlanConfigurationCache
+        when(vlanConfigurationCache.getInternalVlan(any(Node.class), eq(NETWORK_ID))).thenReturn(10);
+
+        // test on an existing internal vlan
+        assertEquals("Error, did not return the correct internalVlan" , 10, tenantNetworkManagerImpl.getInternalVlan(mock(Node.class), NETWORK_ID));
+        // test on a missing internal vlan
+        assertEquals("Error, did not return the correct internalVlan", 0, tenantNetworkManagerImpl.getInternalVlan(mock(Node.class), "unexistingNetwork"));
+    }
+
+    /**
+     * Test method {@link TenantNetworkManagerImpl#reclaimInternalVlan(Node, String, NeutronNetwork)}
+     */
+    @Test
+    public void testReclaimInternalVlan() {
+        // configure vlanConfigurationCache
+        when(vlanConfigurationCache.reclaimInternalVlan(any(Node.class), eq(NETWORK_ID))).thenReturn(10);
+
+        // test on an existing internal vlan
+        tenantNetworkManagerImpl.reclaimInternalVlan(mock(Node.class), NETWORK_ID, mock(NeutronNetwork.class));
+        // test on a missing internal vlan
+        tenantNetworkManagerImpl.reclaimInternalVlan(mock(Node.class), "unexistingNetwork", mock(NeutronNetwork.class));
+    }
+
+    /**
+     * Test method {@link TenantNetworkManagerImpl#programInternalVlan(Node, String, NeutronNetwork)}
+     */
+    @Test
+    public void testProgramInternalVlan(){
+        Port port = mock(Port.class);
+        Row row = mock(Row.class);
+        GenericTableSchema tableSchema = mock(GenericTableSchema.class);
+        Status status = mock(Status.class);
+
+        // configure port
+        when(port.getRow()).thenReturn(row);
+        when(port.getSchema()).thenReturn(tableSchema);
+
+        // configure vlanConfigurationCache
+        when(vlanConfigurationCache.getInternalVlan(any(Node.class), anyString())).thenReturn(10);
+        // configure ovsdbConfigurationService
+        when(ovsdbConfigurationService.createTypedRow(any(Node.class), same(Port.class))).thenReturn(port);
+        when(ovsdbConfigurationService.updateRow(any(Node.class), anyString(), anyString(), anyString(), any(Row.class))).thenReturn(status);
+
+        // test
+        tenantNetworkManagerImpl.programInternalVlan(mock(Node.class), NETWORK_ID, mock(NeutronNetwork.class));
+    }
+
+    /**
+     * Test method {@link TenantNetworkManagerImpl#isTenantNetworkPresentInNode(Node, String)}
+     */
+    @Test
+    public void testIsTenantNetworkPresentInNode() {
+        NetworkingProvider networkingProvider = mock(NetworkingProvider.class);
+
+        Interface intf = mock(Interface.class);
+        Column<GenericTableSchema, Map<String, String>> columnMock = mock(Column.class);
+        Map<String, String> externalIds = new HashMap<String, String>();
+        externalIds.put(Constants.EXTERNAL_ID_INTERFACE_ID, "interfaceId");
+
+        Row row = mock(Row.class);
+        Bridge bridge = mock(Bridge.class, RETURNS_DEEP_STUBS);
+
+        ConcurrentHashMap<String, Row> map;
+        map = new ConcurrentHashMap<>();
+        map.put("row", row);
+
+        NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
+        NeutronPort neutronPort = mock(NeutronPort.class);
+        ArrayList<NeutronNetwork> listNeutronNetwork = new ArrayList<NeutronNetwork>();
+        listNeutronNetwork.add(neutronNetwork);
+
+        // configure neutronNetwork
+        when(neutronNetwork.getProviderSegmentationID()).thenReturn("segId");
+        when(neutronNetwork.getNetworkUUID()).thenReturn("networkUUID");
+        when(neutronNetworkCache.getAllNetworks()).thenReturn(listNeutronNetwork);
+
+        // test method getNetworkId
+        assertEquals("Error, did not return the UUID of the correct network", listNeutronNetwork.get(0).getNetworkUUID(), tenantNetworkManagerImpl.getNetworkId("segId"));
+
+        // configure networkingProvider
+        when(networkingProviderManager.getProvider(any(Node.class))).thenReturn(networkingProvider);
+        when(networkingProvider.hasPerTenantTunneling()).thenReturn(true);
+
+        // configure vlanConfigurationCache
+        when(vlanConfigurationCache.getInternalVlan(any(Node.class), anyString())).thenReturn(10);
+
+        // configure ovsdbConfigurationService
+        when(ovsdbConfigurationService.getRows(any(Node.class), anyString())).thenReturn(map);
+        when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Interface.class),
+                any(Row.class))).thenReturn(intf);
+
+        // configure interface
+        when(intf.getExternalIdsColumn()).thenReturn(columnMock);
+        when(columnMock.getData()).thenReturn(externalIds);
+
+        // configure neutronPort
+        when(neutronPortCache.getPort(anyString())).thenReturn(neutronPort);
+        when(neutronPort.getNetworkUUID()).thenReturn("networkUUID");
+
+        // test method isTenantNetworkPresentInNode
+        assertTrue("Error, did not return correct boolean for isTenantNetworkPresentInNode", tenantNetworkManagerImpl.isTenantNetworkPresentInNode(mock(Node.class), "segId"));
+    }
+
+    /**
+     * Test method {@link TenantNetworkManagerImpl#getNetworkId(String)}
+     */
+    @Test
+    public void testGetNetworkId() {
+        NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
+        ArrayList<NeutronNetwork> listNeutronNetwork = new ArrayList<NeutronNetwork>();
+        listNeutronNetwork.add(neutronNetwork);
+
+        // configure neutronNetwork
+        when(neutronNetwork.getProviderSegmentationID()).thenReturn("segId");
+        when(neutronNetwork.getNetworkUUID()).thenReturn("networkUUID");
+        when(neutronNetworkCache.getAllNetworks()).thenReturn(listNeutronNetwork);
+
+        assertEquals("Error, did not return the UUID of the correct network", listNeutronNetwork.get(0).getNetworkUUID(), tenantNetworkManagerImpl.getNetworkId("segId"));
+    }
+
+    /**
+     * Test method {@link TenantNetworkManagerImpl#getTenantNetwork(Interface)}
+     */
+    @Test
+    public void testGetTenantNetwork() {
+        Interface intf = mock(Interface.class);
+        Column<GenericTableSchema, Map<String, String>> columnMock = mock(Column.class);
+        Map<String, String> externalIds = new HashMap<String, String>();
+        externalIds.put(Constants.EXTERNAL_ID_INTERFACE_ID, "tenantValue");
+        NeutronPort neutronPort = mock(NeutronPort.class);
+        NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
+
+        // configure interface
+        when(intf.getExternalIdsColumn()).thenReturn(columnMock);
+        when(columnMock.getData()).thenReturn(externalIds);
+
+        // configure neutronPort
+        when(neutronPort.getNetworkUUID()).thenReturn("neutronUUID");
+        when(neutronPortCache.getPort(anyString())).thenReturn(neutronPort);
+        // configure neutronNetwork
+        when(neutronNetworkCache.getNetwork(anyString())).thenReturn(neutronNetwork);
+
+        assertEquals("Error, did not return the correct tenant", neutronNetwork, tenantNetworkManagerImpl.getTenantNetwork(intf));
+    }
+
+    /**
+     * Test method {@link TenantNetworkManagerImpl#networkCreated(String)}
+     */
+    @Test
+    public void testNetworkCreated() {
+        // configure vlanConfigurationCache
+        when(vlanConfigurationCache.assignInternalVlan(any(Node.class), anyString())).thenReturn(10);
+
+        assertEquals("Error, did not return the correct network id", 10,tenantNetworkManagerImpl.networkCreated(mock(Node.class), NETWORK_ID));
+    }
+}