openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/PortHandlerTest.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.HashMap;
  23 import java.util.HashSet;
  24 import java.util.List;
  25 import java.util.Map;
  26 import java.util.Set;
  27 import java.util.concurrent.ConcurrentHashMap;
  28 import java.util.concurrent.ConcurrentMap;
  29
  30 import org.junit.Test;
  31 import org.junit.runner.RunWith;
  32 import org.mockito.InjectMocks;
  33 import org.mockito.Mock;
  34 import org.mockito.Mockito;
  35 import org.mockito.runners.MockitoJUnitRunner;
  36 import org.opendaylight.neutron.spi.NeutronPort;
  37 import org.opendaylight.ovsdb.lib.notation.Column;
  38 import org.opendaylight.ovsdb.lib.notation.Row;
  39 import org.opendaylight.ovsdb.lib.notation.UUID;
  40 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
  41 import org.opendaylight.ovsdb.openstack.netvirt.api.Action;
  42 import org.opendaylight.ovsdb.openstack.netvirt.api.Constants;
  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 fort {@link PortHandler}
  52  */
  53 @RunWith(MockitoJUnitRunner.class)
  54 public class PortHandlerTest {
  55
  56     @InjectMocks PortHandler portHandler;
  57
  58     @Mock private OvsdbConfigurationService ovsdbConfigurationService;
  59     @Mock private OvsdbConnectionService connectionService;
  60     @Mock private NeutronL3Adapter neutronL3Adapter;
  61
  62     @Test
  63     public void testCanCreatePort() {
  64         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, portHandler.canCreatePort(mock(NeutronPort.class)));
  65     }
  66
  67     @Test
  68     public void testCanUpdatePort() {
  69         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, portHandler.canUpdatePort(mock(NeutronPort.class), mock(NeutronPort.class)));
  70     }
  71
  72     @Test
  73     public void testCanDeletePort() {
  74         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, portHandler.canDeletePort(mock(NeutronPort.class)));
  75     }
  76
  77     @Test
  78     public void testProcessEvent() {
  79         PortHandler portHandlerSpy = Mockito.spy(portHandler);
  80
  81         NeutronPort neutronPort = mock(NeutronPort.class);
  82         when(neutronPort.getTenantID()).thenReturn("tenantID");
  83         when(neutronPort.getNetworkUUID()).thenReturn("networkUUID");
  84         when(neutronPort.getID()).thenReturn("ID");
  85         when(neutronPort.getPortUUID()).thenReturn("portUUID");
  86
  87         NorthboundEvent ev = mock(NorthboundEvent.class);
  88         when(ev.getPort()).thenReturn(neutronPort);
  89
  90         when(ev.getAction()).thenReturn(Action.ADD);
  91         portHandlerSpy.processEvent(ev);
  92         verify(neutronL3Adapter, times(1)).handleNeutronPortEvent(neutronPort, Action.ADD);
  93
  94         when(ev.getAction()).thenReturn(Action.UPDATE);
  95         portHandlerSpy.processEvent(ev);
  96         verify(neutronL3Adapter, times(1)).handleNeutronPortEvent(neutronPort, Action.UPDATE);
  97
  98
  99         Node node = mock(Node.class);
 100         List<Node> nodes = new ArrayList();
 101         nodes.add(node);
 102         when(connectionService.getNodes()).thenReturn(nodes);
 103
 104         Row row = mock(Row.class);
 105         ConcurrentMap<String, Row> portRows = new ConcurrentHashMap();
 106         portRows.put("key", row);
 107         when(ovsdbConfigurationService.getRows(any(Node.class), anyString())).thenReturn(portRows );
 108
 109         Port port = mock(Port.class);
 110         Column<GenericTableSchema, Set<UUID>> itfaceColumns = mock(Column.class);
 111         when(port.getInterfacesColumn()).thenReturn(itfaceColumns);
 112         Set<UUID> ifaceUUIDs = new HashSet();
 113         ifaceUUIDs.add(mock(UUID.class));
 114         when(itfaceColumns.getData()).thenReturn(ifaceUUIDs );
 115         when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Port.class), any(Row.class))).thenReturn(port);
 116
 117         Interface itface = mock(Interface.class);
 118         Column<GenericTableSchema, Map<String, String>> externalIdColumns = mock(Column.class);
 119         Map<String, String> externalIds = new HashMap();
 120         externalIds.put(Constants.EXTERNAL_ID_INTERFACE_ID, "portUUID");
 121         when(externalIdColumns.getData()).thenReturn(externalIds);
 122         when(itface.getExternalIdsColumn()).thenReturn(externalIdColumns);
 123         when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Interface.class), any(Row.class))).thenReturn(itface);
 124
 125
 126         when(ev.getAction()).thenReturn(Action.DELETE);
 127         portHandlerSpy.processEvent(ev);
 128         verify(neutronL3Adapter, times(1)).handleNeutronPortEvent(neutronPort, Action.DELETE);
 129     }
 130 }