neutronvpn/impl/src/test/java/org/opendaylight/netvirt/neutronvpn/NeutronPortChangeListenerTest.java

   1 /*
   2  * Copyright (c) 2016 Hewlett-Packard Development Company, L.P. 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.netvirt.neutronvpn;
  10
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.when;
  14
  15 import java.util.ArrayList;
  16 import java.util.List;
  17 import java.util.Optional;
  18 import org.junit.After;
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.junit.runner.RunWith;
  22 import org.mockito.Mock;
  23 import org.mockito.junit.MockitoJUnitRunner;
  24 import org.opendaylight.genius.datastoreutils.listeners.DataTreeEventCallbackRegistrar;
  25 import org.opendaylight.infrautils.caches.baseimpl.internal.CacheManagersRegistryImpl;
  26 import org.opendaylight.infrautils.caches.guava.internal.GuavaCacheProvider;
  27 import org.opendaylight.infrautils.jobcoordinator.internal.JobCoordinatorImpl;
  28 import org.opendaylight.infrautils.metrics.MetricProvider;
  29 import org.opendaylight.infrautils.metrics.testimpl.TestMetricProviderImpl;
  30 import org.opendaylight.mdsal.binding.api.DataBroker;
  31 import org.opendaylight.mdsal.binding.api.ReadTransaction;
  32 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  33 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  34 import org.opendaylight.netvirt.elanmanager.api.IElanService;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Address;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.IdManagerService;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.ElanInstances;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstance;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstanceKey;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.config.rev160806.NeutronvpnConfig;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIps;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIpsBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.Port;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.PortBuilder;
  50 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
  51 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  52
  53 @RunWith(MockitoJUnitRunner.class)
  54 public class NeutronPortChangeListenerTest {
  55
  56     NeutronPortChangeListener neutronPortChangeListener;
  57
  58     @Mock
  59     DataBroker dataBroker;
  60     @Mock
  61     NeutronvpnManager neutronvpnManager;
  62     @Mock
  63     NeutronvpnNatManager neutronvpnNatManager;
  64     @Mock
  65     WriteTransaction mockWriteTx;
  66     @Mock
  67     ReadTransaction mockReadTx;
  68     @Mock
  69     Network mockNetwork;
  70     @Mock
  71     ElanInstance elanInstance;
  72     @Mock
  73     NeutronSubnetGwMacResolver gwMacResolver;
  74     @Mock
  75     IElanService elanService;
  76     @Mock
  77     IdManagerService idManager;
  78     @Mock
  79     IPV6InternetDefaultRouteProgrammer ipV6InternetDefRt;
  80     @Mock
  81     DataTreeEventCallbackRegistrar dataTreeEventCallbackRegistrar;
  82     @Mock
  83     NeutronvpnConfig neutronvpnConfig;
  84
  85
  86     MetricProvider metricProvider = new TestMetricProviderImpl();
  87
  88     private final JobCoordinatorImpl jobCoordinator = new JobCoordinatorImpl(metricProvider);
  89
  90     @Before
  91     public void setUp() {
  92         doReturn(mockReadTx).when(dataBroker).newReadOnlyTransaction();
  93         when(mockReadTx.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class)))
  94             .thenReturn(FluentFutures.immediateFluentFuture(Optional.of(mockNetwork)));
  95         neutronPortChangeListener = new NeutronPortChangeListener(dataBroker, neutronvpnManager, neutronvpnNatManager,
  96                 gwMacResolver, elanService, jobCoordinator, new NeutronvpnUtils(dataBroker, idManager, jobCoordinator,
  97                         ipV6InternetDefRt),
  98                 new HostConfigCache(dataBroker, new GuavaCacheProvider(new CacheManagersRegistryImpl())),
  99                 dataTreeEventCallbackRegistrar,neutronvpnConfig);
 100         InstanceIdentifier<ElanInstance> elanIdentifierId = InstanceIdentifier.builder(ElanInstances.class)
 101                 .child(ElanInstance.class,
 102                         new ElanInstanceKey(new Uuid("12345678-1234-1234-1234-123456789012").getValue())).build();
 103     }
 104
 105     @After
 106     public void cleanup() {
 107         jobCoordinator.destroy();
 108     }
 109
 110     @Test
 111     public void addPort__Ipv6FixedIps() throws Exception {
 112         PortBuilder pb = new PortBuilder();
 113         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 114         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 115         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 116         IpAddress ipv6 = new IpAddress(new Ipv6Address("1::1"));
 117         FixedIpsBuilder fib = new FixedIpsBuilder();
 118         fib.setIpAddress(ipv6);
 119         List<FixedIps> fixedIps = new ArrayList<>();
 120         fixedIps.add(fib.build());
 121         pb.setFixedIps(fixedIps);
 122         Port port = pb.build();
 123         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 124     }
 125
 126     @Test
 127     public void addPort__Ipv4FixedIps() throws Exception {
 128         PortBuilder pb = new PortBuilder();
 129         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 130         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 131         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 132         IpAddress ipv4 = new IpAddress(new Ipv4Address("2.2.2.2"));
 133         FixedIpsBuilder fib = new FixedIpsBuilder();
 134         fib.setIpAddress(ipv4);
 135         List<FixedIps> fixedIps = new ArrayList<>();
 136         fixedIps.add(fib.build());
 137         pb.setFixedIps(fixedIps);
 138         Port port = pb.build();
 139         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 140     }
 141
 142     @Test
 143     public void addPort__NoFixedIps() throws Exception {
 144         PortBuilder pb = new PortBuilder();
 145         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 146         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 147         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 148         List<FixedIps> fixedIps = new ArrayList<>();
 149         pb.setFixedIps(fixedIps);
 150         Port port = pb.build();
 151         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 152     }
 153
 154 }