1 package org.opendaylight.groupbasedpolicy.neutron.mapper.mapping.rule;
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;
8 import java.util.ArrayList;
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.GbpDataBrokerTest;
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.security.groups.attributes.SecurityGroupsBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.groups.attributes.security.groups.SecurityGroup;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.rules.attributes.security.rules.SecurityRule;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.secgroups.rev150712.security.rules.attributes.security.rules.SecurityRuleBuilder;
32 public class NeutronSecurityRuleAwareTest extends GbpDataBrokerTest {
34 private static final String RULE_ID = "00000000-0000-0000-0000-000000000001";
35 private static final String RULE_TENANT_ID = "00000000-0000-0000-0000-000000000002";
36 private static final String RULE_GROUP_ID = "00000000-0000-0000-0000-000000000003";
39 public final void testIsDirectionOpposite_InIn() {
40 assertFalse(NeutronSecurityRuleAware.isDirectionOpposite(Direction.In, Direction.In));
44 public final void testIsDirectionOpposite_OutOut() {
45 assertFalse(NeutronSecurityRuleAware.isDirectionOpposite(Direction.Out, Direction.Out));
49 public final void testIsDirectionOpposite_InOut() {
50 assertTrue(NeutronSecurityRuleAware.isDirectionOpposite(Direction.In, Direction.Out));
54 public final void testIsDirectionOpposite_OutIn() {
55 assertTrue(NeutronSecurityRuleAware.isDirectionOpposite(Direction.Out, Direction.In));
59 public void testNeutronSecurityRuleCreatedAndDeleted() throws Exception {
60 DataBroker dataProvider = getDataBroker();
61 NeutronSecurityRuleAware neutronSecurityRuleAware = new NeutronSecurityRuleAware(dataProvider);
63 //create security rule and put to DS
64 SecurityRule neutronRule = buildNeutronSecurityRule();
65 List<SecurityGroup> secGroups = new ArrayList<>();
66 secGroups.add(NeutronEntityFactory.securityGroup(neutronRule.getSecurityGroupId().getValue(),
67 neutronRule.getTenantId().getValue()));
68 Neutron neutron = new NeutronBuilder()
69 .setSecurityGroups(new SecurityGroupsBuilder().setSecurityGroup(secGroups).build()).build();
70 neutronSecurityRuleAware.onCreated(neutronRule, neutron);
73 PolicyAssert.assertContractExists(dataProvider, RULE_TENANT_ID, RULE_ID);
76 Contract readContract = ConfigDataStoreReader.readContract(dataProvider, RULE_TENANT_ID, RULE_ID).get();
77 assertNotNull(readContract);
78 assertEquals(readContract.getId().getValue(), RULE_ID);
81 neutronSecurityRuleAware.onDeleted(neutronRule, neutron, null);
82 PolicyAssert.assertContractNotExists(dataProvider, RULE_TENANT_ID, RULE_ID);
85 // create neutron security rule
86 private SecurityRule buildNeutronSecurityRule() {
87 return new SecurityRuleBuilder().setId(new Uuid(RULE_ID))
88 .setTenantId(new Uuid(RULE_TENANT_ID))
89 .setSecurityGroupId(new Uuid(RULE_GROUP_ID))
90 .setRemoteIpPrefix(new IpPrefix(new Ipv4Prefix("192.0.0.1/24")))
91 .setPortRangeMin(1000)
92 .setPortRangeMax(5000)
93 .setProtocol(ProtocolTcp.class)
94 .setEthertype(EthertypeV4.class)
95 .setDirection(DirectionIngress.class)