Use the memorized service reference
[netvirt.git] / openstack / net-virt / src / test / java / org / opendaylight / ovsdb / openstack / netvirt / impl / VlanConfigurationCacheImplTest.java
index 58d4d1f4ad5130e2b88f59dabb65276afb9b9f6b..d30fdacb15f4ad0a9e7c652043c89a713423b99c 100644 (file)
@@ -9,107 +9,95 @@ package org.opendaylight.ovsdb.openstack.netvirt.impl;
 
 import static org.junit.Assert.assertEquals;
 import static org.mockito.Matchers.any;
-import static org.mockito.Matchers.anyString;
-import static org.mockito.Matchers.same;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
-import java.util.HashSet;
-import java.util.Set;
-import java.util.concurrent.ConcurrentHashMap;
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.List;
 
 import org.junit.Before;
 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.NeutronNetwork;
-import org.opendaylight.ovsdb.lib.notation.Column;
-import org.opendaylight.ovsdb.lib.notation.Row;
-import org.opendaylight.ovsdb.lib.notation.UUID;
-import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
-import org.opendaylight.ovsdb.plugin.api.OvsdbConfigurationService;
-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;
+import org.opendaylight.ovsdb.openstack.netvirt.api.Southbound;
+import org.opendaylight.ovsdb.openstack.netvirt.api.TenantNetworkManager;
+import org.opendaylight.ovsdb.utils.servicehelper.ServiceHelper;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.VlanId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbTerminationPointAugmentation;
+import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
+
+import org.osgi.framework.ServiceReference;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
 
 /**
  * Unit test for {@link VlanConfigurationCacheImpl}
  */
-@RunWith(MockitoJUnitRunner.class)
+@RunWith(PowerMockRunner.class)
+@PrepareForTest(ServiceHelper.class)
 public class VlanConfigurationCacheImplTest {
 
-    @Mock private OvsdbConfigurationService ovsdbConfigurationService;
-
     @InjectMocks public VlanConfigurationCacheImpl vlanConfigurationCacheImpl;
-    @InjectMocks private TenantNetworkManagerImpl tenantNetworkManagerImpl = mock(TenantNetworkManagerImpl.class);
+
+    @Mock private TenantNetworkManager tenantNetworkManager;
+    @Mock private Southbound southbound;
 
     private static final String NODE_UUID = "nodeUUID";
     private static final String NETWORK_ID= "networkId";
-
+    private static final int VLAN_ID = 4;
     /**
      * Function configuring the node
      */
     @Before
     public void setUp(){
-        Row row = mock(Row.class);
-        Port port = mock(Port.class);
-
-        ConcurrentHashMap<String, Row> ovsTable;
-        ovsTable = new ConcurrentHashMap<>();
-        ovsTable.put(NODE_UUID, row);
-
-        Set<Long> tags = new HashSet<Long>();
-        tags.add(Long.valueOf(1));
-
-        UUID uuid = mock(UUID.class);
-        Set<UUID> uuidSet = new HashSet<>();
-        uuidSet.add(uuid);
-
-        Column<GenericTableSchema, Set<Long>> longColumnMock = mock(Column.class);
-        Column<GenericTableSchema, Set<UUID>> uuidColumnMock = mock(Column.class);
-
-        Interface iface = mock(Interface.class);
+        when(southbound.getOvsdbNodeUUID(any(Node.class))).thenReturn(NODE_UUID);
+        List<OvsdbTerminationPointAugmentation> ports = new ArrayList<OvsdbTerminationPointAugmentation>();
+        OvsdbTerminationPointAugmentation port = mock(OvsdbTerminationPointAugmentation.class);
+        VlanId vlanId = mock(VlanId.class);
+        when(vlanId.getValue()).thenReturn(VLAN_ID);
+        when(port.getVlanTag()).thenReturn(vlanId);
+        when(southbound.getTerminationPointsOfBridge(any(Node.class))).thenReturn(ports );
         NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
-
-        when(ovsdbConfigurationService.getRows(any(Node.class), anyString())).thenReturn(ovsTable);
-        when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Port.class), any(Row.class))).thenReturn(port);
-
-        when(port.getTagColumn()).thenReturn(longColumnMock);
-        when(longColumnMock.getData()).thenReturn(tags);
-        when(port.getInterfacesColumn()).thenReturn(uuidColumnMock);
-        when(uuidColumnMock.getData()).thenReturn(uuidSet);
-
-        when(ovsdbConfigurationService.getRow(any(Node.class), anyString(), anyString())).thenReturn(row);
-        when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Interface.class), any(Row.class))).thenReturn(iface);
-
-        when(tenantNetworkManagerImpl.getTenantNetwork(any(Interface.class))).thenReturn(neutronNetwork);
         when(neutronNetwork.getNetworkUUID()).thenReturn(NETWORK_ID);
+        when(tenantNetworkManager.getTenantNetwork(any(OvsdbTerminationPointAugmentation.class))).thenReturn(neutronNetwork );
     }
 
     /**
      * Test method {@link VlanConfigurationCacheImpl#assignInternalVlan(Node, String)}
      */
     @Test
-    public void testAssignInternalVlan() {
-        assertEquals("Error, did not return the correct internalVlanId (first added)", 1, (int) vlanConfigurationCacheImpl.assignInternalVlan(any(Node.class), NETWORK_ID));
-        assertEquals("Error, did not return the correct internalVlanId (second added)", 2, (int) vlanConfigurationCacheImpl.assignInternalVlan(any(Node.class), NETWORK_ID + "1"));
+    public void testAssignReclaimAndGetInternalVlan() {
+        assertEquals("Error, assignInternalVlan() did not return the correct internalVlanId (first added)", 1, (int) vlanConfigurationCacheImpl.assignInternalVlan(any(Node.class), NETWORK_ID));
+        assertEquals("Error, assignInternalVlan () did not return the correct internalVlanId (second added)", 2, (int) vlanConfigurationCacheImpl.assignInternalVlan(any(Node.class), NETWORK_ID + "1"));
+
+        assertEquals("Error, getInternalVlan() did not return the correct internalVlan", 1, (int) vlanConfigurationCacheImpl.getInternalVlan(any(Node.class), NETWORK_ID));
+
+        assertEquals("Error, reclaimInternalVlan() did not return the correct internalVlanId", 1, (int) vlanConfigurationCacheImpl.reclaimInternalVlan(any(Node.class), NETWORK_ID));
+        assertEquals("Error, reclaimInternalVlan() did not return the correct internalVlanId", 2, (int) vlanConfigurationCacheImpl.reclaimInternalVlan(any(Node.class), NETWORK_ID + "1"));
     }
 
-    /**
-     * Test method {@link VlanConfigurationCacheImpl#reclaimInternalVlan(Node, String)}
-     */
     @Test
-    public void testReclaimInternalVlan(){
-        assertEquals("Error, did not return the correct internalVlanId", 1, (int) vlanConfigurationCacheImpl.reclaimInternalVlan(any(Node.class), NETWORK_ID));
+    public void testSetDependencies() throws Exception {
+        TenantNetworkManager tenantNetworkManager = mock(TenantNetworkManager.class);
+        Southbound southbound = mock(Southbound.class);
+
+        PowerMockito.mockStatic(ServiceHelper.class);
+        PowerMockito.when(ServiceHelper.getGlobalInstance(TenantNetworkManager.class, vlanConfigurationCacheImpl)).thenReturn(tenantNetworkManager);
+        PowerMockito.when(ServiceHelper.getGlobalInstance(Southbound.class, vlanConfigurationCacheImpl)).thenReturn(southbound);
+
+        vlanConfigurationCacheImpl.setDependencies(mock(ServiceReference.class));
+
+        assertEquals("Error, did not return the correct object", getField("tenantNetworkManager"), tenantNetworkManager);
+        assertEquals("Error, did not return the correct object", getField("southbound"), southbound);
     }
 
-    /**
-     * Test method {@link VlanConfigurationCacheImpl#getInternalVlan(Node, String)}
-     */
-    @Test
-    public void testGetInternalVlan(){
-        assertEquals("Error, did not return the correct internalVlan", 1, (int) vlanConfigurationCacheImpl.getInternalVlan(any(Node.class), NETWORK_ID));
+    private Object getField(String fieldName) throws Exception {
+        Field field = VlanConfigurationCacheImpl.class.getDeclaredField(fieldName);
+        field.setAccessible(true);
+        return field.get(vlanConfigurationCacheImpl);
     }
 }