neutron-mapper/src/test/java/org/opendaylight/groupbasedpolicy/neutron/mapper/NeutronMapperTest.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;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertNotEquals;
  13 import static org.junit.Assert.fail;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.spy;
  16 import static org.mockito.Mockito.verify;
  17 import static org.mockito.Mockito.when;
  18
  19 import com.google.common.base.Optional;
  20 import com.google.common.collect.ImmutableList;
  21 import com.google.common.collect.ImmutableSet;
  22
  23 import java.io.IOException;
  24 import java.util.Set;
  25
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  29 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
  30 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
  31 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
  32 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  33 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  34 import org.opendaylight.groupbasedpolicy.neutron.mapper.mapping.NeutronNetworkAware;
  35 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronMapperDataBrokerTest;
  36 import org.opendaylight.groupbasedpolicy.neutron.mapper.util.MappingUtils;
  37 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  38 import org.opendaylight.groupbasedpolicy.util.IidFactory;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.BaseEndpointService;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ContextId;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.L3ContextId;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.EndpointService;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.forwarding.context.L3Context;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.l3.ext.rev150712.NetworkL3Extension;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.l3.ext.rev150712.NetworkL3ExtensionBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.NetworkBuilder;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.NeutronBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.SecurityGroups;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.SecurityGroupsBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.security.groups.SecurityGroup;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.security.groups.SecurityGroupBuilder;
  58 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  59
  60 public class NeutronMapperTest extends NeutronMapperDataBrokerTest {
  61
  62     private final Uuid tenantUuid = new Uuid("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa");
  63     private final Uuid networkUuid = new Uuid("dddddddd-dddd-dddd-dddd-dddddddddddd");
  64     private final Uuid networkUuid2 = new Uuid("dddddddd-dddd-dddd-dddd-ddddddddddd2");
  65     private static final long METADATA_IPV4_SERVER_PORT = 80;
  66     private static final IpPrefix METADATA_IP_PREFIX = new IpPrefix(new Ipv4Prefix("169.254.169.254/32"));
  67
  68     private DataBroker dataBroker;
  69     private EndpointService epService;
  70     private BaseEndpointService baseEpService;
  71
  72     private DataObjectModification<Neutron> rootNode;
  73     private Set<DataTreeModification<Neutron>> changes;
  74     private Neutron oldNeutronBefore;
  75     private Neutron oldNeutronAfter;
  76
  77     private NeutronMapper mapper;
  78     private DataObjectModification<Network> networkModif;
  79     private Network networkBefore;
  80     private Network networkAfter;
  81     private NetworkL3Extension networkL3Extension;
  82
  83     @Before
  84     public void init() {
  85         dataBroker = getDataBroker();
  86         epService = mock(EndpointService.class);
  87         baseEpService = mock(BaseEndpointService.class);
  88
  89         mapper = new NeutronMapper(dataBroker, epService, baseEpService, METADATA_IP_PREFIX, METADATA_IPV4_SERVER_PORT);
  90
  91         networkL3Extension = new NetworkL3ExtensionBuilder().setExternal(true).build();
  92
  93         rootNode = mock(DataObjectModification.class);
  94
  95         networkModif = mock(DataObjectModification.class);
  96         networkBefore = new NetworkBuilder().setTenantId(tenantUuid).setUuid(networkUuid).build();
  97         networkAfter = new NetworkBuilder().setTenantId(tenantUuid)
  98             .setUuid(networkUuid2)
  99             .addAugmentation(NetworkL3Extension.class, networkL3Extension)
 100             .build();
 101         when(networkModif.getDataType()).thenReturn(NeutronNetworkAware.NETWORK_WILDCARD_IID.getTargetType());
 102
 103         when(networkModif.getDataBefore()).thenReturn(networkBefore);
 104         when(networkModif.getDataAfter()).thenReturn(networkAfter);
 105
 106         when(rootNode.getModifiedChildren()).thenReturn(ImmutableSet.of(networkModif));
 107
 108         DataTreeModification<Neutron> change = mock(DataTreeModification.class);
 109         when(change.getRootNode()).thenReturn(rootNode);
 110
 111         changes = ImmutableSet.of(change);
 112
 113         oldNeutronBefore = new NeutronBuilder().build();
 114         SecurityGroup sg = new SecurityGroupBuilder().setTenantId(tenantUuid).setUuid(MappingUtils.EIG_UUID).build();
 115         SecurityGroups securityGroups = new SecurityGroupsBuilder().setSecurityGroup(ImmutableList.of(sg)).build();
 116         oldNeutronAfter = new NeutronBuilder().setSecurityGroups(securityGroups).build();
 117     }
 118
 119     @Test
 120     public void testConstructor() throws IOException {
 121         DataBroker dataBrokerSpy = spy(dataBroker);
 122         NeutronMapper other = new NeutronMapper(dataBrokerSpy, epService, baseEpService, METADATA_IP_PREFIX,
 123             METADATA_IPV4_SERVER_PORT);
 124
 125         verify(dataBrokerSpy).registerDataTreeChangeListener(new DataTreeIdentifier<>(
 126                 LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.builder(Neutron.class).build()), other);
 127
 128         other.close();
 129     }
 130
 131     @Test
 132     public void test_Write_withNetworkCreate() {
 133         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
 134         when(networkModif.getDataBefore()).thenReturn(null);
 135         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 136         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 137
 138         mapper.onDataTreeChanged(changes);
 139         assertNetworkExists(networkAfter);
 140     }
 141
 142     @Test
 143     public void test_Write_withNetworkUpdate() {
 144         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
 145         when(networkModif.getDataBefore()).thenReturn(null);
 146         when(networkModif.getDataAfter()).thenReturn(networkBefore);
 147         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 148         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 149
 150         mapper.onDataTreeChanged(changes);
 151
 152         assertNetworkExists(networkBefore);
 153         assertNetworkNotExists(networkAfter);
 154
 155         when(networkModif.getDataBefore()).thenReturn(networkBefore);
 156         when(networkModif.getDataAfter()).thenReturn(networkAfter);
 157
 158         mapper.onDataTreeChanged(changes);
 159         // no-op in NeutronNetworkAware#onUpdated
 160     }
 161
 162     @Test
 163     public void test_Delete() {
 164         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.DELETE);
 165         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 166         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 167
 168         mapper.onDataTreeChanged(changes);
 169     }
 170
 171     @Test
 172     public void test_SubtreeModified() {
 173         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.SUBTREE_MODIFIED);
 174         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 175         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 176
 177         mapper.onDataTreeChanged(changes);
 178     }
 179
 180     private void assertNetworkExists(Network network) {
 181         Optional<L3Context> opt = getL3ContextOptional(network);
 182         if (opt.isPresent()) {
 183             assertEquals(network.getUuid().getValue(), opt.get().getId().getValue());
 184         } else {
 185             fail("no network in DS, Uuid:" + network.getUuid());
 186         }
 187     }
 188
 189     private void assertNetworkNotExists(Network network) {
 190         Optional<L3Context> opt = getL3ContextOptional(network);
 191         if (opt.isPresent()) {
 192             assertNotEquals(network.getUuid().getValue(), opt.get().getId().getValue());
 193         }
 194     }
 195
 196     private Optional<L3Context> getL3ContextOptional(Network network) {
 197         ReadOnlyTransaction readTx = dataBroker.newReadOnlyTransaction();
 198         TenantId tenantId = new TenantId(network.getTenantId().getValue());
 199         ContextId l3CtxId = new ContextId(network.getUuid().getValue());
 200         L3ContextId l3ContextId = new L3ContextId(l3CtxId);
 201         InstanceIdentifier<L3Context> l3ContextIid = IidFactory.l3ContextIid(tenantId, l3ContextId);
 202         return DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION, l3ContextIid, readTx);
 203     }
 204 }