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

   1 /*
   2  * Copyright (c) 2015 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.ovsdb.openstack.netvirt.impl;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Matchers.anyString;
  13 import static org.mockito.Matchers.same;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.when;
  16
  17 import java.util.HashSet;
  18 import java.util.Random;
  19 import java.util.Set;
  20 import java.util.concurrent.ConcurrentHashMap;
  21
  22 import org.junit.Before;
  23 import org.junit.Test;
  24 import org.junit.runner.RunWith;
  25 import org.mockito.InjectMocks;
  26 import org.mockito.Mock;
  27 import org.mockito.runners.MockitoJUnitRunner;
  28 import org.opendaylight.neutron.spi.NeutronNetwork;
  29 import org.opendaylight.ovsdb.lib.notation.Column;
  30 import org.opendaylight.ovsdb.lib.notation.Row;
  31 import org.opendaylight.ovsdb.lib.notation.UUID;
  32 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
  33 import org.opendaylight.ovsdb.plugin.api.OvsdbConfigurationService;
  34 import org.opendaylight.ovsdb.schema.openvswitch.Interface;
  35 import org.opendaylight.ovsdb.schema.openvswitch.Port;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  37
  38 /**
  39  * Unit test for class VlanConfigurationCacheImpl
  40  */
  41 @RunWith(MockitoJUnitRunner.class)
  42 public class VlanConfigurationCacheImplTest {
  43
  44     @Mock private OvsdbConfigurationService ovsdbConfigurationService;
  45
  46     @InjectMocks public VlanConfigurationCacheImpl vlanConfigurationCacheImpl;
  47     @InjectMocks private TenantNetworkManagerImpl tenantNetworkManagerImpl = mock(TenantNetworkManagerImpl.class);
  48
  49     private static final String NODE_UUID = "nodeUUID";
  50     private static final String NETWORK_ID= "networkId";
  51
  52     /**
  53      * Function configuring the node
  54      */
  55     @Before
  56     public void setUp(){
  57         Row row = mock(Row.class);
  58         Port port = mock(Port.class);
  59
  60         ConcurrentHashMap<String, Row> ovsTable;
  61         ovsTable = new ConcurrentHashMap<>();
  62         ovsTable.put(NODE_UUID, row);
  63
  64         ConcurrentHashMap<String, Row> portRows = ovsTable;
  65
  66         Set<Long> tags = new HashSet<Long>();
  67         tags.add(new Random().nextLong());
  68
  69         UUID uuid = mock(UUID.class);
  70         Set<UUID> uuidSet = new HashSet<>();
  71         uuidSet.add(uuid);
  72
  73         Column<GenericTableSchema, Set<Long>> longColumnMock = mock(Column.class);
  74         Column<GenericTableSchema, Set<UUID>> uuidColumnMock = mock(Column.class);
  75
  76         Interface iface = mock(Interface.class);
  77         NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
  78
  79         // configure ovsdbConfigurationService
  80         when(ovsdbConfigurationService.getRows(any(Node.class), anyString())).thenReturn(ovsTable);
  81         when(ovsdbConfigurationService.getRows(any(Node.class), anyString())).thenReturn(portRows);
  82         when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Port.class), any(Row.class))).thenReturn(port);
  83
  84         // configure port
  85         when(port.getTagColumn()).thenReturn(longColumnMock);
  86         when(longColumnMock.getData()).thenReturn(tags);
  87         when(port.getInterfacesColumn()).thenReturn(uuidColumnMock);
  88         when(uuidColumnMock.getData()).thenReturn(uuidSet);
  89
  90         // configure ovsdbConfigurationService
  91         when(ovsdbConfigurationService.getRow(any(Node.class), anyString(), anyString())).thenReturn(row);
  92         when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Interface.class), any(Row.class))).thenReturn(iface);
  93
  94         // configure neutronNetwork
  95         when(tenantNetworkManagerImpl.getTenantNetwork(any(Interface.class))).thenReturn(neutronNetwork);
  96         when(neutronNetwork.getNetworkUUID()).thenReturn("networkUUID");
  97     }
  98
  99     /**
 100      * Test method {@link VlanConfigurationCacheImpl#assignInternalVlan(Node, String)}
 101      */
 102     @Test
 103     public void testAssignInternalVlan() {
 104         // test
 105         assertEquals("Error, did not return the correct internalVlanId (first added)", 1, (int) vlanConfigurationCacheImpl.assignInternalVlan(mock(Node.class), NETWORK_ID));
 106         assertEquals("Error, did not return the correct internalVlanId (second added)",2, (int) vlanConfigurationCacheImpl.assignInternalVlan(mock(Node.class), NETWORK_ID + "1"));
 107         }
 108
 109     /**
 110      * Test method {@linkp VlanConfigurationCacheImpl#reclaimInternalVlan(Node, String)}
 111      */
 112     @Test
 113     public void testReclaimInternalVlan(){
 114         // test
 115         assertEquals("Error, did not return the correct internalVlanId",0, (int) vlanConfigurationCacheImpl.reclaimInternalVlan(any(Node.class), NETWORK_ID));
 116     }
 117 }