neutron-mapper/src/test/java/org/opendaylight/groupbasedpolicy/neutron/mapper/mapping/NeutronPortAwareDataStoreTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. 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.groupbasedpolicy.neutron.mapper.mapping;
  10
  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.util.Collections;
  16 import java.util.List;
  17
  18 import org.junit.Before;
  19 import org.junit.Rule;
  20 import org.junit.Test;
  21 import org.junit.rules.ExpectedException;
  22 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  23 import org.opendaylight.groupbasedpolicy.neutron.mapper.EndpointRegistrator;
  24 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronMapperAssert;
  25 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronMapperDataBrokerTest;
  26 import org.opendaylight.groupbasedpolicy.neutron.mapper.util.PortUtils;
  27 import org.opendaylight.groupbasedpolicy.neutron.mapper.util.Utils;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.RegisterEndpointInput;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.UnregisterEndpointInput;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIps;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIpsBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.Ports;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.PortsBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.Port;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.PortBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.Subnets;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.SubnetsBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.Subnet;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.SubnetBuilder;
  47
  48 import com.google.common.collect.ImmutableList;
  49
  50 public class NeutronPortAwareDataStoreTest extends NeutronMapperDataBrokerTest {
  51
  52     private final Uuid tenantUuid = new Uuid("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa");
  53     private final Uuid portUuid = new Uuid("bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb");
  54     private final Uuid subnetUuid = new Uuid("cccccccc-cccc-cccc-cccc-cccccccccccc");
  55     private final Uuid networkUuid = new Uuid("dddddddd-dddd-dddd-dddd-dddddddddddd");
  56     private final Uuid uuidReserved3 = new Uuid("dddddddd-dddd-dddd-dddd-ddddddddddd3");
  57
  58     private DataBroker dataBroker;
  59     private NeutronPortAware portAware;
  60     private EndpointRegistrator epRegistrator;
  61     private Neutron neutron;
  62
  63     @Before
  64     public void init() {
  65         dataBroker = getDataBroker();
  66         neutron = mock(Neutron.class);
  67         epRegistrator = mock(EndpointRegistrator.class);
  68         when(epRegistrator.registerEndpoint(any(RegisterEndpointInput.class))).thenReturn(true);
  69         when(epRegistrator.registerEndpoint(
  70                 any(org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.RegisterEndpointInput.class)))
  71                     .thenReturn(true);
  72         when(epRegistrator.unregisterEndpoint(any(UnregisterEndpointInput.class))).thenReturn(true);
  73         when(epRegistrator.unregisterEndpoint(
  74                 any(org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.UnregisterEndpointInput.class)))
  75                     .thenReturn(true);
  76
  77         portAware = new NeutronPortAware(dataBroker, epRegistrator, new IpPrefix(new Ipv4Prefix("192.168.192.168/32")));
  78     }
  79
  80     @Rule
  81     public ExpectedException thrown = ExpectedException.none();
  82
  83     @Test
  84     public void testConstructor_invalidArgument() throws NullPointerException {
  85         thrown.expect(NullPointerException.class);
  86         new NeutronPortAware(null, null, null);
  87     }
  88
  89     @Test
  90     public void test_createAndDeleteDhcpPort() {
  91         IpAddress ipAddress = new IpAddress(new Ipv4Address("10.0.0.2"));
  92         Subnets subnets = createSubnets();
  93         when(neutron.getSubnets()).thenReturn(subnets);
  94
  95         FixedIps portIpWithSubnet = createFixedIps(ipAddress);
  96         Port port = newBasePort().setDeviceOwner(PortUtils.DEVICE_OWNER_DHCP)
  97             .setFixedIps(ImmutableList.of(portIpWithSubnet))
  98             .build();
  99         when(neutron.getPorts()).thenReturn(new PortsBuilder().setPort(Collections.singletonList(port)).build());
 100
 101         portAware.onCreated(port, neutron);
 102         NeutronMapperAssert.assertPortExists(dataBroker, port.getUuid());
 103
 104         portAware.onDeleted(port, neutron, neutron);
 105         NeutronMapperAssert.assertPortNotExists(dataBroker, port.getUuid());
 106     }
 107
 108     @Test
 109     public void test_createDhcpPort_noFixedIps() {
 110         Subnets subnets = createSubnets();
 111         when(neutron.getSubnets()).thenReturn(subnets);
 112
 113         Port port = newBasePort().setDeviceOwner(PortUtils.DEVICE_OWNER_DHCP).build();
 114         portAware.onCreated(port, neutron);
 115         NeutronMapperAssert.assertPortNotExists(dataBroker, port.getUuid());
 116     }
 117
 118     @Test
 119     public void test_createAndDeleteNormalPort() throws Exception {
 120         IpAddress ipAddress = new IpAddress(new Ipv4Address("10.0.0.2"));
 121         Subnets subnets = createSubnets();
 122         when(neutron.getSubnets()).thenReturn(subnets);
 123
 124         FixedIps portIpWithSubnet = createFixedIps(ipAddress);
 125         Port port = newBasePort().setDeviceOwner("owner for normal port")
 126             .setFixedIps(ImmutableList.of(portIpWithSubnet))
 127             .build();
 128         when(neutron.getPorts()).thenReturn(new PortsBuilder().setPort(Collections.singletonList(port)).build());
 129
 130         portAware.onCreated(port, neutron);
 131         NeutronMapperAssert.assertPortExists(dataBroker, port.getUuid());
 132
 133         portAware.onDeleted(port, neutron, neutron);
 134         NeutronMapperAssert.assertPortNotExists(dataBroker, port.getUuid());
 135     }
 136
 137     @Test
 138     public void test_createAndUpdateNormalPort() {
 139         IpAddress ipAddress = new IpAddress(new Ipv4Address("10.0.0.2"));
 140         Subnets subnets = createSubnets();
 141         when(neutron.getSubnets()).thenReturn(subnets);
 142
 143         FixedIps portIpWithSubnet = createFixedIps(ipAddress);
 144         Port port = newBasePort().setDeviceOwner("owner for normal port")
 145             .setFixedIps(ImmutableList.of(portIpWithSubnet))
 146             .build();
 147         portAware.onCreated(port, neutron);
 148         NeutronMapperAssert.assertPortExists(dataBroker, port.getUuid());
 149
 150         Port newPort = new PortBuilder(port).setName("updatedName").build();
 151         portAware.onUpdated(port, newPort, neutron, neutron);
 152         NeutronMapperAssert.assertPortExists(dataBroker, port.getUuid());
 153     }
 154
 155     @Test
 156     public void test_createNormalPort_noFixedIps() {
 157         Subnets subnets = createSubnets();
 158         when(neutron.getSubnets()).thenReturn(subnets);
 159
 160         Port port = newBasePort().setDeviceOwner("owner for normal port").build();
 161         portAware.onCreated(port, neutron);
 162         NeutronMapperAssert.assertPortExists(dataBroker, port.getUuid());
 163     }
 164
 165     @Test
 166     public void test_createAndDeleteRouterInterfacePort() {
 167         IpAddress ipAddress = new IpAddress(new Ipv4Address("10.0.0.2"));
 168         Subnets subnets = createSubnets();
 169         when(neutron.getSubnets()).thenReturn(subnets);
 170
 171         FixedIps portIpWithSubnet = createFixedIps(ipAddress);
 172         Port neutronPort1 = new PortBuilder().setTenantId(new Uuid(tenantUuid))
 173             .setFixedIps(ImmutableList.of(portIpWithSubnet))
 174             .setName("portName1")
 175             .setUuid(portUuid)
 176             .setDeviceId("deviceId")
 177             .setFixedIps(ImmutableList.of(portIpWithSubnet))
 178             .setNetworkId(networkUuid)
 179             .setMacAddress(new MacAddress("00:00:00:00:35:02"))
 180             .build();
 181         Port neutronPort2 = new PortBuilder().setTenantId(new Uuid(tenantUuid))
 182             .setFixedIps(ImmutableList.of(portIpWithSubnet))
 183             .setName("portName2")
 184             .setUuid(portUuid)
 185             .setDeviceId("deviceId")
 186             .setDeviceOwner(PortUtils.DEVICE_OWNER_DHCP)
 187             .setFixedIps(ImmutableList.of(portIpWithSubnet))
 188             .setNetworkId(networkUuid)
 189             .setMacAddress(new MacAddress("00:00:00:00:35:02"))
 190             .build();
 191         Ports neutronPorts = new PortsBuilder().setPort(ImmutableList.of(neutronPort1, neutronPort2)).build();
 192         when(neutron.getPorts()).thenReturn(neutronPorts);
 193
 194         Subnet subnet = subnets.getSubnet().get(0);
 195         Port port = newBasePort().setDeviceOwner(PortUtils.DEVICE_OWNER_ROUTER_IFACE)
 196             .setFixedIps(ImmutableList.of(portIpWithSubnet))
 197             .setDeviceId("37ffd9b3-30c6-4d21-96df-00c04fc9552f")
 198             .build();
 199         when(neutron.getPorts()).thenReturn(new PortsBuilder().setPort(Collections.singletonList(port)).build());
 200         portAware.onCreated(port, neutron);
 201         NeutronMapperAssert.assertNetworkDomainExists(dataBroker, port, subnet, neutron, ipAddress);
 202
 203         portAware.onDeleted(port, neutron, neutron);
 204         NeutronMapperAssert.assertNetworkDomainExists(dataBroker, port, subnet, neutron, ipAddress);
 205         NeutronMapperAssert.assertPortNotExists(dataBroker, port.getUuid());
 206     }
 207
 208     @Test
 209     public void test_createRouterInterfacePort_noFixedIps() {
 210         Subnets subnets = createSubnets();
 211         when(neutron.getSubnets()).thenReturn(subnets);
 212
 213         Subnet subnet = subnets.getSubnet().get(0);
 214         Port port = newBasePort().setDeviceOwner(PortUtils.DEVICE_OWNER_ROUTER_IFACE).build();
 215         portAware.onCreated(port, neutron);
 216         NeutronMapperAssert.assertNetworkDomainNotExists(dataBroker, port, subnet, neutron, null);
 217     }
 218
 219     @Test
 220     public void test_createAndDeleteRouterGatewayPort() {
 221         Port port = new PortBuilder().setUuid(portUuid).setDeviceOwner(PortUtils.DEVICE_OWNER_ROUTER_GATEWAY).build();
 222         portAware.onCreated(port, neutron);
 223         portAware.onDeleted(port, neutron, neutron);
 224         // no op
 225     }
 226
 227     @Test
 228     public void test_createAndDeleteFloatingIpPort() {
 229         Port port = new PortBuilder().setUuid(portUuid).setDeviceOwner(PortUtils.DEVICE_OWNER_FLOATING_IP).build();
 230         portAware.onCreated(port, neutron);
 231         portAware.onDeleted(port, neutron, neutron);
 232         // no op
 233     }
 234
 235     private PortBuilder newBasePort() {
 236         List<Uuid> secGroups = ImmutableList.of(uuidReserved3);
 237         return new PortBuilder().setTenantId(new Uuid(tenantUuid))
 238             .setSecurityGroups(secGroups)
 239             .setName("portName")
 240             .setUuid(portUuid)
 241             .setDeviceId("deviceId")
 242             .setNetworkId(networkUuid)
 243             .setMacAddress(new MacAddress("00:00:00:00:35:02"));
 244     }
 245
 246     private Subnets createSubnets() {
 247         Subnet subnet = new SubnetBuilder().setTenantId(tenantUuid)
 248             .setUuid(subnetUuid)
 249             .setName("subnetName")
 250             .setNetworkId(networkUuid)
 251             .setCidr(Utils.createIpPrefix("10.0.0.0/24"))
 252             .build();
 253         return new SubnetsBuilder().setSubnet(ImmutableList.of(subnet)).build();
 254     }
 255
 256     private FixedIps createFixedIps(IpAddress ipAddress) {
 257         return new FixedIpsBuilder().setSubnetId(subnetUuid).setIpAddress(ipAddress).build();
 258     }
 259
 260 }