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