vpnservice/neutronvpn/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.Matchers.eq;
  13 import static org.mockito.Mockito.doReturn;
  14 import static org.mockito.Mockito.when;
  15
  16 import com.google.common.base.Optional;
  17 import com.google.common.util.concurrent.Futures;
  18 import java.util.ArrayList;
  19 import java.util.List;
  20 import org.junit.Before;
  21 import org.junit.Test;
  22 import org.junit.runner.RunWith;
  23 import org.mockito.Mock;
  24 import org.mockito.runners.MockitoJUnitRunner;
  25 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  26 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  27 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
  28 import org.opendaylight.controller.md.sal.binding.api.NotificationService;
  29 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  30 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  31 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
  32 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  33 import org.opendaylight.netvirt.elanmanager.api.IElanService;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Address;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rpcs.rev160406.OdlInterfaceRpcService;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.ElanInstances;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstance;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstanceKey;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIps;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIpsBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.Port;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.PortBuilder;
  48 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  49 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  50
  51 @RunWith(MockitoJUnitRunner.class)
  52 public class NeutronPortChangeListenerTest {
  53
  54     NeutronPortChangeListener neutronPortChangeListener;
  55
  56     @Mock
  57     DataBroker dataBroker;
  58     @Mock
  59     NeutronvpnManager neutronvpnManager;
  60     @Mock
  61     NeutronvpnNatManager neutronvpnNatManager;
  62     @Mock
  63     NotificationPublishService notiPublishService;
  64     @Mock
  65     NotificationService notiService;
  66     @Mock
  67     OdlInterfaceRpcService odlInterfaceRpcService;
  68     @Mock
  69     NeutronFloatingToFixedIpMappingChangeListener floatingIpMapListener;
  70     @Mock
  71     ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  72     @Mock
  73     WriteTransaction mockWriteTx;
  74     @Mock
  75     ReadOnlyTransaction mockReadTx;
  76     @Mock
  77     Network mockNetwork;
  78     @Mock
  79     ElanInstance elanInstance;
  80     @Mock
  81     NeutronSubnetGwMacResolver gwMacResolver;
  82     @Mock
  83     IElanService elanService;
  84
  85     @Before
  86     public void setUp() {
  87         when(dataBroker.registerDataChangeListener(any(LogicalDatastoreType.class), //
  88                 any(InstanceIdentifier.class), //
  89                 any(DataChangeListener.class), //
  90                 any(AsyncDataBroker.DataChangeScope.class))). //
  91                 thenReturn(dataChangeListenerRegistration);
  92         doReturn(mockWriteTx).when(dataBroker).newWriteOnlyTransaction();
  93         doReturn(Futures.immediateCheckedFuture(null)).when(mockWriteTx).submit();
  94         doReturn(mockReadTx).when(dataBroker).newReadOnlyTransaction();
  95         when(mockReadTx.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class)))
  96             .thenReturn(Futures.immediateCheckedFuture(Optional.of(mockNetwork)));
  97         neutronPortChangeListener = new NeutronPortChangeListener(dataBroker, neutronvpnManager, neutronvpnNatManager,
  98                 notiPublishService, gwMacResolver, odlInterfaceRpcService, elanService);
  99         InstanceIdentifier<ElanInstance> elanIdentifierId = InstanceIdentifier.builder(ElanInstances.class)
 100                 .child(ElanInstance.class,
 101                         new ElanInstanceKey(new Uuid("12345678-1234-1234-1234-123456789012").getValue())).build();
 102         when(mockReadTx.read(any(LogicalDatastoreType.class), eq(elanIdentifierId)))
 103             .thenReturn(Futures.immediateCheckedFuture(Optional.of(elanInstance)));
 104     }
 105
 106     @Test
 107     public void addPort__Ipv6FixedIps() throws Exception {
 108         PortBuilder pb = new PortBuilder();
 109         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 110         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 111         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 112         IpAddress ipv6 = new IpAddress(new Ipv6Address("1::1"));
 113         FixedIpsBuilder fib = new FixedIpsBuilder();
 114         fib.setIpAddress(ipv6);
 115         List<FixedIps> fixedIps = new ArrayList<>();
 116         fixedIps.add(fib.build());
 117         pb.setFixedIps(fixedIps);
 118         Port port = pb.build();
 119         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 120     }
 121
 122     @Test
 123     public void addPort__Ipv4FixedIps() throws Exception {
 124         PortBuilder pb = new PortBuilder();
 125         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 126         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 127         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 128         IpAddress ipv4 = new IpAddress(new Ipv4Address("2.2.2.2"));
 129         FixedIpsBuilder fib = new FixedIpsBuilder();
 130         fib.setIpAddress(ipv4);
 131         List<FixedIps> fixedIps = new ArrayList<>();
 132         fixedIps.add(fib.build());
 133         pb.setFixedIps(fixedIps);
 134         Port port = pb.build();
 135         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 136     }
 137
 138     @Test
 139     public void addPort__NoFixedIps() throws Exception {
 140         PortBuilder pb = new PortBuilder();
 141         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 142         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 143         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 144         List<FixedIps> fixedIps = new ArrayList<>();
 145         pb.setFixedIps(fixedIps);
 146         Port port = pb.build();
 147         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 148     }
 149
 150 }