neutron-mapper/src/test/java/org/opendaylight/groupbasedpolicy/neutron/mapper/mapping/rule/NeutronSecurityRuleAwareTest.java

   1 package org.opendaylight.groupbasedpolicy.neutron.mapper.mapping.rule;
   2
   3 import static org.junit.Assert.assertEquals;
   4 import static org.junit.Assert.assertFalse;
   5 import static org.junit.Assert.assertNotNull;
   6 import static org.junit.Assert.assertTrue;
   7
   8 import java.util.ArrayList;
   9 import java.util.List;
  10
  11 import org.junit.Test;
  12 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  13 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.ConfigDataStoreReader;
  14 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronMapperDataBrokerTest;
  15 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronEntityFactory;
  16 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.PolicyAssert;
  17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpPrefix;
  18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Prefix;
  19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.HasDirection.Direction;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.policy.Contract;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.DirectionIngress;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.EthertypeV4;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.ProtocolTcp;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.NeutronBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.SecurityRuleAttributes;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.SecurityGroupsBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.security.groups.SecurityGroup;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.rules.attributes.security.rules.SecurityRule;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.rules.attributes.security.rules.SecurityRuleBuilder;
  32
  33 public class NeutronSecurityRuleAwareTest extends NeutronMapperDataBrokerTest {
  34
  35     private static final String RULE_ID = "00000000-0000-0000-0000-000000000001";
  36     private static final String RULE_TENANT_ID = "00000000-0000-0000-0000-000000000002";
  37     private static final String RULE_GROUP_ID = "00000000-0000-0000-0000-000000000003";
  38
  39     @Test
  40     public final void testIsDirectionOpposite_InIn() {
  41         assertFalse(NeutronSecurityRuleAware.isDirectionOpposite(Direction.In, Direction.In));
  42     }
  43
  44     @Test
  45     public final void testIsDirectionOpposite_OutOut() {
  46         assertFalse(NeutronSecurityRuleAware.isDirectionOpposite(Direction.Out, Direction.Out));
  47     }
  48
  49     @Test
  50     public final void testIsDirectionOpposite_InOut() {
  51         assertTrue(NeutronSecurityRuleAware.isDirectionOpposite(Direction.In, Direction.Out));
  52     }
  53
  54     @Test
  55     public final void testIsDirectionOpposite_OutIn() {
  56         assertTrue(NeutronSecurityRuleAware.isDirectionOpposite(Direction.Out, Direction.In));
  57     }
  58
  59     @Test
  60     public void testNeutronSecurityRuleCreatedAndDeleted() throws Exception {
  61         DataBroker dataProvider = getDataBroker();
  62         NeutronSecurityRuleAware neutronSecurityRuleAware = new NeutronSecurityRuleAware(dataProvider);
  63
  64         //create security rule and put to DS
  65         SecurityRule neutronRule = buildNeutronSecurityRule();
  66         List<SecurityGroup> secGroups = new ArrayList<>();
  67         secGroups.add(NeutronEntityFactory.securityGroup(neutronRule.getSecurityGroupId().getValue(),
  68                 neutronRule.getTenantId().getValue()));
  69         Neutron neutron = new NeutronBuilder()
  70             .setSecurityGroups(new SecurityGroupsBuilder().setSecurityGroup(secGroups).build()).build();
  71         neutronSecurityRuleAware.onCreated(neutronRule, neutron);
  72
  73         //read security rule
  74         PolicyAssert.assertContractExists(dataProvider, RULE_TENANT_ID, RULE_ID);
  75
  76         //compare
  77         Contract readContract = ConfigDataStoreReader.readContract(dataProvider, RULE_TENANT_ID, RULE_ID).get();
  78         assertNotNull(readContract);
  79         assertEquals(readContract.getId().getValue(), RULE_ID);
  80
  81         //delete rule
  82         neutronSecurityRuleAware.onDeleted(neutronRule, neutron, null);
  83         PolicyAssert.assertContractNotExists(dataProvider, RULE_TENANT_ID, RULE_ID);
  84     }
  85
  86     // create neutron security rule
  87     private SecurityRule buildNeutronSecurityRule() {
  88         SecurityRuleAttributes.Protocol protocolTcp = new SecurityRuleAttributes.Protocol(ProtocolTcp.class);
  89         return new SecurityRuleBuilder().setUuid(new Uuid(RULE_ID))
  90             .setTenantId(new Uuid(RULE_TENANT_ID))
  91             .setSecurityGroupId(new Uuid(RULE_GROUP_ID))
  92             .setRemoteIpPrefix(new IpPrefix(new Ipv4Prefix("192.0.0.1/24")))
  93             .setPortRangeMin(1000)
  94             .setPortRangeMax(5000)
  95             .setProtocol(protocolTcp)
  96             .setEthertype(EthertypeV4.class)
  97             .setDirection(DirectionIngress.class)
  98             .build();
  99     }
 100
 101 }