d93a4bf1cf1bcd82820426b17179bbc6f5c65bbd
[hc2vpp.git] /
1 /*
2  * Copyright (c) 2016 Cisco and/or its affiliates.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at:
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package io.fd.hc2vpp.routing.write.factory;
18
19 import io.fd.hc2vpp.common.test.util.NamingContextHelper;
20 import io.fd.hc2vpp.common.translate.util.NamingContext;
21 import io.fd.hc2vpp.routing.Ipv4RouteData;
22 import io.fd.hc2vpp.routing.helpers.ClassifyTableTestHelper;
23 import io.fd.hc2vpp.routing.helpers.RoutingRequestTestHelper;
24 import io.fd.hc2vpp.routing.helpers.SchemaContextTestHelper;
25 import io.fd.hc2vpp.routing.write.trait.RouteRequestProducer;
26 import io.fd.hc2vpp.v3po.vppclassifier.VppClassifierContextManager;
27 import io.fd.honeycomb.test.tools.HoneycombTestRunner;
28 import io.fd.honeycomb.test.tools.annotations.InjectTestData;
29 import io.fd.honeycomb.translate.MappingContext;
30 import io.fd.honeycomb.translate.util.RWUtils;
31 import io.fd.vpp.jvpp.core.dto.IpAddDelRoute;
32 import org.junit.Before;
33 import org.junit.Test;
34 import org.junit.runner.RunWith;
35 import org.mockito.Mock;
36 import org.mockito.MockitoAnnotations;
37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ipv4.unicast.routing.rev140524.StaticRoutes1;
38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ipv4.unicast.routing.rev140524.routing.routing.instance.routing.protocols.routing.protocol._static.routes.ipv4.Route;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev140524.SpecialNextHopGrouping;
40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev140524.routing.routing.instance.routing.protocols.routing.protocol.StaticRoutes;
41
42 import static io.fd.hc2vpp.routing.write.factory.SpecialNextHopRequestFactory.forContexts;
43 import static org.junit.Assert.assertEquals;
44 import static org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev140524.SpecialNextHopGrouping.SpecialNextHop.*;
45
46 @RunWith(HoneycombTestRunner.class)
47 public class SpecialNextHopRequestFactoryIpv4Test
48         implements RouteRequestProducer, RoutingRequestTestHelper, ClassifyTableTestHelper,
49         SchemaContextTestHelper,NamingContextHelper {
50
51     private static final String PARENT_PROTOCOL_4 = "parent-protocol-4";
52     private static final int PARENT_PROTOCOL_4_INDEX = 4;
53
54     @Mock
55     private VppClassifierContextManager classifierContextManager;
56
57     @Mock
58     private MappingContext mappingContext;
59
60     private NamingContext interfaceContext;
61     private NamingContext routingProtocolContextContext;
62
63     private SpecialNextHopRequestFactory factory;
64
65     @Before
66     public void init() {
67         MockitoAnnotations.initMocks(this);
68
69         interfaceContext = new NamingContext("iface", "interface-context");
70         routingProtocolContextContext = new NamingContext("routingProtocol", "routing-protocol-context");
71         factory = forContexts(classifierContextManager, interfaceContext, routingProtocolContextContext);
72
73         addMapping(classifierContextManager, CLASSIFY_TABLE_NAME, CLASSIFY_TABLE_INDEX, mappingContext);
74         defineMapping(mappingContext, PARENT_PROTOCOL_4, PARENT_PROTOCOL_4_INDEX, "routing-protocol-context");
75     }
76
77     @Test
78     public void testIpv4WithClassifierBlackhole(
79             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteBlackhole.json", id = STATIC_ROUTE_PATH)
80                     StaticRoutes routes) {
81         final IpAddDelRoute request =
82                 factory.createIpv4SpecialHopRequest(true, PARENT_PROTOCOL_4, extractSingleRoute(routes, 1L), mappingContext,
83                         SpecialNextHopGrouping.SpecialNextHop.Blackhole);
84
85         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, 24, 1, 0, 0, 0, PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI), request);
86     }
87
88     @Test
89     public void testIpv4WithClassifierReceive(
90             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteReceive.json", id = STATIC_ROUTE_PATH)
91                     StaticRoutes routes) {
92         final IpAddDelRoute request =
93                 factory.createIpv4SpecialHopRequest(true, PARENT_PROTOCOL_4, extractSingleRoute(routes, 1L), mappingContext, Receive);
94
95         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, 24, 0, 1, 0, 0, PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI), request);
96     }
97
98     @Test
99     public void testIpv4WithClassifierUnreach(
100             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteUnreachable.json", id = STATIC_ROUTE_PATH)
101                     StaticRoutes routes) {
102         final IpAddDelRoute request =
103                 factory.createIpv4SpecialHopRequest(true, PARENT_PROTOCOL_4, extractSingleRoute(routes, 1L), mappingContext, Unreachable);
104
105         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, 24, 0, 0, 1, 0, PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI), request);
106     }
107
108     @Test
109     public void testIpv4WithClassifierProhibited(
110             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteProhibited.json", id = STATIC_ROUTE_PATH)
111                     StaticRoutes routes) {
112         final IpAddDelRoute request =
113                 factory.createIpv4SpecialHopRequest(true,PARENT_PROTOCOL_4,  extractSingleRoute(routes, 1L), mappingContext, Prohibit);
114
115         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, 24, 0, 0, 0, 1, PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI), request);
116     }
117
118     private Route extractSingleRoute(final StaticRoutes staticRoutes, final long id) {
119         return staticRoutes.getAugmentation(StaticRoutes1.class).getIpv4().getRoute().stream()
120                 .filter(route -> route.getId() == id)
121                 .collect(RWUtils.singleItemCollector());
122     }
123 }