2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.groupbasedpolicy.neutron.mapper.mapping;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Mockito.mock;
13 import static org.mockito.Mockito.when;
15 import java.util.concurrent.ExecutionException;
16 import java.util.concurrent.Future;
18 import com.google.common.collect.ImmutableList;
19 import org.junit.Before;
20 import org.junit.Rule;
21 import org.junit.Test;
22 import org.junit.rules.ExpectedException;
23 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
24 import org.opendaylight.groupbasedpolicy.neutron.mapper.EndpointRegistrator;
25 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronMapperAssert;
26 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronMapperDataBrokerTest;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.BaseEndpointService;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.EndpointService;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.Networks;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.NetworksBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.NetworkBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.NetworkProviderExtension;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.NetworkProviderExtensionBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.Subnet;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.SubnetBuilder;
43 import org.opendaylight.yangtools.yang.common.RpcResult;
45 public class NeutronSubnetAwareDataStoreTest extends NeutronMapperDataBrokerTest {
47 private static final Uuid tenantUuid = new Uuid("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa");
48 private static final Uuid networkUuid = new Uuid("bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb");
49 private static final Uuid subnetUuid = new Uuid("cccccccc-cccc-cccc-cccc-cccccccccccc");
50 private static final Uuid subnetUuid2 = new Uuid("cccccccc-cccc-cccc-cccc-ccccccccccc2");
51 private static final IpAddress ipAddress = new IpAddress(new Ipv4Address("10.0.0.2"));
52 private static final IpAddress ipAddress2 = new IpAddress(new Ipv4Address("10.0.2.2"));
53 private static final IpPrefix ipPrefix = new IpPrefix(new Ipv4Prefix("10.0.0.0/24"));
55 private DataBroker dataBroker;
56 private Neutron neutron;
57 private EndpointRegistrator epRegistrator;
58 private EndpointService epService;
59 private BaseEndpointService baseEpService;
60 private Future<RpcResult<Void>> futureRpcResult;
61 private RpcResult<Void> rpcResult;
62 private RpcResult<Void> rpcFail;
63 private NeutronSubnetAware subnetAware;
64 private static final Subnet subnet = new SubnetBuilder().setTenantId(tenantUuid)
66 .setName("subnetName")
67 .setNetworkId(networkUuid)
68 .setGatewayIp(ipAddress)
71 private static final Subnet subnet2 = new SubnetBuilder().setTenantId(tenantUuid)
73 .setName("subnetName2")
74 .setNetworkId(networkUuid)
75 .setGatewayIp(ipAddress2)
80 public void init() throws ExecutionException, InterruptedException {
81 futureRpcResult = mock(Future.class);
82 rpcResult = mock(RpcResult.class);
83 rpcFail = mock(RpcResult.class);
84 when(rpcResult.isSuccessful()).thenReturn(true);
85 when(rpcFail.isSuccessful()).thenReturn(false);
87 dataBroker = getDataBroker();
88 neutron = mock(Neutron.class);
90 epService = mock(EndpointService.class);
91 baseEpService = mock(BaseEndpointService.class);
93 when(futureRpcResult.get()).thenReturn(rpcResult);
94 when(epService.registerL3PrefixEndpoint(
95 any(org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.RegisterL3PrefixEndpointInput.class)))
96 .thenReturn(futureRpcResult);
98 epRegistrator = new EndpointRegistrator(epService, baseEpService);
100 subnetAware = new NeutronSubnetAware(dataBroker, epRegistrator);
104 public ExpectedException thrown = ExpectedException.none();
107 public void testConstructor_invalidArgument() throws NullPointerException {
108 thrown.expect(NullPointerException.class);
109 new NeutronSubnetAware(null, null);
113 public void testOnCreated() {
115 NetworkProviderExtension networkProviderExtension =
116 new NetworkProviderExtensionBuilder().setPhysicalNetwork("physicalNetwork").build();
117 Network network = new NetworkBuilder().setUuid(networkUuid)
118 .addAugmentation(NetworkProviderExtension.class, networkProviderExtension)
120 Networks networks = new NetworksBuilder().setNetwork(ImmutableList.of(network)).build();
121 when(neutron.getNetworks()).thenReturn(networks);
123 subnetAware.onCreated(subnet, neutron);
125 NeutronMapperAssert.assertNetworkDomainExists(dataBroker, tenantUuid, subnet, ipAddress);
129 public void testOnCreated_noNetwork() {
131 Networks networks = new NetworksBuilder().setNetwork(ImmutableList.of()).build();
132 when(neutron.getNetworks()).thenReturn(networks);
134 subnetAware.onCreated(subnet, neutron);
136 NeutronMapperAssert.assertNetworkDomainNotExists(dataBroker, tenantUuid, subnet, ipAddress);
140 public void testOnUpdated() {
142 Network network = new NetworkBuilder().setUuid(networkUuid).build();
143 Networks networks = new NetworksBuilder().setNetwork(ImmutableList.of(network)).build();
144 when(neutron.getNetworks()).thenReturn(networks);
146 subnetAware.onCreated(subnet, neutron);
148 NeutronMapperAssert.assertNetworkDomainExists(dataBroker, tenantUuid, subnet, ipAddress);
149 NeutronMapperAssert.assertNetworkDomainNotExists(dataBroker, tenantUuid, subnet2, ipAddress);
151 subnetAware.onUpdated(subnet, subnet2, neutron, neutron);
153 NeutronMapperAssert.assertNetworkDomainExists(dataBroker, tenantUuid, subnet2, ipAddress);
157 public void testOnDeleted() {
159 Network network = new NetworkBuilder().setUuid(networkUuid).build();
160 Networks networks = new NetworksBuilder().setNetwork(ImmutableList.of(network)).build();
161 when(neutron.getNetworks()).thenReturn(networks);
163 subnetAware.onCreated(subnet, neutron);
165 NeutronMapperAssert.assertNetworkDomainExists(dataBroker, tenantUuid, subnet, ipAddress);
167 subnetAware.onDeleted(subnet, neutron, neutron);
169 NeutronMapperAssert.assertNetworkDomainNotExists(dataBroker, tenantUuid, subnet, ipAddress);
173 public void testOnDeleted_wrongSubnet() {
175 Network network = new NetworkBuilder().setUuid(networkUuid).build();
176 Networks networks = new NetworksBuilder().setNetwork(ImmutableList.of(network)).build();
177 when(neutron.getNetworks()).thenReturn(networks);
179 subnetAware.onCreated(subnet, neutron);
181 NeutronMapperAssert.assertNetworkDomainExists(dataBroker, tenantUuid, subnet, ipAddress);
183 subnetAware.onDeleted(subnet2, neutron, neutron);
185 NeutronMapperAssert.assertNetworkDomainExists(dataBroker, tenantUuid, subnet, ipAddress);