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