neutron-mapper/src/test/java/org/opendaylight/groupbasedpolicy/neutron/mapper/test/NeutronMapperAssert.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.test;
  10
  11 import static org.junit.Assert.assertFalse;
  12 import static org.junit.Assert.assertTrue;
  13
  14 import com.google.common.base.Optional;
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  18 import org.opendaylight.groupbasedpolicy.domain_extension.l2_l3.util.L2L3IidFactory;
  19 import org.opendaylight.groupbasedpolicy.neutron.gbp.util.NeutronGbpIidFactory;
  20 import org.opendaylight.groupbasedpolicy.neutron.mapper.mapping.NeutronSubnetAware;
  21 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.UniqueId;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.rev160427.forwarding.forwarding.by.tenant.NetworkDomain;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.neutron.gbp.mapper.rev150513.mappings.gbp.by.neutron.mappings.endpoints.by.ports.EndpointByPort;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIps;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.Port;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.Subnet;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.SubnetBuilder;
  32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  33
  34 public class NeutronMapperAssert {
  35
  36     // asserts for port
  37
  38     public static void assertPortExists(DataBroker dataBroker, Uuid portUuid) {
  39         Optional<EndpointByPort> portOptional = getEndpointByPortOptional(dataBroker, portUuid);
  40         assertTrue(portOptional.isPresent());
  41     }
  42
  43     public static void assertPortNotExists(DataBroker dataBroker, Uuid portUuid) {
  44         Optional<EndpointByPort> portOptional = getEndpointByPortOptional(dataBroker, portUuid);
  45         assertFalse(portOptional.isPresent());
  46     }
  47
  48     private static Optional<EndpointByPort> getEndpointByPortOptional(DataBroker dataBroker, Uuid portUuid) {
  49         UniqueId portId = new UniqueId(portUuid.getValue());
  50         InstanceIdentifier<EndpointByPort> iid = NeutronGbpIidFactory.endpointByPortIid(portId);
  51         Optional<EndpointByPort> portOptional;
  52         try (ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction()) {
  53             portOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.OPERATIONAL, iid, rTx);
  54         }
  55         return portOptional;
  56     }
  57
  58     public static void assertNetworkDomainExists(DataBroker dataBroker, Port port, Subnet subnet, IpAddress ipAddress) {
  59         Optional<NetworkDomain> portOptional =
  60                 getNetworkDomainOptional(dataBroker, port.getTenantId(), ipAddress, subnet);
  61         assertTrue(portOptional.isPresent());
  62     }
  63
  64     public static void assertNetworkDomainNotExists(DataBroker dataBroker, Port port, Subnet subnet,
  65             IpAddress ipAddress) {
  66         Optional<NetworkDomain> portOptional =
  67                 getNetworkDomainOptional(dataBroker, port.getTenantId(), ipAddress, subnet);
  68         assertFalse(portOptional.isPresent());
  69     }
  70
  71     private static Optional<NetworkDomain> getNetworkDomainOptional(DataBroker dataBroker, Uuid tenantUuid,
  72             IpAddress ipAddress, Subnet subnet) {
  73         InstanceIdentifier<NetworkDomain> iid;
  74         NetworkDomain subnetDomain = NeutronSubnetAware.createSubnet(subnet, ipAddress);
  75         TenantId tenantId = new TenantId(tenantUuid.getValue());
  76         iid = L2L3IidFactory.subnetIid(tenantId, subnetDomain.getNetworkDomainId());
  77         Optional<NetworkDomain> optional;
  78         try (ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction()) {
  79             optional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION, iid, rTx);
  80         }
  81         return optional;
  82     }
  83
  84 }