2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.groupbasedpolicy.neutron.mapper.mapping.rule;
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;
16 import java.util.ArrayList;
17 import java.util.List;
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.NeutronEntityFactory;
25 import org.opendaylight.groupbasedpolicy.neutron.mapper.test.NeutronMapperDataBrokerTest;
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;
43 public class NeutronSecurityRuleAwareTest extends NeutronMapperDataBrokerTest {
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";
49 private EndpointRegistrator epRegistrator;
53 epRegistrator = getEpRegistrator();
57 public final void testIsDirectionOpposite_InIn() {
58 assertFalse(NeutronSecurityRuleAware.isDirectionOpposite(Direction.In, Direction.In));
62 public final void testIsDirectionOpposite_OutOut() {
63 assertFalse(NeutronSecurityRuleAware.isDirectionOpposite(Direction.Out, Direction.Out));
67 public final void testIsDirectionOpposite_InOut() {
68 assertTrue(NeutronSecurityRuleAware.isDirectionOpposite(Direction.In, Direction.Out));
72 public final void testIsDirectionOpposite_OutIn() {
73 assertTrue(NeutronSecurityRuleAware.isDirectionOpposite(Direction.Out, Direction.In));
77 public void testNeutronSecurityRuleCreatedAndDeleted() throws Exception {
78 DataBroker dataProvider = getDataBroker();
79 NeutronSecurityRuleAware neutronSecurityRuleAware = new NeutronSecurityRuleAware(dataProvider, epRegistrator);
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);
91 PolicyAssert.assertContractExists(dataProvider, RULE_TENANT_ID, RULE_ID);
94 Contract readContract = ConfigDataStoreReader.readContract(dataProvider, RULE_TENANT_ID, RULE_ID).get();
95 assertNotNull(readContract);
96 assertEquals(readContract.getId().getValue(), RULE_ID);
99 neutronSecurityRuleAware.onDeleted(neutronRule, neutron, null);
100 PolicyAssert.assertContractNotExists(dataProvider, RULE_TENANT_ID, RULE_ID);
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)