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.rev130715.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.ports.attributes.ports.Port;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.Subnet;
  31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  32
  33 public final class NeutronMapperAssert {
  34     private NeutronMapperAssert() {
  35     }
  36
  37     // asserts for port
  38
  39     public static void assertPortExists(DataBroker dataBroker, Uuid portUuid) {
  40         Optional<EndpointByPort> portOptional = getEndpointByPortOptional(dataBroker, portUuid);
  41         assertTrue(portOptional.isPresent());
  42     }
  43
  44     public static void assertPortNotExists(DataBroker dataBroker, Uuid portUuid) {
  45         Optional<EndpointByPort> portOptional = getEndpointByPortOptional(dataBroker, portUuid);
  46         assertFalse(portOptional.isPresent());
  47     }
  48
  49     private static Optional<EndpointByPort> getEndpointByPortOptional(DataBroker dataBroker, Uuid portUuid) {
  50         UniqueId portId = new UniqueId(portUuid.getValue());
  51         InstanceIdentifier<EndpointByPort> iid = NeutronGbpIidFactory.endpointByPortIid(portId);
  52         Optional<EndpointByPort> portOptional;
  53         try (ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction()) {
  54             portOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.OPERATIONAL, iid, rTx);
  55         }
  56         return portOptional;
  57     }
  58
  59     public static void assertNetworkDomainExists(DataBroker dataBroker, Port port, Subnet subnet, Neutron neutron,
  60         IpAddress ipAddress) {
  61         Optional<NetworkDomain> opt =
  62                 getNetworkDomainOptional(dataBroker, port.getTenantId(), neutron, ipAddress, subnet);
  63         assertTrue(opt.isPresent());
  64     }
  65
  66     public static void assertNetworkDomainExists(DataBroker dataBroker, Uuid tenantUuid, Subnet subnet,
  67         Neutron neutron, IpAddress ipAddress) {
  68         Optional<NetworkDomain> opt = getNetworkDomainOptional(dataBroker, tenantUuid, neutron, ipAddress, subnet);
  69         assertTrue(opt.isPresent());
  70     }
  71
  72     public static void assertNetworkDomainNotExists(DataBroker dataBroker, Port port, Subnet subnet,
  73             Neutron neutron, IpAddress ipAddress) {
  74         Optional<NetworkDomain> opt =
  75                 getNetworkDomainOptional(dataBroker, port.getTenantId(), neutron, ipAddress, subnet);
  76         assertFalse(opt.isPresent());
  77     }
  78
  79     public static void assertNetworkDomainNotExists(DataBroker dataBroker, Uuid tenantUuid, Subnet subnet,
  80             Neutron neutron, IpAddress ipAddress) {
  81         Optional<NetworkDomain> opt = getNetworkDomainOptional(dataBroker, tenantUuid, neutron, ipAddress, subnet);
  82         assertFalse(opt.isPresent());
  83     }
  84
  85     private static Optional<NetworkDomain> getNetworkDomainOptional(DataBroker dataBroker, Uuid tenantUuid,
  86             Neutron neutron, IpAddress ipAddress, Subnet subnet) {
  87         InstanceIdentifier<NetworkDomain> iid;
  88         NetworkDomain subnetDomain = NeutronSubnetAware.createSubnet(subnet, neutron, ipAddress);
  89         TenantId tenantId = new TenantId(tenantUuid.getValue());
  90         iid = L2L3IidFactory.subnetIid(tenantId, subnetDomain.getNetworkDomainId());
  91         Optional<NetworkDomain> optional;
  92         try (ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction()) {
  93             optional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION, iid, rTx);
  94         }
  95         return optional;
  96     }
  97
  98 }