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

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