gerrit.fd Code Review - hc2vpp.git/blob

   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 static io.fd.hc2vpp.routing.write.factory.SpecialNextHopRequestFactory.forContexts;
  20 import static org.junit.Assert.assertEquals;
  21 import static org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev180313.SpecialNextHop.SpecialNextHopEnum.Blackhole;
  22 import static org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev180313.SpecialNextHop.SpecialNextHopEnum.Prohibit;
  23 import static org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev180313.SpecialNextHop.SpecialNextHopEnum.Receive;
  24 import static org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev180313.SpecialNextHop.SpecialNextHopEnum.Unreachable;
  25
  26 import io.fd.hc2vpp.common.test.util.NamingContextHelper;
  27 import io.fd.hc2vpp.common.translate.util.NamingContext;
  28 import io.fd.hc2vpp.routing.Ipv4RouteData;
  29 import io.fd.hc2vpp.routing.helpers.ClassifyTableTestHelper;
  30 import io.fd.hc2vpp.routing.helpers.RoutingRequestTestHelper;
  31 import io.fd.hc2vpp.routing.helpers.SchemaContextTestHelper;
  32 import io.fd.hc2vpp.routing.write.trait.RouteRequestProducer;
  33 import io.fd.hc2vpp.vpp.classifier.context.VppClassifierContextManager;
  34 import io.fd.honeycomb.test.tools.HoneycombTestRunner;
  35 import io.fd.honeycomb.test.tools.annotations.InjectTestData;
  36 import io.fd.honeycomb.translate.MappingContext;
  37 import io.fd.honeycomb.translate.util.RWUtils;
  38 import io.fd.vpp.jvpp.core.dto.IpAddDelRoute;
  39 import org.junit.Before;
  40 import org.junit.Test;
  41 import org.junit.runner.RunWith;
  42 import org.mockito.Mock;
  43 import org.mockito.MockitoAnnotations;
  44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  45 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ipv4.unicast.routing.rev180313.StaticRoutes1;
  46 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ipv4.unicast.routing.rev180313.routing.control.plane.protocols.control.plane.protocol._static.routes.ipv4.Route;
  47 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.routing.rev180313.routing.control.plane.protocols.control.plane.protocol.StaticRoutes;
  48
  49 @RunWith(HoneycombTestRunner.class)
  50 public class SpecialNextHopRequestFactoryIpv4Test
  51         implements RouteRequestProducer, RoutingRequestTestHelper, ClassifyTableTestHelper,
  52         SchemaContextTestHelper,NamingContextHelper {
  53
  54     private static final String PARENT_PROTOCOL_4 = "parent-protocol-4";
  55     private static final int PARENT_PROTOCOL_4_INDEX = 4;
  56     public static final Ipv4Prefix IPV_4_PREFIX = new Ipv4Prefix("192.168.2.1/24");
  57     public static final int DST_PREFIX = 24;
  58
  59     @Mock
  60     private VppClassifierContextManager classifierContextManager;
  61
  62     @Mock
  63     private MappingContext mappingContext;
  64
  65     private NamingContext interfaceContext;
  66     private NamingContext routingProtocolContextContext;
  67
  68     private SpecialNextHopRequestFactory factory;
  69
  70     @Before
  71     public void init() {
  72         MockitoAnnotations.initMocks(this);
  73
  74         interfaceContext = new NamingContext("iface", "interface-context");
  75         routingProtocolContextContext = new NamingContext("routingProtocol", "routing-protocol-context");
  76         factory = forContexts(classifierContextManager, interfaceContext, routingProtocolContextContext);
  77
  78         addMapping(classifierContextManager, CLASSIFY_TABLE_NAME, CLASSIFY_TABLE_INDEX, mappingContext);
  79         defineMapping(mappingContext, PARENT_PROTOCOL_4, PARENT_PROTOCOL_4_INDEX, "routing-protocol-context");
  80     }
  81
  82     @Test
  83     public void testIpv4WithClassifierBlackhole(
  84             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteBlackhole.json", id = STATIC_ROUTE_PATH)
  85                 StaticRoutes routes) {
  86         final IpAddDelRoute request =
  87                 factory.createIpv4SpecialHopRequest(true, PARENT_PROTOCOL_4, extractSingleRoute(routes, IPV_4_PREFIX),
  88                                                     mappingContext, Blackhole);
  89
  90         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, DST_PREFIX, 1, 0, 0, 0,
  91                                           PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI),
  92                      request);
  93     }
  94
  95     @Test
  96     public void testIpv4WithClassifierReceive(
  97             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteReceive.json", id = STATIC_ROUTE_PATH)
  98                     StaticRoutes routes) {
  99         final IpAddDelRoute request =
 100                 factory.createIpv4SpecialHopRequest(true, PARENT_PROTOCOL_4, extractSingleRoute(routes, IPV_4_PREFIX),
 101                                                     mappingContext, Receive);
 102
 103         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, DST_PREFIX, 0, 1, 0, 0,
 104                                           PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI),
 105                      request);
 106     }
 107
 108     @Test
 109     public void testIpv4WithClassifierUnreach(
 110             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteUnreachable.json", id = STATIC_ROUTE_PATH)
 111                     StaticRoutes routes) {
 112         final IpAddDelRoute request =
 113                 factory.createIpv4SpecialHopRequest(true, PARENT_PROTOCOL_4, extractSingleRoute(routes, IPV_4_PREFIX),
 114                                                     mappingContext, Unreachable);
 115
 116         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, DST_PREFIX, 0, 0, 1, 0,
 117                                           PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI),
 118                      request);
 119     }
 120
 121     @Test
 122     public void testIpv4WithClassifierProhibited(
 123             @InjectTestData(resourcePath = "/ipv4/specialhop/specialHopRouteProhibited.json", id = STATIC_ROUTE_PATH)
 124                     StaticRoutes routes) {
 125         final IpAddDelRoute request =
 126                 factory.createIpv4SpecialHopRequest(true, PARENT_PROTOCOL_4, extractSingleRoute(routes, IPV_4_PREFIX),
 127                                                     mappingContext, Prohibit);
 128
 129         assertEquals(desiredSpecialResult(1, 0, Ipv4RouteData.FIRST_ADDRESS_AS_ARRAY, DST_PREFIX, 0, 0, 0, 1,
 130                                           PARENT_PROTOCOL_4_INDEX, DEFAULT_VNI),
 131                      request);
 132     }
 133
 134     private Route extractSingleRoute(final StaticRoutes staticRoutes, final Ipv4Prefix id) {
 135         return staticRoutes.augmentation(StaticRoutes1.class).getIpv4().getRoute().stream()
 136                 .filter(route -> route.getDestinationPrefix().getValue().equals(id.getValue()))
 137                 .collect(RWUtils.singleItemCollector());
 138     }
 139 }