openstack/net-virt/src/test/java/org/opendaylight/netvirt/openstack/netvirt/impl/VlanConfigurationCacheImplTest.java

   1 /*
   2  * Copyright (c) 2015, 2016 Inocybe 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.netvirt.openstack.netvirt.impl;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Mockito.mock;
  13 import static org.mockito.Mockito.when;
  14
  15 import java.lang.reflect.Field;
  16 import java.util.ArrayList;
  17 import java.util.List;
  18
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.junit.runner.RunWith;
  22 import org.mockito.InjectMocks;
  23 import org.mockito.Mock;
  24 import org.mockito.runners.MockitoJUnitRunner;
  25 import org.opendaylight.netvirt.openstack.netvirt.api.Southbound;
  26 import org.opendaylight.netvirt.openstack.netvirt.api.TenantNetworkManager;
  27 import org.opendaylight.netvirt.openstack.netvirt.translator.NeutronNetwork;
  28 import org.opendaylight.netvirt.utils.servicehelper.ServiceHelper;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.VlanId;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105
  31         .OvsdbTerminationPointAugmentation;
  32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology
  33         .Node;
  34 import org.osgi.framework.ServiceReference;
  35
  36 /**
  37  * Unit test for {@link VlanConfigurationCacheImpl}
  38  */
  39 @RunWith(MockitoJUnitRunner.class)
  40 public class VlanConfigurationCacheImplTest {
  41
  42     @InjectMocks public VlanConfigurationCacheImpl vlanConfigurationCacheImpl;
  43
  44     @Mock private TenantNetworkManager tenantNetworkManager;
  45     @Mock private Southbound southbound;
  46
  47     private static final String NODE_UUID = "nodeUUID";
  48     private static final String NETWORK_ID= "networkId";
  49     private static final int VLAN_ID = 4;
  50     /**
  51      * Function configuring the node
  52      */
  53     @Before
  54     public void setUp(){
  55         when(southbound.getOvsdbNodeUUID(any(Node.class))).thenReturn(NODE_UUID);
  56         List<OvsdbTerminationPointAugmentation> ports = new ArrayList<>();
  57         OvsdbTerminationPointAugmentation port = mock(OvsdbTerminationPointAugmentation.class);
  58         VlanId vlanId = mock(VlanId.class);
  59         when(vlanId.getValue()).thenReturn(VLAN_ID);
  60         when(port.getVlanTag()).thenReturn(vlanId);
  61         when(southbound.getTerminationPointsOfBridge(any(Node.class))).thenReturn(ports );
  62         NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
  63         when(neutronNetwork.getNetworkUUID()).thenReturn(NETWORK_ID);
  64         when(tenantNetworkManager.getTenantNetwork(any(OvsdbTerminationPointAugmentation.class))).thenReturn(neutronNetwork );
  65     }
  66
  67     /**
  68      * Test method {@link VlanConfigurationCacheImpl#assignInternalVlan(Node, String)}
  69      */
  70     @Test
  71     public void testAssignReclaimAndGetInternalVlan() {
  72         assertEquals("Error, assignInternalVlan() did not return the correct internalVlanId (first added)", 1, (int) vlanConfigurationCacheImpl.assignInternalVlan(any(Node.class), NETWORK_ID));
  73         assertEquals("Error, assignInternalVlan () did not return the correct internalVlanId (second added)", 2, (int) vlanConfigurationCacheImpl.assignInternalVlan(any(Node.class), NETWORK_ID + "1"));
  74
  75         assertEquals("Error, getInternalVlan() did not return the correct internalVlan", 1, (int) vlanConfigurationCacheImpl.getInternalVlan(any(Node.class), NETWORK_ID));
  76
  77         assertEquals("Error, reclaimInternalVlan() did not return the correct internalVlanId", 1, (int) vlanConfigurationCacheImpl.reclaimInternalVlan(any(Node.class), NETWORK_ID));
  78         assertEquals("Error, reclaimInternalVlan() did not return the correct internalVlanId", 2, (int) vlanConfigurationCacheImpl.reclaimInternalVlan(any(Node.class), NETWORK_ID + "1"));
  79     }
  80
  81     @Test
  82     public void testSetDependencies() throws Exception {
  83         TenantNetworkManager tenantNetworkManager = mock(TenantNetworkManager.class);
  84         Southbound southbound = mock(Southbound.class);
  85
  86         ServiceHelper.overrideGlobalInstance(TenantNetworkManager.class, tenantNetworkManager);
  87         ServiceHelper.overrideGlobalInstance(Southbound.class, southbound);
  88
  89         vlanConfigurationCacheImpl.setDependencies(mock(ServiceReference.class));
  90
  91         assertEquals("Error, did not return the correct object", getField("tenantNetworkManager"), tenantNetworkManager);
  92         assertEquals("Error, did not return the correct object", getField("southbound"), southbound);
  93     }
  94
  95     private Object getField(String fieldName) throws Exception {
  96         Field field = VlanConfigurationCacheImpl.class.getDeclaredField(fieldName);
  97         field.setAccessible(true);
  98         return field.get(vlanConfigurationCacheImpl);
  99     }
 100 }