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
  16 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  17 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  19 import org.opendaylight.groupbasedpolicy.domain_extension.l2_l3.util.L2L3IidFactory;
  20 import org.opendaylight.groupbasedpolicy.neutron.gbp.util.NeutronGbpIidFactory;
  21 import org.opendaylight.groupbasedpolicy.neutron.mapper.mapping.NeutronSubnetAware;
  22 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.UniqueId;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.rev160427.forwarding.forwarding.by.tenant.NetworkDomain;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.neutron.gbp.mapper.rev150513.mappings.gbp.by.neutron.mappings.endpoints.by.ports.EndpointByPort;
  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.rev150712.Neutron;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.subnets.rev150712.subnets.attributes.subnets.Subnet;
  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, Neutron neutron,
  59         IpAddress ipAddress) {
  60         Optional<NetworkDomain> opt =
  61                 getNetworkDomainOptional(dataBroker, port.getTenantId(), neutron, ipAddress, subnet);
  62         assertTrue(opt.isPresent());
  63     }
  64
  65     public static void assertNetworkDomainExists(DataBroker dataBroker, Uuid tenantUuid, Subnet subnet,
  66         Neutron neutron, IpAddress ipAddress) {
  67         Optional<NetworkDomain> opt = getNetworkDomainOptional(dataBroker, tenantUuid, neutron, ipAddress, subnet);
  68         assertTrue(opt.isPresent());
  69     }
  70
  71     public static void assertNetworkDomainNotExists(DataBroker dataBroker, Port port, Subnet subnet,
  72             Neutron neutron, IpAddress ipAddress) {
  73         Optional<NetworkDomain> opt =
  74                 getNetworkDomainOptional(dataBroker, port.getTenantId(), neutron, ipAddress, subnet);
  75         assertFalse(opt.isPresent());
  76     }
  77
  78     public static void assertNetworkDomainNotExists(DataBroker dataBroker, Uuid tenantUuid, Subnet subnet,
  79             Neutron neutron, IpAddress ipAddress) {
  80         Optional<NetworkDomain> opt = getNetworkDomainOptional(dataBroker, tenantUuid, neutron, ipAddress, subnet);
  81         assertFalse(opt.isPresent());
  82     }
  83
  84     private static Optional<NetworkDomain> getNetworkDomainOptional(DataBroker dataBroker, Uuid tenantUuid,
  85             Neutron neutron, IpAddress ipAddress, Subnet subnet) {
  86         InstanceIdentifier<NetworkDomain> iid;
  87         NetworkDomain subnetDomain = NeutronSubnetAware.createSubnet(subnet, neutron, ipAddress);
  88         TenantId tenantId = new TenantId(tenantUuid.getValue());
  89         iid = L2L3IidFactory.subnetIid(tenantId, subnetDomain.getNetworkDomainId());
  90         Optional<NetworkDomain> optional;
  91         try (ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction()) {
  92             optional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION, iid, rTx);
  93         }
  94         return optional;
  95     }
  96
  97 }