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 java.io.IOException;
  20 import java.util.Set;
  21
  22 import com.google.common.base.Optional;
  23 import com.google.common.collect.ImmutableList;
  24 import com.google.common.collect.ImmutableSet;
  25 import org.junit.Before;
  26 import org.junit.Test;
  27 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  28 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
  29 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
  30 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
  31 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  32 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  33 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
  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.yang.types.rev130715.Uuid;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.BaseEndpointService;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ContextId;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.L3ContextId;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.EndpointService;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.forwarding.context.L3Context;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.l3.ext.rev150712.NetworkL3Extension;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.l3.ext.rev150712.NetworkL3ExtensionBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.NetworkBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.NeutronBuilder;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.SecurityGroups;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.SecurityGroupsBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.security.groups.SecurityGroup;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.security.groups.SecurityGroupBuilder;
  56 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  57
  58 public class NeutronMapperTest extends NeutronMapperDataBrokerTest {
  59
  60     private final Uuid tenantUuid = new Uuid("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa");
  61     private final Uuid networkUuid = new Uuid("dddddddd-dddd-dddd-dddd-dddddddddddd");
  62     private final Uuid networkUuid2 = new Uuid("dddddddd-dddd-dddd-dddd-ddddddddddd2");
  63
  64     private DataBroker dataBroker;
  65     private RpcProviderRegistry rpcProvider;
  66     private EndpointService epService;
  67     private BaseEndpointService baseEpService;
  68
  69     private DataObjectModification<Neutron> rootNode;
  70     private Set<DataTreeModification<Neutron>> changes;
  71     private Neutron oldNeutronBefore;
  72     private Neutron oldNeutronAfter;
  73
  74     private NeutronMapper mapper;
  75     private DataObjectModification<Network> networkModif;
  76     private Network networkBefore;
  77     private Network networkAfter;
  78     private NetworkL3Extension networkL3Extension;
  79
  80     @Before
  81     public void init() {
  82         dataBroker = getDataBroker();
  83         epService = mock(EndpointService.class);
  84         baseEpService = mock(BaseEndpointService.class);
  85         rpcProvider = mock(RpcProviderRegistry.class);
  86         when(rpcProvider.getRpcService(EndpointService.class)).thenReturn(epService);
  87         when(rpcProvider.getRpcService(BaseEndpointService.class)).thenReturn(baseEpService);
  88
  89         mapper = new NeutronMapper(dataBroker, rpcProvider);
  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, rpcProvider);
 123
 124         verify(dataBrokerSpy).registerDataTreeChangeListener(new DataTreeIdentifier<>(
 125                 LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.builder(Neutron.class).build()), other);
 126
 127         other.close();
 128     }
 129
 130     @Test
 131     public void test_Write_withNetworkCreate() {
 132         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
 133         when(networkModif.getDataBefore()).thenReturn(null);
 134         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 135         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 136
 137         mapper.onDataTreeChanged(changes);
 138         assertNetworkExists(networkAfter);
 139     }
 140
 141     @Test
 142     public void test_Write_withNetworkUpdate() {
 143         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
 144         when(networkModif.getDataBefore()).thenReturn(null);
 145         when(networkModif.getDataAfter()).thenReturn(networkBefore);
 146         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 147         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 148
 149         mapper.onDataTreeChanged(changes);
 150
 151         assertNetworkExists(networkBefore);
 152         assertNetworkNotExists(networkAfter);
 153
 154         when(networkModif.getDataBefore()).thenReturn(networkBefore);
 155         when(networkModif.getDataAfter()).thenReturn(networkAfter);
 156
 157         mapper.onDataTreeChanged(changes);
 158         // no-op in NeutronNetworkAware#onUpdated
 159     }
 160
 161     @Test
 162     public void test_Delete() {
 163         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.DELETE);
 164         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 165         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 166
 167         mapper.onDataTreeChanged(changes);
 168     }
 169
 170     @Test
 171     public void test_SubtreeModified() {
 172         when(networkModif.getModificationType()).thenReturn(DataObjectModification.ModificationType.SUBTREE_MODIFIED);
 173         when(rootNode.getDataBefore()).thenReturn(oldNeutronBefore);
 174         when(rootNode.getDataAfter()).thenReturn(oldNeutronAfter);
 175
 176         mapper.onDataTreeChanged(changes);
 177     }
 178
 179     private void assertNetworkExists(Network network) {
 180         Optional<L3Context> opt = getL3ContextOptional(network);
 181         if (opt.isPresent()) {
 182             assertEquals(network.getUuid().getValue(), opt.get().getId().getValue());
 183         } else {
 184             fail("no network in DS, Uuid:" + network.getUuid());
 185         }
 186     }
 187
 188     private void assertNetworkNotExists(Network network) {
 189         Optional<L3Context> opt = getL3ContextOptional(network);
 190         if (opt.isPresent()) {
 191             assertNotEquals(network.getUuid().getValue(), opt.get().getId().getValue());
 192         }
 193     }
 194
 195     private Optional<L3Context> getL3ContextOptional(Network network) {
 196         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
 197         TenantId tenantId = new TenantId(network.getTenantId().getValue());
 198         ContextId l3CtxId = new ContextId(network.getUuid().getValue());
 199         L3ContextId l3ContextId = new L3ContextId(l3CtxId);
 200         InstanceIdentifier<L3Context> l3ContextIid = IidFactory.l3ContextIid(tenantId, l3ContextId);
 201         return DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION, l3ContextIid, rTx);
 202     }
 203 }