import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyBoolean;
+import static org.mockito.Matchers.anyInt;
+import static org.mockito.Matchers.anyLong;
import static org.mockito.Matchers.anyString;
+import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
import static org.mockito.Mockito.when;
+import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
-import java.lang.reflect.Method;
+import java.net.InetAddress;
import java.util.ArrayList;
+import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
+import org.apache.commons.lang3.tuple.Pair;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
-import org.mockito.InjectMocks;
import org.mockito.Mock;
+import org.mockito.Mockito;
import org.opendaylight.neutron.spi.INeutronNetworkCRUD;
import org.opendaylight.neutron.spi.INeutronPortCRUD;
import org.opendaylight.neutron.spi.INeutronSubnetCRUD;
import org.opendaylight.ovsdb.openstack.netvirt.api.RoutingProvider;
import org.opendaylight.ovsdb.openstack.netvirt.api.Southbound;
import org.opendaylight.ovsdb.openstack.netvirt.api.Status;
+import org.opendaylight.ovsdb.openstack.netvirt.api.StatusCode;
import org.opendaylight.ovsdb.openstack.netvirt.api.TenantNetworkManager;
-import org.opendaylight.ovsdb.utils.config.ConfigProperties;
import org.opendaylight.ovsdb.utils.servicehelper.ServiceHelper;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
+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.NodeId;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceReference;
import org.powermock.api.mockito.PowerMockito;
+import org.powermock.api.support.membermodification.MemberMatcher;
+import org.powermock.api.support.membermodification.MemberModifier;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
+import org.powermock.reflect.Whitebox;
/**
* Unit test for {@link NeutronL3Adapter}
*/
-@PrepareForTest({ConfigProperties.class, ServiceHelper.class})
+@PrepareForTest({ServiceHelper.class, InetAddress.class, NeutronL3Adapter.class})
@RunWith(PowerMockRunner.class)
public class NeutronL3AdapterTest {
- @InjectMocks private NeutronL3Adapter neutronL3Adapter;
+ @Mock private NeutronL3Adapter neutronL3Adapter;
+
+ private static final String ID = "45";
+ private static final String IP = "127.0.0.1";
+ private static final String MALFORM_IP = "127.0.0.1.5";
+ private static final String INTF_NAME = "br-int";
+ private static final String EXTERNAL_ROUTER_MAC_UPDATE = "";
+ private static final String UUID = "7da709ff-397f-4778-a0e8-994811272fdb";
+ private static final String FIXED_IP_ADDRESS = "192.168.1.0";
+ private static final String FLOATING_IP_ADDRESS = "192.168.1.1";
+ private static final String OWNER_ROUTER_INTERFACE = "network:router_interface";
+ private static final String OWNER_FLOATING_IP = "network:floatingip";
+ private static final String MAC_ADDRESS = "00:00:5E:00:02:01";
+ private static final String MAC_ADDRESS_2 = "00:00:5E:00:02:02";
+ private static final String PORT_INT = "port_int";
+ private static final String SEG_ID = "2";
+ private static final String CIDR = "192.168.100.0/24";
+ private static final String OFPort = "OFPort|45";
+ private static final String IP_MASK = "127.0.0.1/32";
+
+ @SuppressWarnings("rawtypes")
+ private Class floatingIpClass;
+ private Object floatingIpObject;
- @Mock private ConfigurationService configurationService;
- @Mock private TenantNetworkManager tenantNetworkManager;
- @Mock private INeutronNetworkCRUD neutronNetworkCache;
- @Mock private INeutronSubnetCRUD neutronSubnetCache;
- @Mock private INeutronPortCRUD neutronPortCache;
- @Mock private NodeCacheManager nodeCacheManager;
- @Mock private Southbound southbound;
-
- @Mock private NeutronPort neutronPort;
+ @Before
+ public void setUp() throws Exception{
+ neutronL3Adapter = PowerMockito.mock(NeutronL3Adapter.class, Mockito.CALLS_REAL_METHODS);
- private Set<String> inboundIpRewriteCache;
- private Set<String> outboundIpRewriteCache;
- private Set<String> outboundIpRewriteExclusionCache;
- private Set<String> routerInterfacesCache;
- private Set<String> staticArpEntryCache;
- private Set<String> l3ForwardingCache;
- private Map<String, String> networkIdToRouterMacCache;
- private Map<String, NeutronRouter_Interface> subnetIdToRouterInterfaceCache;
+ // init instance variables
+ MemberModifier.field(NeutronL3Adapter.class, "enabled").set(neutronL3Adapter, true);
- private static final String HOST_ADDRESS = "127.0.0.1";
+ // floating ip (nested private class from NeutronL3Adapter)
+ floatingIpClass = Whitebox.getInnerClassType(NeutronL3Adapter.class, "FloatIpData");
+ floatingIpObject = createFloatingIpObject();
+ }
- @Before
- public void setUp() throws Exception{
- PowerMockito.mockStatic(ConfigProperties.class);
- PowerMockito.when(ConfigProperties.getProperty(neutronL3Adapter.getClass(), "ovsdb.l3.fwd.enabled")).thenReturn("yes");
- PowerMockito.when(ConfigProperties.getProperty(neutronL3Adapter.getClass(), "ovsdb.l3.arp.responder.disabled")).thenReturn("no");
+ @Test
+ public void testUpdateExternalRouterMac() throws Exception {
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "flushExistingIpRewrite"));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "rebuildExistingIpRewrite"));
- when(configurationService.isL3ForwardingEnabled()).thenReturn(true);
+ neutronL3Adapter.updateExternalRouterMac(EXTERNAL_ROUTER_MAC_UPDATE);
- this.getMethod("initL3AdapterMembers").invoke(neutronL3Adapter);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("flushExistingIpRewrite");
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("rebuildExistingIpRewrite");
+ }
- this.getNeutronL3AdapterFields();
- this.setUpVar();
+ @Test
+ public void testhandleNeutronSubnetEvent() throws Exception {
+ // Nothing to be done here
+ neutronL3Adapter.handleNeutronSubnetEvent(mock(NeutronSubnet.class), Action.ADD);
}
@Test
- public void test() {
+ public void testHandleNeutronPortEvent() throws Exception {
+ Map<String, NeutronRouter_Interface> subnetIdToRouterInterfaceCache = new HashMap<String, NeutronRouter_Interface>();
+ // Mock variables
+ Neutron_IPs neutronIP = mock(Neutron_IPs.class);
+ when(neutronIP.getSubnetUUID()).thenReturn(UUID);
+ List<Neutron_IPs> list_neutronIP = new ArrayList<Neutron_IPs>();
+ list_neutronIP.add(neutronIP);
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ when(neutronPort.getDeviceOwner()).thenReturn(OWNER_ROUTER_INTERFACE);
+ when(neutronPort.getFixedIPs()).thenReturn(list_neutronIP);
+
+ // init instance variables
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortCache").set(neutronL3Adapter , mock(INeutronPortCRUD.class));
+ subnetIdToRouterInterfaceCache.put(UUID, mock(NeutronRouter_Interface.class));
+ MemberModifier.field(NeutronL3Adapter.class, "subnetIdToRouterInterfaceCache").set(neutronL3Adapter , subnetIdToRouterInterfaceCache);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "updateL3ForNeutronPort", NeutronPort.class, boolean.class));
+ Mockito.doNothing().when(neutronL3Adapter).handleNeutronRouterInterfaceEvent(any(NeutronRouter.class), any(NeutronRouter_Interface.class), any(Action.class));
+ neutronL3Adapter.handleNeutronPortEvent(neutronPort, Action.ADD);
+ Mockito.verify(neutronL3Adapter).handleNeutronRouterInterfaceEvent(any(NeutronRouter.class), any(NeutronRouter_Interface.class), eq(Action.ADD));
+
+ when(neutronPort.getDeviceOwner()).thenReturn("");
+ neutronL3Adapter.handleNeutronPortEvent(neutronPort, Action.ADD);
+ Mockito.verify(neutronL3Adapter, times(2)).handleNeutronRouterInterfaceEvent(any(NeutronRouter.class), any(NeutronRouter_Interface.class), eq(Action.ADD));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("updateL3ForNeutronPort", any(NeutronPort.class), eq(false));
+
+ neutronL3Adapter.handleNeutronPortEvent(neutronPort, Action.DELETE);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("updateL3ForNeutronPort", any(NeutronPort.class), eq(true));
}
- private void getNeutronL3AdapterFields() throws Exception{
- inboundIpRewriteCache = (Set<String>) getField("inboundIpRewriteCache");
- outboundIpRewriteCache = (Set<String>) getField("outboundIpRewriteCache");
- outboundIpRewriteExclusionCache = (Set<String>) getField("outboundIpRewriteExclusionCache");
- routerInterfacesCache = (Set<String>) getField("routerInterfacesCache");
- staticArpEntryCache = (Set<String>) getField("staticArpEntryCache");
- l3ForwardingCache = (Set<String>) getField("l3ForwardingCache");
- networkIdToRouterMacCache = (Map<String, String>) getField("networkIdToRouterMacCache");
- subnetIdToRouterInterfaceCache = (Map<String, NeutronRouter_Interface>) getField("subnetIdToRouterInterfaceCache");
+ @Test
+ public void testhandleNeutronRouterEvent() throws Exception {
+ // Nothing to be done here
+ neutronL3Adapter.handleNeutronRouterEvent(mock(NeutronRouter.class), Action.ADD);
}
- private void setUpVar(){
+ @Test
+ public void testHandleNeutronRouterInterfaceEvent() throws Exception {
+ // Mock variables
+ NeutronRouter_Interface neutronRouterInterface = mock(NeutronRouter_Interface.class);
+ when(neutronRouterInterface.getPortUUID()).thenReturn(UUID);
+ when(neutronRouterInterface.getSubnetUUID()).thenReturn(UUID);
+
Neutron_IPs neutronIP = mock(Neutron_IPs.class);
- NeutronRouter neutronRouter = mock(NeutronRouter.class);
+ when(neutronIP.getSubnetUUID()).thenReturn(UUID);
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ List<Neutron_IPs> list_neutronIP = new ArrayList<Neutron_IPs>();
+ list_neutronIP.add(neutronIP);
+ when(neutronPort.getFixedIPs()).thenReturn(list_neutronIP);
+ List<NeutronPort> list_neutronPort = new ArrayList<>();
+ list_neutronPort.add(neutronPort);
+ INeutronPortCRUD neutronPortCache = mock(INeutronPortCRUD.class);
+ when(neutronPortCache.getAllPorts()).thenReturn(list_neutronPort);
- NeutronSubnet neutronSubnet = mock(NeutronSubnet.class);
+ // init instance variables
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortCache").set(neutronL3Adapter , neutronPortCache);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programFlowsForNeutronRouterInterface", NeutronRouter_Interface.class, Boolean.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "updateL3ForNeutronPort", NeutronPort.class, boolean.class));
+
+ neutronL3Adapter.handleNeutronRouterInterfaceEvent(mock(NeutronRouter.class), neutronRouterInterface, Action.ADD);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForNeutronRouterInterface", any(NeutronRouter_Interface.class), eq(false));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("updateL3ForNeutronPort", any(NeutronPort.class), eq(false));
+
+ neutronL3Adapter.handleNeutronRouterInterfaceEvent(mock(NeutronRouter.class), neutronRouterInterface, Action.DELETE);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForNeutronRouterInterface", any(NeutronRouter_Interface.class), eq(true));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("updateL3ForNeutronPort", any(NeutronPort.class), eq(true));
+ }
+
+ @Test
+ public void testHandleNeutronFloatingIPEvent() throws Exception {
+ // Mock variables
+ NeutronFloatingIP neutronFloatingIP = mock(NeutronFloatingIP.class);
+ when(neutronFloatingIP.getFixedIPAddress()).thenReturn(FIXED_IP_ADDRESS);
+ when(neutronFloatingIP.getFloatingIPAddress()).thenReturn(FLOATING_IP_ADDRESS);
+ when(neutronFloatingIP.getFloatingIPUUID()).thenReturn(UUID);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programFlowsForFloatingIPArpAdd", NeutronFloatingIP.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programFlowsForFloatingIPInbound", NeutronFloatingIP.class, Action.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programFlowsForFloatingIPOutbound", NeutronFloatingIP.class, Action.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programFlowsForFloatingIPArpDelete", String.class));
+
+ neutronL3Adapter.handleNeutronFloatingIPEvent(neutronFloatingIP, Action.ADD);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForFloatingIPArpAdd", any(NeutronFloatingIP.class));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForFloatingIPInbound", any(NeutronFloatingIP.class), eq(Action.ADD));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForFloatingIPOutbound", any(NeutronFloatingIP.class), eq(Action.ADD));
+
+ neutronL3Adapter.handleNeutronFloatingIPEvent(neutronFloatingIP, Action.DELETE);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForFloatingIPArpDelete", anyString());
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForFloatingIPInbound", any(NeutronFloatingIP.class), eq(Action.DELETE));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForFloatingIPOutbound", any(NeutronFloatingIP.class), eq(Action.DELETE));
+ }
+
+ @SuppressWarnings({ "unchecked", "rawtypes" })
+ @Test
+ public void testProgramFlowsForFloatingIPInbound() throws Exception {
+ Map floatIpDataMapCache = new HashMap();
+
+ NeutronFloatingIP neutronFloatingIp = mock(NeutronFloatingIP.class);
+ when(neutronFloatingIp.getID()).thenReturn(ID);
+
+ // init instance variables
+ floatIpDataMapCache .put(ID, floatingIpObject);
+ MemberModifier.field(NeutronL3Adapter.class, "floatIpDataMapCache").set(neutronL3Adapter , floatIpDataMapCache);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programInboundIpRewriteStage1", Long.class, Long.class, String.class, String.class, String.class, Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForFloatingIPInbound", neutronFloatingIp, Action.ADD);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programInboundIpRewriteStage1", anyLong(), anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+ }
+
+ @SuppressWarnings({ "unchecked", "rawtypes" })
+ @Test
+ public void testProgramFlowsForFloatingIPOutbound() throws Exception {
+ Map floatIpDataMapCache = new HashMap();
+
+ NeutronFloatingIP neutronFloatingIp = mock(NeutronFloatingIP.class);
+ when(neutronFloatingIp.getID()).thenReturn(ID);
+
+ // init instance variables
+ floatIpDataMapCache.put(ID, floatingIpObject);
+ MemberModifier.field(NeutronL3Adapter.class, "floatIpDataMapCache").set(neutronL3Adapter , floatIpDataMapCache);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programOutboundIpRewriteStage1", floatingIpClass, Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForFloatingIPOutbound", neutronFloatingIp, Action.ADD);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programOutboundIpRewriteStage1", any(floatingIpClass), eq(Action.ADD));
+ }
+
+ @SuppressWarnings({ "unchecked", "rawtypes" })
+ @Test
+ public void testFlushExistingIpRewrite() throws Exception {
+ Map floatIpDataMapCache = new HashMap();
+
+ // init instance variables
+ floatIpDataMapCache.put(ID, floatingIpObject);
+ MemberModifier.field(NeutronL3Adapter.class, "floatIpDataMapCache").set(neutronL3Adapter , floatIpDataMapCache);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programOutboundIpRewriteStage1", floatingIpClass, Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "flushExistingIpRewrite");
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programOutboundIpRewriteStage1", any(floatingIpClass), eq(Action.DELETE));
+ }
+
+ @SuppressWarnings({ "unchecked", "rawtypes" })
+ @Test
+ public void testRebuildExistingIpRewrite() throws Exception {
+ Map floatIpDataMapCache = new HashMap();
+
+ // init instance variables
+ floatIpDataMapCache.put(ID, floatingIpObject);
+ MemberModifier.field(NeutronL3Adapter.class, "floatIpDataMapCache").set(neutronL3Adapter , floatIpDataMapCache);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programOutboundIpRewriteStage1", floatingIpClass, Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "rebuildExistingIpRewrite");
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programOutboundIpRewriteStage1", any(floatingIpClass), eq(Action.ADD));
+ }
+
+ @SuppressWarnings({ "unchecked", "rawtypes" })
+ @Test
+ public void testProgramFlowsForFloatingIPArpAdd() throws Exception {
+ Map<String, Pair> neutronPortToDpIdCache = new HashMap<String, Pair>();
+ Map<String, String> networkIdToRouterMacCache = new HashMap<String, String>();
+ Map floatIpDataMapCache = new HashMap();
+
+ NeutronFloatingIP neutronFloatingIP = mock(NeutronFloatingIP.class);
+ when(neutronFloatingIP.getFixedIPAddress()).thenReturn(FIXED_IP_ADDRESS);
+ when(neutronFloatingIP.getFloatingIPAddress()).thenReturn(FLOATING_IP_ADDRESS);
+ when(neutronFloatingIP.getPortUUID()).thenReturn(UUID);
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ when(neutronPort.getMacAddress()).thenReturn(MAC_ADDRESS);
NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
- Node node = mock(Node.class);
- NodeId nodeID = mock(NodeId.class);
- // TODO SB_MIGRATION
- //Row row = mock(Row.class);
- //Bridge bridge = mock(Bridge.class);
- Status status = mock(Status.class);
+ when(neutronNetwork.getProviderSegmentationID()).thenReturn(ID);
+ when(neutronNetwork.getID()).thenReturn(ID);
+
+ // init instance variables
+ floatIpDataMapCache.put(ID, floatingIpObject);
+ neutronPortToDpIdCache.put(UUID, mock(Pair.class));
+ networkIdToRouterMacCache.put(ID, MAC_ADDRESS);
+ INeutronNetworkCRUD neutronNetworkCache = mock(INeutronNetworkCRUD.class);
+ when(neutronNetworkCache.getNetwork(anyString())).thenReturn(neutronNetwork);
+ INeutronPortCRUD neutronPortCache = mock(INeutronPortCRUD.class);
+ when(neutronPortCache.getPort(anyString())).thenReturn(neutronPort);
+ PowerMockito.doReturn(neutronPort).when(neutronL3Adapter, "findNeutronPortForFloatingIp", anyString());
+ PowerMockito.doReturn(Long.valueOf(15)).when(neutronL3Adapter, "findOFPortForExtPatch", anyLong());
+ MemberModifier.field(NeutronL3Adapter.class, "floatIpDataMapCache").set(neutronL3Adapter , floatIpDataMapCache);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortToDpIdCache").set(neutronL3Adapter , neutronPortToDpIdCache);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortCache").set(neutronL3Adapter , neutronPortCache);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronNetworkCache").set(neutronL3Adapter , neutronNetworkCache);
+ MemberModifier.field(NeutronL3Adapter.class, "networkIdToRouterMacCache").set(neutronL3Adapter , networkIdToRouterMacCache);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "findNeutronPortForFloatingIp", String.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "findOFPortForExtPatch", Long.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programStaticArpStage1", Long.class, String.class, String.class, String.class, Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForFloatingIPArpAdd", neutronFloatingIP);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("findNeutronPortForFloatingIp", anyString());
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("findOFPortForExtPatch", anyLong());
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage1", anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+ }
- List<Neutron_IPs> list_neutronIP = new ArrayList<Neutron_IPs>();
- list_neutronIP.add(neutronIP);
+ @SuppressWarnings({ "unchecked", "rawtypes" })
+ @Test
+ public void testProgramFlowsForFloatingIPArpDelete() throws Exception {
+ Map floatIpDataMapCache = new HashMap();
+
+ // init instance variables
+ floatIpDataMapCache.put(ID, floatingIpObject);
+ MemberModifier.field(NeutronL3Adapter.class, "floatIpDataMapCache").set(neutronL3Adapter , floatIpDataMapCache);
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programStaticArpStage1", Long.class, String.class, String.class, String.class, Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForFloatingIPArpDelete", ID);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage1", anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE));
+ }
+
+ @Test
+ public void testFindNeutronPortForFloatingIp() throws Exception {
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ when(neutronPort.getDeviceOwner()).thenReturn(OWNER_FLOATING_IP);
+ when(neutronPort.getDeviceID()).thenReturn(ID);
List<NeutronPort> list_neutronPort = new ArrayList<>();
list_neutronPort.add(neutronPort);
+ INeutronPortCRUD neutronPortCache = mock(INeutronPortCRUD.class);
+ when(neutronPortCache.getAllPorts()).thenReturn(list_neutronPort);
- List<Node> list_nodes = new ArrayList<Node>();
- list_nodes.add(node);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortCache").set(neutronL3Adapter , neutronPortCache);
- //ConcurrentMap<String, Row> rowMap = mock(ConcurrentMap.class);
- //rowMap.put("key", row);
+ assertEquals("Error, did not return the correct NeutronPort", neutronPort, Whitebox.invokeMethod(neutronL3Adapter, "findNeutronPortForFloatingIp", ID));
+ }
- //Column<GenericTableSchema, Set<String>> bridgeColumnIds = mock(Column.class);
- Set<String> dpids = new HashSet();
- dpids.add("11111");
+ @Test
+ public void testFindOFPortForExtPatch() throws Exception {
+ ConfigurationService configurationService = mock(ConfigurationService.class);
+ when(configurationService.getPatchPortName(any(Pair.class))).thenReturn(PORT_INT);
+ MemberModifier.field(NeutronL3Adapter.class, "configurationService").set(neutronL3Adapter , configurationService);
+ List<Node> nodes = new ArrayList<Node>();
+ nodes.add(mock(Node.class));
+ NodeCacheManager nodeCacheManager = mock(NodeCacheManager.class);
+ when(nodeCacheManager.getBridgeNodes()).thenReturn(nodes);
+ MemberModifier.field(NeutronL3Adapter.class, "nodeCacheManager").set(neutronL3Adapter , nodeCacheManager);
+ Southbound southbound = mock(Southbound.class);
+ when(southbound.getDataPathId(any(Node.class))).thenReturn(Long.valueOf(ID));
+ OvsdbTerminationPointAugmentation terminationPointOfBridge = mock(OvsdbTerminationPointAugmentation.class);
+ when(terminationPointOfBridge.getOfport()).thenReturn(Long.valueOf(ID));
+ when(southbound.getTerminationPointOfBridge(any(Node.class), anyString())).thenReturn(terminationPointOfBridge);
+ MemberModifier.field(NeutronL3Adapter.class, "southbound").set(neutronL3Adapter , southbound);
- when(neutronPort.getFixedIPs()).thenReturn(list_neutronIP);
- when(neutronPort.getPortUUID()).thenReturn("portUUID");
- when(neutronPort.getTenantID()).thenReturn("tenantID");
- when(neutronPort.getNetworkUUID()).thenReturn("networkUUID");
- when(neutronPort.getMacAddress()).thenReturn("macAddress1").thenReturn("macAddress2");
+ assertEquals("Error, did not return the correct NeutronPort", Long.valueOf(ID), Whitebox.invokeMethod(neutronL3Adapter, "findOFPortForExtPatch", Long.valueOf(ID)));
+ }
- when(neutronIP.getSubnetUUID()).thenReturn("subnetUUID");
- when(neutronIP.getIpAddress()).thenReturn(HOST_ADDRESS);
+ @Test
+ public void testHandleNeutronNetworkEvent() throws Exception {
+ // Nothing to be done here
+ Whitebox.invokeMethod(neutronL3Adapter, "handleNeutronNetworkEvent", mock(NeutronNetwork.class), Action.ADD);
+ }
- when(neutronPortCache.getAllPorts()).thenReturn(list_neutronPort);
- when(neutronPortCache.getPort(anyString())).thenReturn(neutronPort);
+ @Test
+ public void testHandleInterfaceEvent() throws Exception {
+ Map<String, Pair<Long, Uuid>> neutronPortToDpIdCache = new HashMap<String, Pair<Long,Uuid>>();
+ // init instance variables
+ TenantNetworkManager tenantNetworkManager = mock(TenantNetworkManager.class);
+ MemberModifier.field(NeutronL3Adapter.class, "tenantNetworkManager").set(neutronL3Adapter , tenantNetworkManager);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortToDpIdCache").set(neutronL3Adapter , neutronPortToDpIdCache);
+
+ // Mock variables
+ NodeId nodeId = mock(NodeId.class);
+ when(nodeId.getValue()).thenReturn(ID);
+ Node node = mock(Node.class);
+ when(node.getNodeId()).thenReturn(nodeId);
+
+ OvsdbTerminationPointAugmentation intf = mock(OvsdbTerminationPointAugmentation.class);
+ when(intf.getInterfaceUuid()).thenReturn(mock(Uuid.class));
+ when(intf.getName()).thenReturn(INTF_NAME);
+
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ when(neutronPort.getPortUUID()).thenReturn(UUID);
- when(neutronSubnetCache.getSubnet(anyString())).thenReturn(neutronSubnet);
+ when(tenantNetworkManager.getTenantPort(intf)).thenReturn(neutronPort);
- when(neutronSubnet.getNetworkUUID()).thenReturn("networkUUID");
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "getDpidForIntegrationBridge", Node.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "handleInterfaceEventAdd", String.class, Long.class, Uuid.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "handleInterfaceEventDelete", OvsdbTerminationPointAugmentation.class, Long.class));
+ PowerMockito.when(neutronL3Adapter, "getDpidForIntegrationBridge", any(Node.class)).thenReturn(Long.valueOf(45));
+ Mockito.doNothing().when(neutronL3Adapter).handleNeutronPortEvent(any(NeutronPort.class), any(Action.class));
+
+ neutronL3Adapter.handleInterfaceEvent(node, intf, mock(NeutronNetwork.class), Action.ADD);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("getDpidForIntegrationBridge", any(Node.class));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("handleInterfaceEventAdd", any(String.class), anyLong(), any(Uuid.class));
+ Mockito.verify(neutronL3Adapter).handleNeutronPortEvent(neutronPort, Action.ADD);
+
+ neutronL3Adapter.handleInterfaceEvent(node, intf, mock(NeutronNetwork.class), Action.DELETE);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("handleInterfaceEventDelete", any(OvsdbTerminationPointAugmentation.class), anyLong());
+ Mockito.verify(neutronL3Adapter).handleNeutronPortEvent(neutronPort, Action.DELETE);
+ }
+
+ @Test
+ public void testHandleInterfaceEventAdd() throws Exception {
+ Map<String, Pair<Long, Uuid>> neutronPortToDpIdCache = new HashMap<String, Pair<Long,Uuid>>();
+ // init instance variables
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortToDpIdCache").set(neutronL3Adapter , neutronPortToDpIdCache);
+ int temp = neutronPortToDpIdCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "handleInterfaceEventAdd", "", Long.valueOf(5), mock(Uuid.class));
+
+ assertEquals("Error, did not add the port", temp+1, neutronPortToDpIdCache.size());
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void testHandleInterfaceEventDelete() throws Exception {
+ Map<String, Pair<Long, Uuid>> neutronPortToDpIdCache = new HashMap<String, Pair<Long,Uuid>>();
+ OvsdbTerminationPointAugmentation intf = mock(OvsdbTerminationPointAugmentation.class);
+ Uuid uuid = mock(Uuid.class);
+ when(intf.getInterfaceUuid()).thenReturn(uuid );
+ Pair<Long, Uuid> pair = mock(Pair.class);
+ when(pair.getRight()).thenReturn(uuid);
+
+ // init instance variables
+ neutronPortToDpIdCache.put("key", pair);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortToDpIdCache").set(neutronL3Adapter , neutronPortToDpIdCache);
+ int temp = neutronPortToDpIdCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "handleInterfaceEventDelete", intf, Long.valueOf(ID));
+
+ assertEquals("Error, did not remove the port", temp-1, neutronPortToDpIdCache.size());
+ }
+
+ @Test
+ public void testUpdateL3ForNeutronPort() throws Exception {
+ Map<String, String> networkIdToRouterMacCache = new HashMap<String, String>();
+
+ Neutron_IPs neutronIp = mock(Neutron_IPs.class);
+ when(neutronIp.getIpAddress()).thenReturn(FIXED_IP_ADDRESS);
+ List<Neutron_IPs> neutronIps = new ArrayList<Neutron_IPs>();
+ neutronIps.add(neutronIp);
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ when(neutronPort.getNetworkUUID()).thenReturn(UUID);
+ when(neutronPort.getMacAddress()).thenReturn(MAC_ADDRESS_2);
+ when(neutronPort.getFixedIPs()).thenReturn(neutronIps);
+ NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
+ when(neutronNetwork.getProviderSegmentationID()).thenReturn(ID);
+ List<Node> nodes = new ArrayList<Node>();
+ nodes.add(mock(Node.class));
+ PowerMockito.doReturn(Long.valueOf(15)).when(neutronL3Adapter, "getDpidForIntegrationBridge", any(Node.class));
+
+ // init instance variables
+ networkIdToRouterMacCache .put(UUID, MAC_ADDRESS);
+ MemberModifier.field(NeutronL3Adapter.class, "networkIdToRouterMacCache").set(neutronL3Adapter , networkIdToRouterMacCache);
+ INeutronNetworkCRUD neutronNetworkCache = mock(INeutronNetworkCRUD.class);
when(neutronNetworkCache.getNetwork(anyString())).thenReturn(neutronNetwork);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronNetworkCache").set(neutronL3Adapter , neutronNetworkCache);
+ NodeCacheManager nodeCacheManager = mock(NodeCacheManager.class);
+ when(nodeCacheManager.getBridgeNodes()).thenReturn(nodes);
+ MemberModifier.field(NeutronL3Adapter.class, "nodeCacheManager").set(neutronL3Adapter , nodeCacheManager);
+ MemberModifier.field(NeutronL3Adapter.class, "flgDistributedARPEnabled").set(neutronL3Adapter , true);
+
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "getDpidForIntegrationBridge", Node.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programL3ForwardingStage1", Node.class, Long.class, String.class, String.class, String.class, Action.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programStaticArpStage1", Long.class, String.class, String.class, String.class, Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "updateL3ForNeutronPort", neutronPort, false);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("getDpidForIntegrationBridge", any(Node.class));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programL3ForwardingStage1", any(Node.class), anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage1", anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "updateL3ForNeutronPort", neutronPort, true);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(2)).invoke("getDpidForIntegrationBridge", any(Node.class));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programL3ForwardingStage1", any(Node.class), anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage1", anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE));
+ }
- when(neutronNetwork.getRouterExternal()).thenReturn(false); // default true
- when(neutronNetwork.getProviderSegmentationID()).thenReturn("providerSegmentationId1","providerSegmentationId2", "providerSegmentationId3");
- when(neutronNetwork.getTenantID()).thenReturn("tenantId");
- when(neutronNetwork.getNetworkUUID()).thenReturn("networkUUID");
+ // either add or remove item in l3ForwardingCache
+ @Test
+ public void testProgramL3ForwardingStage1() throws Exception {
+ Set<String> l3ForwardingCache = new HashSet<String>();
- when(neutronSubnet.getGatewayIP()).thenReturn("gatewayIp");
- when(neutronSubnet.getCidr()).thenReturn(HOST_ADDRESS + "/32");
- when(neutronSubnet.getSubnetUUID()).thenReturn("subnetUUID");
+ NodeId nodeId = mock(NodeId.class);
+ when(nodeId.getValue()).thenReturn(ID);
+ Node node = mock(Node.class);
+ when(node.getNodeId()).thenReturn(nodeId);
- when(tenantNetworkManager.isTenantNetworkPresentInNode(any(Node.class), anyString())).thenReturn(false);
- when(tenantNetworkManager.isTenantNetworkPresentInNode(any(Node.class), anyString())).thenReturn(true);
+ // Suppress the called to these functions
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programL3ForwardingStage2", Node.class, Long.class, String.class, String.class, String.class, Action.class));
- when(node.getNodeId()).thenReturn(nodeID);
+ // init instance variables
+ MemberModifier.field(NeutronL3Adapter.class, "l3ForwardingCache").set(neutronL3Adapter , l3ForwardingCache );
+ PowerMockito.when(neutronL3Adapter, "programL3ForwardingStage2", any(Node.class), anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD)).thenReturn(new Status(StatusCode.SUCCESS));
+ PowerMockito.when(neutronL3Adapter, "programL3ForwardingStage2", any(Node.class), anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE)).thenReturn(new Status(StatusCode.SUCCESS));
- when(nodeID.getValue()).thenReturn("nodeId");
+ int temp = l3ForwardingCache.size();
- when(status.isSuccess()).thenReturn(true);
+ Whitebox.invokeMethod(neutronL3Adapter, "programL3ForwardingStage1", node, Long.valueOf(45), SEG_ID, MAC_ADDRESS, IP, Action.ADD);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programL3ForwardingStage2", any(Node.class), anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+ assertEquals("Error, did not add the value", temp, l3ForwardingCache.size()-1);
- /* TODO SB_MIGRATION */
- //when(connectionService.getBridgeNodes()).thenReturn(list_nodes);
+ l3ForwardingCache.add(node.getNodeId().getValue() + ":" + SEG_ID + ":" + IP);
+ temp = l3ForwardingCache.size();
- when(configurationService.getDefaultGatewayMacAddress(any(Node.class))).thenReturn("defaultGatewayMacAddress");
- when(configurationService.getIntegrationBridgeName()).thenReturn("brName");
+ Whitebox.invokeMethod(neutronL3Adapter, "programL3ForwardingStage1", node, Long.valueOf(45), SEG_ID, MAC_ADDRESS, IP, Action.DELETE);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programL3ForwardingStage2", any(Node.class), anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE));
+ assertEquals("Error, did not delete the value", temp, l3ForwardingCache.size()+1);
+ }
- //when(ovsdbConfigurationService.getRows(any(Node.class), anyString())).thenReturn(rowMap);
- //when(ovsdbConfigurationService.getTypedRow(any(Node.class), same(Bridge.class), any(Row.class))).thenReturn(bridge);
- //when(ovsdbConfigurationService.getRow(any(Node.class), anyString(), anyString())).thenReturn(row);
+ @Test
+ public void testProgramL3ForwardingStage2() throws Exception {
+ NodeId nodeId = mock(NodeId.class);
+ when(nodeId.getValue()).thenReturn(ID);
+ Node node = mock(Node.class);
+ when(node.getNodeId()).thenReturn(nodeId);
- //when(bridge.getName()).thenReturn("brName");
- //when(bridge.getDatapathIdColumn()).thenReturn(bridgeColumnIds);
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.BADREQUEST), Whitebox.invokeMethod(neutronL3Adapter, "programL3ForwardingStage2", node, Long.valueOf(45), SEG_ID, MAC_ADDRESS, MALFORM_IP, Action.ADD));
- //when(bridgeColumnIds.getData()).thenReturn(dpids);
+ PowerMockito.mockStatic(InetAddress.class);
+ InetAddress inetAddress = mock(InetAddress.class);
+ PowerMockito.when(InetAddress.getByName(anyString())).thenReturn(inetAddress);
- when(nodeCacheManager.getBridgeNodes()).thenReturn(list_nodes);
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.SUCCESS), Whitebox.invokeMethod(neutronL3Adapter, "programL3ForwardingStage2", node, Long.valueOf(45), SEG_ID, MAC_ADDRESS, IP, Action.ADD));
}
+ @Test
+ public void testProgramFlowsForNeutronRouterInterface() throws Exception {
+ Map<String, String> networkIdToRouterMacCache = new HashMap<String, String>();
+ Map<String, List<Neutron_IPs>> networkIdToRouterIpListCache = new HashMap<String, List<Neutron_IPs>>();
+ Map<String, NeutronRouter_Interface> subnetIdToRouterInterfaceCache = new HashMap<String, NeutronRouter_Interface>();
+
+ NeutronRouter_Interface intf = mock(NeutronRouter_Interface.class);
+ when(intf.getPortUUID()).thenReturn(UUID);
+ when(intf.getSubnetUUID()).thenReturn(UUID);
+ Neutron_IPs neutronIp = mock(Neutron_IPs.class);
+ when(neutronIp.getIpAddress()).thenReturn(FIXED_IP_ADDRESS);
+ List<Neutron_IPs> ips = new ArrayList<Neutron_IPs>();
+ ips.add(neutronIp);
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ when(neutronPort.getMacAddress()).thenReturn(MAC_ADDRESS);
+ when(neutronPort.getFixedIPs()).thenReturn(ips);
+ NeutronSubnet neutronSubnet = mock(NeutronSubnet.class);
+ when(neutronSubnet.getNetworkUUID()).thenReturn(UUID);
+ when(neutronSubnet.getGatewayIP()).thenReturn(IP);
+ when(neutronSubnet.getCidr()).thenReturn("cidr");
+ NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
+ when(neutronNetwork.getProviderSegmentationID()).thenReturn(ID);
+ when(neutronNetwork.getRouterExternal()).thenReturn(false); //might change that to true
+ when(neutronNetwork.getNetworkUUID()).thenReturn(UUID);
- /**
- * Test method {@link NeutronL3Adapter#handleNeutronPortEvent(NeutronPort, Action)}
- * Device owner = network:router_interface
- */
-// @Test
- public void testHandleNeutronPortEvent1() {
- when(neutronPort.getDeviceOwner()).thenReturn("network:router_interface");
+ Node node = mock(Node.class);
+ List<Node> nodes = new ArrayList<Node>();
+ nodes.add(node);
+
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "getDpidForIntegrationBridge", Node.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programFlowsForNeutronRouterInterfacePair", Node.class, Long.class, NeutronRouter_Interface.class, NeutronRouter_Interface.class, NeutronNetwork.class, String.class, String.class, String.class, int.class, Action.class, Boolean.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programFlowForNetworkFromExternal", Node.class, Long.class, String.class, String.class, String.class, int.class, Action.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programStaticArpStage1", Long.class, String.class, String.class, String.class, Action.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programIpRewriteExclusionStage1", Node.class, Long.class, String.class, String.class, Action.class));
+
+ // init instance variables
+ INeutronPortCRUD neutronPortCache = mock(INeutronPortCRUD.class);
+ PowerMockito.when(neutronPortCache.getPort(anyString())).thenReturn(neutronPort);
+ INeutronSubnetCRUD neutronSubnetCache = mock(INeutronSubnetCRUD.class);
+ PowerMockito.when(neutronSubnetCache.getSubnet(anyString())).thenReturn(neutronSubnet);
+ INeutronNetworkCRUD neutronNetworkCache = mock(INeutronNetworkCRUD.class);
+ PowerMockito.when(neutronNetworkCache.getNetwork(anyString())).thenReturn(neutronNetwork);
+ NodeCacheManager nodeCacheManager = mock(NodeCacheManager.class);
+ PowerMockito.when(nodeCacheManager.getBridgeNodes()).thenReturn(nodes);
+ MemberModifier.field(NeutronL3Adapter.class, "networkIdToRouterMacCache").set(neutronL3Adapter , networkIdToRouterMacCache);
+ MemberModifier.field(NeutronL3Adapter.class, "networkIdToRouterIpListCache").set(neutronL3Adapter , networkIdToRouterIpListCache);
+ MemberModifier.field(NeutronL3Adapter.class, "subnetIdToRouterInterfaceCache").set(neutronL3Adapter , subnetIdToRouterInterfaceCache);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortCache").set(neutronL3Adapter , neutronPortCache);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronSubnetCache").set(neutronL3Adapter , neutronSubnetCache);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronNetworkCache").set(neutronL3Adapter , neutronNetworkCache);
+ MemberModifier.field(NeutronL3Adapter.class, "nodeCacheManager").set(neutronL3Adapter , nodeCacheManager);
+ PowerMockito.when(neutronL3Adapter, "getDpidForIntegrationBridge", any(Node.class)).thenReturn(Long.valueOf(45));
+
+ int networkIdToRouterMacCacheSize, networkIdToRouterIpListCacheSize, subnetIdToRouterInterfaceCacheSize;
+ networkIdToRouterMacCacheSize = networkIdToRouterMacCache.size();
+ networkIdToRouterIpListCacheSize = networkIdToRouterIpListCache.size();
+ subnetIdToRouterInterfaceCacheSize = subnetIdToRouterInterfaceCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForNeutronRouterInterface", intf, false);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("getDpidForIntegrationBridge", any(Node.class));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForNeutronRouterInterfacePair", any(Node.class), anyLong(), any(NeutronRouter_Interface.class), any(NeutronRouter_Interface.class), any(NeutronNetwork.class), anyString(), anyString(), anyString(), anyInt(), eq(Action.ADD), anyBoolean());
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowForNetworkFromExternal", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyInt(), eq(Action.ADD));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage1", anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programIpRewriteExclusionStage1", any(Node.class), anyLong(), anyString(), anyString(), eq(Action.ADD));
+ assertEquals("Error, did not add the RouterMac", networkIdToRouterMacCacheSize, networkIdToRouterMacCache.size() -1);
+ assertEquals("Error, did not add the RouterIP", networkIdToRouterIpListCacheSize, networkIdToRouterIpListCache.size() -1);
+ assertEquals("Error, did not add the RouterInterface", subnetIdToRouterInterfaceCacheSize, subnetIdToRouterInterfaceCache.size() -1);
+
+ networkIdToRouterMacCache.put(UUID, MAC_ADDRESS);
+ networkIdToRouterIpListCache.put(UUID, ips);
+ networkIdToRouterMacCacheSize = networkIdToRouterMacCache.size();
+ networkIdToRouterIpListCacheSize = networkIdToRouterIpListCache.size();
+ subnetIdToRouterInterfaceCacheSize = subnetIdToRouterInterfaceCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForNeutronRouterInterface", intf, true);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(2)).invoke("getDpidForIntegrationBridge", any(Node.class));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowsForNeutronRouterInterfacePair", any(Node.class), anyLong(), any(NeutronRouter_Interface.class), any(NeutronRouter_Interface.class), any(NeutronNetwork.class), anyString(), anyString(), anyString(), anyInt(), eq(Action.DELETE), anyBoolean());
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programFlowForNetworkFromExternal", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyInt(), eq(Action.DELETE));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage1", anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programIpRewriteExclusionStage1", any(Node.class), anyLong(), anyString(), anyString(), eq(Action.DELETE));
+ assertEquals("Error, did not remove the RouterMac", networkIdToRouterMacCacheSize, networkIdToRouterMacCache.size() +1);
+ assertEquals("Error, did not remove the RouterIP", networkIdToRouterIpListCacheSize, networkIdToRouterIpListCache.size() +1);
+ assertEquals("Error, did not remove the RouterInterface", subnetIdToRouterInterfaceCacheSize, subnetIdToRouterInterfaceCache.size() +1);
+ }
- neutronL3Adapter.handleNeutronPortEvent(neutronPort, Action.ADD);
- // Affected by the add
- assertEquals("Error, did not return the correct networkIdToRouterMacCache size", 1, networkIdToRouterMacCache.size());
- assertEquals("Error, did not return the correct subnetIdToRouterInterfaceCache size", 1, subnetIdToRouterInterfaceCache.size());
- /* TODO SB_MIGRATION */
- //assertEquals("Error, did not return the correct routerInterfacesCache size", 2, routerInterfacesCache.size());
-// assertEquals("Error, did not return the correct staticArpEntryCache size", 2, staticArpEntryCache.size());
-// assertEquals("Error, did not return the correct outboundIpRewriteExclusionCache size", 1, outboundIpRewriteExclusionCache.size());
-// assertEquals("Error, did not return the correct l3ForwardingCache size", 1, l3ForwardingCache.size());
- // Unchanged
- assertEquals("Error, did not return the correct inboundIpRewriteCache size", 0, inboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteCache size", 0, outboundIpRewriteCache.size());
+ @Test
+ public void testProgramFlowForNetworkFromExternal() throws Exception {
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programRouterInterfaceStage1", Node.class, Long.class, String.class, String.class, String.class, String.class, int.class, Action.class));
- neutronL3Adapter.handleNeutronPortEvent(neutronPort, Action.DELETE);
- // Affected by the delete
- assertEquals("Error, did not return the correct networkIdToRouterMacCache size", 0, networkIdToRouterMacCache.size());
- assertEquals("Error, did not return the correct subnetIdToRouterInterfaceCache size", 0, subnetIdToRouterInterfaceCache.size());
-// assertEquals("Error, did not return the correct staticArpEntryCache size", 1, staticArpEntryCache.size());
- // Unchanged
- assertEquals("Error, did not return the correct routerInterfacesCache size", 2, routerInterfacesCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteExclusionCache size", 1, outboundIpRewriteExclusionCache.size());
- assertEquals("Error, did not return the correct l3ForwardingCache size", 1, l3ForwardingCache.size());
- assertEquals("Error, did not return the correct inboundIpRewriteCache size", 0, inboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteCache size", 0, outboundIpRewriteCache.size());
- }
-
- /**
- * Test method {@link NeutronL3Adapter#handleNeutronPortEvent(NeutronPort, Action)}
- * Device owner = ""
- */
-// @Test
- public void testHandleNeutronPortEvent2() {
- when(neutronPort.getDeviceOwner()).thenReturn("");
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowForNetworkFromExternal", mock(Node.class), Long.valueOf(12), "", "", "", 4, Action.ADD);
- // populate subnetIdToRouterInterfaceCache to pass the
- // if (neutronRouterInterface != null)
- NeutronRouter_Interface neutronRouterInterface = mock(NeutronRouter_Interface.class);
- when(neutronRouterInterface.getPortUUID()).thenReturn("portUUID");
- when(neutronRouterInterface.getSubnetUUID()).thenReturn("subnetUUID");
- subnetIdToRouterInterfaceCache.put("subnetUUID", neutronRouterInterface);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programRouterInterfaceStage1", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyString(), anyInt(), any(Action.class));
+ }
- /* device owner = "" */
- neutronL3Adapter.handleNeutronPortEvent(neutronPort, Action.ADD);
- // Affected by the add
- /* TODO SB_MIGRATION */
- //assertEquals("Error, did not return the correct routerInterfacesCache size", 2, routerInterfacesCache.size());
- assertEquals("Error, did not return the correct staticArpEntryCache size", 2, staticArpEntryCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteExclusionCache size", 1, outboundIpRewriteExclusionCache.size());
- assertEquals("Error, did not return the correct l3ForwardingCache size", 1, l3ForwardingCache.size());
- assertEquals("Error, did not return the correct networkIdToRouterMacCache size", 1, networkIdToRouterMacCache.size());
- // Added above
- assertEquals("Error, did not return the correct subnetIdToRouterInterfaceCache size", 1, subnetIdToRouterInterfaceCache.size());
- // Unchanged
- assertEquals("Error, did not return the correct inboundIpRewriteCache size", 0, inboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteCache size", 0, outboundIpRewriteCache.size());
+ @Test
+ public void testProgramFlowsForNeutronRouterInterfacePair() throws Exception {
+ NeutronRouter_Interface srcNeutronRouterInterface = mock(NeutronRouter_Interface.class);
+ when(srcNeutronRouterInterface.getSubnetUUID()).thenReturn(UUID);
+ when(srcNeutronRouterInterface.getID()).thenReturn(ID);
+ when(srcNeutronRouterInterface.getPortUUID()).thenReturn(UUID);
+ NeutronRouter_Interface dstNeutronRouterInterface = mock(NeutronRouter_Interface.class);
+ NeutronNetwork neutronNetwork = mock(NeutronNetwork.class);
+ when(neutronNetwork.getTenantID()).thenReturn(ID);
+ when(neutronNetwork.getProviderSegmentationID()).thenReturn(ID);
+ Neutron_IPs ip = mock(Neutron_IPs.class);
+ when(ip.getIpAddress()).thenReturn(IP);
+ List<Neutron_IPs> ips = new ArrayList<Neutron_IPs>();
+ ips.add(ip);
+ ips.add(ip);
+ ips.add(ip);
+ NeutronPort neutronPort = mock(NeutronPort.class);
+ when(neutronPort.getMacAddress()).thenReturn(MAC_ADDRESS);
+ when(neutronPort.getFixedIPs()).thenReturn(ips);
+ NeutronSubnet neutronSubnet = mock(NeutronSubnet.class);
+ when(neutronSubnet.getNetworkUUID()).thenReturn(UUID);
+ when(neutronSubnet.getCidr()).thenReturn(CIDR);
+
+ INeutronSubnetCRUD neutronSubnetCache = mock(INeutronSubnetCRUD.class);
+ PowerMockito.when(neutronSubnetCache.getSubnet(anyString())).thenReturn(neutronSubnet);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronSubnetCache").set(neutronL3Adapter , neutronSubnetCache);
+ INeutronNetworkCRUD neutronNetworkCache = mock(INeutronNetworkCRUD.class);
+ PowerMockito.when(neutronNetworkCache.getNetwork(anyString())).thenReturn(neutronNetwork);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronNetworkCache").set(neutronL3Adapter , neutronNetworkCache);
+ INeutronPortCRUD neutronPortCache = mock(INeutronPortCRUD.class);
+ PowerMockito.when(neutronPortCache.getPort(anyString())).thenReturn(neutronPort);
+ MemberModifier.field(NeutronL3Adapter.class, "neutronPortCache").set(neutronL3Adapter , neutronPortCache);
+
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programRouterInterfaceStage1", Node.class, Long.class, String.class, String.class, String.class, String.class, int.class, Action.class));
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "getMaskLenFromCidr", String.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForNeutronRouterInterfacePair", mock(Node.class), Long.valueOf(12), srcNeutronRouterInterface, dstNeutronRouterInterface, neutronNetwork, SEG_ID, MAC_ADDRESS, IP, 4, Action.ADD, false);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programRouterInterfaceStage1", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyString(), anyInt(), any(Action.class));
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programFlowsForNeutronRouterInterfacePair", mock(Node.class), Long.valueOf(12), srcNeutronRouterInterface, dstNeutronRouterInterface, neutronNetwork, SEG_ID, MAC_ADDRESS, IP, 4, Action.ADD, true);
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(2)).invoke("programRouterInterfaceStage1", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyString(), anyInt(), any(Action.class));
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("getMaskLenFromCidr", anyString());
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(4)).invoke("programFlowsForNeutronRouterInterfacePair", any(Node.class), anyLong(), any(NeutronRouter_Interface.class), any(NeutronRouter_Interface.class), any(NeutronNetwork.class), anyString(), anyString(), anyString(), anyInt(), any(Action.class), eq(false)); //3 + 1 above
+ }
- neutronL3Adapter.handleNeutronPortEvent(neutronPort, Action.DELETE);
- // Affected by the delete
- assertEquals("Error, did not return the correct staticArpEntryCache size", 1, staticArpEntryCache.size());
- assertEquals("Error, did not return the correct l3ForwardingCache size", 0, l3ForwardingCache.size());
- // Unchanged
- assertEquals("Error, did not return the correct routerInterfacesCache size", 2, routerInterfacesCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteExclusionCache size", 1, outboundIpRewriteExclusionCache.size());
- assertEquals("Error, did not return the correct networkIdToRouterMacCache size", 1, networkIdToRouterMacCache.size());
- assertEquals("Error, did not return the correct inboundIpRewriteCache size", 0, inboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteCache size", 0, outboundIpRewriteCache.size());
- // Added above
- assertEquals("Error, did not return the correct subnetIdToRouterInterfaceCache size", 1, subnetIdToRouterInterfaceCache.size());
- }
-
- /**
- * Test method {@link NeutronL3Adapter#handleNeutronFloatingIPEvent(NeutronFloatingIP, Action)}
- */
-// @Test
- public void testandleNeutronFloatingIPEvent() throws Exception{
- NeutronFloatingIP neutronFloatingIP = mock(NeutronFloatingIP.class);
- when(neutronFloatingIP.getFixedIPAddress()).thenReturn(HOST_ADDRESS);
- when(neutronFloatingIP.getFloatingIPAddress()).thenReturn(HOST_ADDRESS);
- when(neutronFloatingIP.getFloatingNetworkUUID()).thenReturn("floatingNetworkUUID");
+ @Test
+ public void testProgramRouterInterfaceStage1() throws Exception {
+ Set<String> routerInterfacesCache = new HashSet<String>();
- networkIdToRouterMacCache.put("floatingNetworkUUID", "routerMacAddress");
+ NodeId nodeId = mock(NodeId.class);
+ when(nodeId.getValue()).thenReturn(ID);
+ Node node = mock(Node.class);
+ when(node.getNodeId()).thenReturn(nodeId);
- neutronL3Adapter.handleNeutronFloatingIPEvent(neutronFloatingIP, Action.ADD);
- // Added above
- assertEquals("Error, did not return the correct networkIdToRouterMacCache size", 1, networkIdToRouterMacCache.size());
- // Affected by the add
- /* TODO SB_MIGRATION */
- //assertEquals("Error, did not return the correct inboundIpRewriteCache size", 1, inboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteCache size", 1, outboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct staticArpEntryCache size", 1, staticArpEntryCache.size());
- // Unchanged
- assertEquals("Error, did not return the correct routerInterfacesCache size", 0, routerInterfacesCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteExclusionCache size", 0, outboundIpRewriteExclusionCache.size());
- assertEquals("Error, did not return the correct subnetIdToRouterInterfaceCache size", 0, subnetIdToRouterInterfaceCache.size());
- assertEquals("Error, did not return the correct l3ForwardingCache size", 0, l3ForwardingCache.size());
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programRouterInterfaceStage2", Node.class, Long.class, String.class, String.class, String.class, String.class, int.class, Action.class));
- neutronL3Adapter.handleNeutronFloatingIPEvent(neutronFloatingIP, Action.DELETE);
- // Unchanged
- assertEquals("Error, did not return the correct networkIdToRouterMacCache size", 1, networkIdToRouterMacCache.size());
- assertEquals("Error, did not return the correct inboundIpRewriteCache size", 1, inboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteCache size", 1, outboundIpRewriteCache.size());
- assertEquals("Error, did not return the correct staticArpEntryCache size", 1, staticArpEntryCache.size());
- assertEquals("Error, did not return the correct routerInterfacesCache size", 0, routerInterfacesCache.size());
- assertEquals("Error, did not return the correct outboundIpRewriteExclusionCache size", 0, outboundIpRewriteExclusionCache.size());
- assertEquals("Error, did not return the correct subnetIdToRouterInterfaceCache size", 0, subnetIdToRouterInterfaceCache.size());
- assertEquals("Error, did not return the correct l3ForwardingCache size", 0, l3ForwardingCache.size());
+ MemberModifier.field(NeutronL3Adapter.class, "routerInterfacesCache").set(neutronL3Adapter , routerInterfacesCache);
+ PowerMockito.when(neutronL3Adapter, "programRouterInterfaceStage2", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyString(), anyInt(), any(Action.class)).thenReturn(new Status(StatusCode.SUCCESS));
+
+ int t = routerInterfacesCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programRouterInterfaceStage1", node, Long.valueOf(12), SEG_ID, SEG_ID, MAC_ADDRESS, IP, 4, Action.ADD);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programRouterInterfaceStage1", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyString(), anyInt(), eq(Action.ADD));
+ assertEquals("Error, did not add the interface", t, routerInterfacesCache.size() - 1);
+
+ t = routerInterfacesCache.size();
+ Whitebox.invokeMethod(neutronL3Adapter, "programRouterInterfaceStage1", node, Long.valueOf(12), SEG_ID, SEG_ID, MAC_ADDRESS, IP, 4, Action.DELETE);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programRouterInterfaceStage1", any(Node.class), anyLong(), anyString(), anyString(), anyString(), anyString(), anyInt(), eq(Action.DELETE));
+ assertEquals("Error, did not delete the interface", t, routerInterfacesCache.size() + 1);
+ }
+
+ // either add or remove routerInterfacesCache
+ @Test
+ public void testProgramRouterInterfaceStage2() throws Exception {
+ NodeId nodeId = mock(NodeId.class);
+ when(nodeId.getValue()).thenReturn(ID);
+ Node node = mock(Node.class);
+ when(node.getNodeId()).thenReturn(nodeId);
+
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.BADREQUEST), Whitebox.invokeMethod(neutronL3Adapter, "programRouterInterfaceStage2", node, Long.valueOf(45), SEG_ID, SEG_ID , MAC_ADDRESS, MALFORM_IP, 4, Action.ADD));
+
+ PowerMockito.mockStatic(InetAddress.class);
+ InetAddress inetAddress = mock(InetAddress.class);
+ PowerMockito.when(InetAddress.getByName(anyString())).thenReturn(inetAddress);
+
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.SUCCESS), Whitebox.invokeMethod(neutronL3Adapter, "programRouterInterfaceStage2", node, Long.valueOf(45), SEG_ID, SEG_ID, MAC_ADDRESS, IP, 4, Action.ADD));
+
+ }
+
+ //either add or remove staticArpEntryCache
+ @Test
+ public void testProgramStaticArpStage1() throws Exception {
+ Set<String> staticArpEntryCache = new HashSet<String>();
+
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programStaticArpStage2", Long.class, String.class, String.class, String.class, Action.class));
+
+ MemberModifier.field(NeutronL3Adapter.class, "staticArpEntryCache").set(neutronL3Adapter , staticArpEntryCache);
+ PowerMockito.when(neutronL3Adapter, "programStaticArpStage2", anyLong(), anyString(), anyString(), anyString(), any(Action.class)).thenReturn(new Status(StatusCode.SUCCESS));
+
+ int t = staticArpEntryCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programStaticArpStage1", Long.valueOf(12), PORT_INT, MAC_ADDRESS, IP, Action.ADD);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage2", anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+ assertEquals("Error, did not add the static arp", t, staticArpEntryCache.size() - 1);
+
+ t = staticArpEntryCache.size();
+ Whitebox.invokeMethod(neutronL3Adapter, "programStaticArpStage1", Long.valueOf(12), PORT_INT, MAC_ADDRESS, IP, Action.DELETE);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programStaticArpStage2", anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE));
+ assertEquals("Error, did not delete the static arp", t, staticArpEntryCache.size() + 1);
+ }
+
+ @Test
+ public void testProgramStaticArpStage2() throws Exception {
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.BADREQUEST), Whitebox.invokeMethod(neutronL3Adapter, "programStaticArpStage2", Long.valueOf(45), PORT_INT, MAC_ADDRESS, MALFORM_IP, Action.ADD));
+
+ PowerMockito.mockStatic(InetAddress.class);
+ InetAddress inetAddress = mock(InetAddress.class);
+ PowerMockito.when(InetAddress.getByName(anyString())).thenReturn(inetAddress);
+
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.SUCCESS), Whitebox.invokeMethod(neutronL3Adapter, "programStaticArpStage2", Long.valueOf(45), PORT_INT, MAC_ADDRESS, IP, Action.ADD));
+ }
+
+ // either add or remove inboundIpRewriteCache
+ @Test
+ public void testProgramInboundIpRewriteStage1() throws Exception {
+ Set<String> inboundIpRewriteCache = new HashSet<String>();
+
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programInboundIpRewriteStage2", Long.class, Long.class, String.class, String.class, String.class, Action.class));
+
+ MemberModifier.field(NeutronL3Adapter.class, "inboundIpRewriteCache").set(neutronL3Adapter , inboundIpRewriteCache);
+ PowerMockito.when(neutronL3Adapter, "programInboundIpRewriteStage2", anyLong(), anyLong(), anyString(), anyString(), anyString(), any(Action.class)).thenReturn(new Status(StatusCode.SUCCESS));
+
+ int t = inboundIpRewriteCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programInboundIpRewriteStage1", Long.valueOf(12), Long.valueOf(12), PORT_INT, MAC_ADDRESS, IP, Action.ADD);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programInboundIpRewriteStage2", anyLong(), anyLong(), anyString(), anyString(), anyString(), eq(Action.ADD));
+ assertEquals("Error, did not add the inboundIpRewrite", t, inboundIpRewriteCache.size() - 1);
+
+ t = inboundIpRewriteCache.size();
+ Whitebox.invokeMethod(neutronL3Adapter, "programInboundIpRewriteStage1", Long.valueOf(12), Long.valueOf(12),PORT_INT, MAC_ADDRESS, IP, Action.DELETE);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programInboundIpRewriteStage2", anyLong(), anyLong(), anyString(), anyString(), anyString(), eq(Action.DELETE));
+ assertEquals("Error, did not delete the inboundIpRewrite", t, inboundIpRewriteCache.size() + 1);
+ }
+
+ @Test
+ public void testProgramInboundIpRewriteStage2() throws Exception {
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.BADREQUEST), Whitebox.invokeMethod(neutronL3Adapter, "programInboundIpRewriteStage2", Long.valueOf(45), Long.valueOf(45), SEG_ID, MAC_ADDRESS, IP, Action.ADD));
+
+ PowerMockito.mockStatic(InetAddress.class);
+ InetAddress inetAddress = mock(InetAddress.class);
+ PowerMockito.when(InetAddress.getByName(anyString())).thenReturn(inetAddress);
+
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.SUCCESS), Whitebox.invokeMethod(neutronL3Adapter, "programInboundIpRewriteStage2", Long.valueOf(45), Long.valueOf(45), SEG_ID, MAC_ADDRESS, IP, Action.ADD));
+ }
+
+ // either add or remove outboundIpRewriteExclusionCache
+ @Test
+ public void testProgramIpRewriteExclusionStage1() throws Exception {
+ Set<String> outboundIpRewriteExclusionCache = new HashSet<String>();
+
+ NodeId nodeId = mock(NodeId.class);
+ when(nodeId.getValue()).thenReturn(ID);
+ Node node = mock(Node.class);
+ when(node.getNodeId()).thenReturn(nodeId);
+
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programIpRewriteExclusionStage2", Node.class, Long.class, String.class, String.class, Action.class));
+
+ MemberModifier.field(NeutronL3Adapter.class, "outboundIpRewriteExclusionCache").set(neutronL3Adapter , outboundIpRewriteExclusionCache);
+ PowerMockito.when(neutronL3Adapter, "programIpRewriteExclusionStage2", any(Node.class), anyLong(), anyString(), anyString(), any(Action.class)).thenReturn(new Status(StatusCode.SUCCESS));
+
+ int t = outboundIpRewriteExclusionCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programIpRewriteExclusionStage1", node, Long.valueOf(12), SEG_ID, CIDR, Action.ADD);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programIpRewriteExclusionStage2", any(Node.class), anyLong(), anyString(), anyString(), eq(Action.ADD));
+ assertEquals("Error, did not add the outboundIpRewriteExclusion", t, outboundIpRewriteExclusionCache.size() - 1);
+
+ t = outboundIpRewriteExclusionCache.size();
+ Whitebox.invokeMethod(neutronL3Adapter, "programIpRewriteExclusionStage1", node, Long.valueOf(12), SEG_ID, CIDR, Action.DELETE);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programIpRewriteExclusionStage2", any(Node.class), anyLong(), anyString(), anyString(), eq(Action.DELETE));
+ assertEquals("Error, did not delete the outboundIpRewriteExclusion", t, outboundIpRewriteExclusionCache.size() + 1);
+ }
+
+ @Test
+ public void testProgramIpRewriteExclusionStage2() throws Exception {
+ NodeId nodeId = mock(NodeId.class);
+ when(nodeId.getValue()).thenReturn(ID);
+ Node node = mock(Node.class);
+ when(node.getNodeId()).thenReturn(nodeId);
+
+ assertEquals("Error, this not return the correct status code", new Status(StatusCode.SUCCESS), Whitebox.invokeMethod(neutronL3Adapter, "programIpRewriteExclusionStage2", node, Long.valueOf(45), SEG_ID, CIDR, Action.ADD));
+ }
+
+ // either add or remove outboundIpRewriteCache
+ @SuppressWarnings("unchecked")
+ @Test
+ public void testProgramOutboundIpRewriteStage1() throws Exception{
+ HashSet<String> outboundIpRewriteCache = new HashSet<String>();
+
+ MemberModifier.suppress(MemberMatcher.method(NeutronL3Adapter.class, "programOutboundIpRewriteStage2", floatingIpClass, Action.class));
+
+ MemberModifier.field(NeutronL3Adapter.class, "outboundIpRewriteCache").set(neutronL3Adapter , outboundIpRewriteCache);
+ PowerMockito.when(neutronL3Adapter, "programOutboundIpRewriteStage2", any(floatingIpClass), any(Action.class)).thenReturn(new Status(StatusCode.SUCCESS));
+
+ int t = outboundIpRewriteCache.size();
+
+ Whitebox.invokeMethod(neutronL3Adapter, "programOutboundIpRewriteStage1", floatingIpObject, Action.ADD);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programOutboundIpRewriteStage2", any(floatingIpClass), eq(Action.ADD));
+ assertEquals("Error, did not add the outbound ip", t, outboundIpRewriteCache.size() - 1);
+
+ t = outboundIpRewriteCache.size();
+ Whitebox.invokeMethod(neutronL3Adapter, "programOutboundIpRewriteStage1", floatingIpObject, Action.DELETE);
+
+ PowerMockito.verifyPrivate(neutronL3Adapter, times(1)).invoke("programOutboundIpRewriteStage2", any(floatingIpClass), eq(Action.DELETE));
+ assertEquals("Error, did not delete the outbound ip", t, outboundIpRewriteCache.size() + 1);
+ }
+
+ @Test
+ public void testPrepareProgramOutboundIpRewriteStage2() throws Exception {
+ assertEquals("Error, did not return the correct status code", new Status(StatusCode.BADREQUEST), Whitebox.invokeMethod(neutronL3Adapter, "programOutboundIpRewriteStage2", floatingIpObject, Action.ADD));
+
+ PowerMockito.mockStatic(InetAddress.class);
+ InetAddress inetAddress = mock(InetAddress.class);
+ PowerMockito.when(InetAddress.getByName(anyString())).thenReturn(inetAddress);
+
+ assertEquals("Error, did not return the correct status code", new Status(StatusCode.SUCCESS), Whitebox.invokeMethod(neutronL3Adapter, "programOutboundIpRewriteStage2", floatingIpObject, Action.ADD));
+ }
+
+ @Test
+ public void testGetMaskLenFromCidr() throws Exception {
+ assertEquals("Error, did not return the correct mask", 32, Whitebox.invokeMethod(neutronL3Adapter, "getMaskLenFromCidr", IP_MASK));
+ }
+
+ @Test
+ public void testGetDpidForIntegrationBridge() throws Exception {
+ Southbound southbound = mock(Southbound.class);
+ ConfigurationService configurationService = mock(ConfigurationService.class);
+
+ MemberModifier.field(NeutronL3Adapter.class, "southbound").set(neutronL3Adapter , southbound);
+ MemberModifier.field(NeutronL3Adapter.class, "configurationService").set(neutronL3Adapter , configurationService);
+
+ PowerMockito.when(southbound.getBridge(any(Node.class), anyString())).thenReturn(mock(OvsdbBridgeAugmentation.class));
+ PowerMockito.when(configurationService.getIntegrationBridgeName()).thenReturn("");
+ PowerMockito.when(southbound.getDataPathId(any(Node.class))).thenReturn(Long.valueOf(45));
+
+ assertEquals("Error, did not return the correct Dpid", Long.valueOf(45), Whitebox.invokeMethod(neutronL3Adapter, "getDpidForIntegrationBridge", mock(Node.class)));
+ }
+
+ @Test
+ public void testencodeExcplicitOFPort() throws Exception {
+ assertEquals("Error, did not correctly encode the port", OFPort, NeutronL3Adapter.encodeExcplicitOFPort(Long.valueOf(45)));
}
@Test
return field.get(neutronL3Adapter);
}
- private Method getMethod(String methodName) throws Exception {
- Method method = neutronL3Adapter.getClass().getDeclaredMethod(methodName, new Class[] {});
- method.setAccessible(true);
- return method;
+ @SuppressWarnings("rawtypes")
+ private Object createFloatingIpObject() throws Exception{
+ Class clazz = Whitebox.getInnerClassType(NeutronL3Adapter.class, "FloatIpData");
+ Constructor [] constructors = clazz.getConstructors();
+ Constructor c = constructors[0];
+ return c.newInstance(new NeutronL3Adapter(), Long.valueOf(415), Long.valueOf(415), "a", "b", "c", "d", "e");
}
}
Code Review / netvirt.git / commitdiff