openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/NetworkHandlerTest.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
   9 package org.opendaylight.ovsdb.openstack.netvirt;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.mockito.Matchers.any;
  13 import static org.mockito.Matchers.anyString;
  14 import static org.mockito.Matchers.same;
  15 import static org.mockito.Mockito.mock;
  16 import static org.mockito.Mockito.times;
  17 import static org.mockito.Mockito.verify;
  18 import static org.mockito.Mockito.when;
  19
  20 import java.net.HttpURLConnection;
  21 import java.util.ArrayList;
  22 import java.util.HashSet;
  23 import java.util.List;
  24 import java.util.Set;
  25 import java.util.concurrent.ConcurrentHashMap;
  26 import java.util.concurrent.ConcurrentMap;
  27
  28 import org.junit.Test;
  29 import org.junit.runner.RunWith;
  30 import org.mockito.InjectMocks;
  31 import org.mockito.Mock;
  32 import org.mockito.Mockito;
  33 import org.mockito.runners.MockitoJUnitRunner;
  34 import org.opendaylight.neutron.spi.INeutronNetworkCRUD;
  35 import org.opendaylight.neutron.spi.NeutronNetwork;
  36 import org.opendaylight.ovsdb.lib.notation.Column;
  37 import org.opendaylight.ovsdb.lib.notation.Row;
  38 import org.opendaylight.ovsdb.lib.notation.UUID;
  39 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
  40 import org.opendaylight.ovsdb.openstack.netvirt.api.Action;
  41 import org.opendaylight.ovsdb.openstack.netvirt.api.BridgeConfigurationManager;
  42 import org.opendaylight.ovsdb.openstack.netvirt.api.TenantNetworkManager;
  43 import org.opendaylight.ovsdb.openstack.netvirt.impl.NeutronL3Adapter;
  44 import org.opendaylight.ovsdb.plugin.api.OvsdbConfigurationService;
  45 import org.opendaylight.ovsdb.plugin.api.OvsdbConnectionService;
  46 import org.opendaylight.ovsdb.schema.openvswitch.Interface;
  47 import org.opendaylight.ovsdb.schema.openvswitch.Port;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  49
  50 /**
  51  * Unit test for {@link NetworkHandler}
  52  */
  53 @RunWith(MockitoJUnitRunner.class)
  54 public class NetworkHandlerTest {
  55
  56     @InjectMocks NetworkHandler networkHandler;
  57
  58     @Mock NeutronNetwork neutronNetwork;
  59
  60     @Mock private NeutronL3Adapter neutronL3Adapter;
  61     @Mock private TenantNetworkManager tenantNetworkManager;
  62     @Mock private INeutronNetworkCRUD neutronNetworkCache;
  63     @Mock private OvsdbConnectionService connectionService;
  64     @Mock private BridgeConfigurationManager bridgeConfigurationManager;
  65     @Mock private OvsdbConfigurationService ovsdbConfigurationService;
  66
  67     /**
  68      * Test method {@link NetworkHandler#canCreateNetwork(NeutronNetwork)}
  69      */
  70     @Test
  71     public void testCanCreateNetwork() {
  72         when(neutronNetwork.isShared())
  73                                     .thenReturn(true)
  74                                     .thenReturn(false);
  75         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_NOT_ACCEPTABLE, networkHandler.canCreateNetwork(neutronNetwork));
  76         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, networkHandler.canCreateNetwork(neutronNetwork));
  77     }
  78
  79     /**
  80      * Test method {@link NetworkHandler#canUpdateNetwork(NeutronNetwork, NeutronNetwork)}
  81      */
  82     @Test
  83     public void testCanUpdateNetwork() {
  84         when(neutronNetwork.isShared())
  85                                     .thenReturn(true)
  86                                     .thenReturn(false);
  87         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_NOT_ACCEPTABLE, networkHandler.canUpdateNetwork(neutronNetwork, neutronNetwork));
  88         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, networkHandler.canUpdateNetwork(neutronNetwork, neutronNetwork));
  89     }
  90
  91     /**
  92      * Test method {@link NetworkHandler#canDeleteNetwork(NeutronNetwork)}
  93      */
  94     @Test
  95     public void testCanDeleteNetwork() {
  96         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, networkHandler.canDeleteNetwork(neutronNetwork));
  97     }
  98
  99     /**
 100      * Test method {@link NetworkHandler#processEvent(AbstractEvent)}
 101      */
 102     @Test
 103     public void testProcessEvent() {
 104         NetworkHandler networkHandlerSpy = Mockito.spy(networkHandler);
 105
 106         NorthboundEvent ev = mock(NorthboundEvent.class);
 107         when(ev.getNeutronNetwork()).thenReturn(neutronNetwork);
 108
 109         when(ev.getAction()).thenReturn(Action.ADD);
 110         networkHandlerSpy.processEvent(ev);
 111         verify(neutronL3Adapter, times(1)).handleNeutronNetworkEvent(any(NeutronNetwork.class), same(Action.ADD));
 112
 113         when(ev.getAction()).thenReturn(Action.UPDATE);
 114         networkHandlerSpy.processEvent(ev);
 115         verify(neutronL3Adapter, times(1)).handleNeutronNetworkEvent(any(NeutronNetwork.class), same(Action.UPDATE));
 116
 117         /* configuration needed to pass doNeutronNetworkDeleted() function*/
 118         Node node = mock(Node.class);
 119         List<Node> nodes = new ArrayList();
 120         nodes.add(node);
 121         when(connectionService.getNodes()).thenReturn(nodes);
 122
 123         ConcurrentMap<String, Row> ports = new ConcurrentHashMap<>();
 124         Row row = mock(Row.class);
 125         ports.put("key", row);
 126         when(ovsdbConfigurationService.getRows(any(Node.class), anyString())).thenReturn(ports);
 127
 128         Port port = mock(Port.class);
 129         Column<GenericTableSchema, Set<UUID>> portColumns = mock(Column.class);
 130         Set<UUID> interfaceUUIDs = new HashSet<UUID>();
 131         UUID uuid = mock(UUID.class);
 132         interfaceUUIDs.add(uuid);
 133         when(port.getInterfacesColumn()).thenReturn(portColumns);
 134         when(portColumns.getData()).thenReturn(interfaceUUIDs);
 135         when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Port.class), any(Row.class))).thenReturn(port);
 136
 137         Interface iface = mock(Interface.class);
 138         Column<GenericTableSchema, String> ifaceColumns = mock(Column.class);
 139         when(iface.getTypeColumn()).thenReturn(ifaceColumns);
 140         when(ifaceColumns.getData()).thenReturn(NetworkHandler.NETWORK_TYPE_VXLAN);
 141         when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Interface.class), any(Row.class))).thenReturn(iface);
 142         /**/
 143
 144         when(ev.getAction()).thenReturn(Action.DELETE);
 145         networkHandlerSpy.processEvent(ev);
 146         verify(neutronL3Adapter, times(1)).handleNeutronNetworkEvent(any(NeutronNetwork.class), same(Action.DELETE));
 147         verify(tenantNetworkManager, times(1)).networkDeleted(anyString());
 148     }
 149 }