bgp/openconfig-rp-spi/src/test/java/org/opendaylight/protocol/bgp/openconfig/routing/policy/spi/AttributesEqualTests.java

   1 /*
   2  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.protocol.bgp.openconfig.routing.policy.spi;
   9
  10 import static org.junit.Assert.assertNotNull;
  11 import static org.junit.Assert.assertNull;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.registry.RouteAttributeContainer.routeAttributeContainerFalse;
  14
  15 import java.util.List;
  16 import org.junit.Before;
  17 import org.junit.Test;
  18 import org.mockito.Mock;
  19 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.registry.RouteAttributeContainer;
  20 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryExportParameters;
  21 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.routing.policy.rev151009.routing.policy.top.routing.policy.policy.definitions.policy.definition.statements.Statement;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Address;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.AttributesBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.LocalPrefBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.MultiExitDiscBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.OriginBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpOrigin;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv6NextHopCaseBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.ipv6.next.hop._case.Ipv6NextHopBuilder;
  33
  34 public class AttributesEqualTests extends AbstractStatementRegistryTest {
  35     @Mock
  36     private BGPRouteEntryExportParameters exportParameters;
  37     private List<Statement> basicStatements;
  38     @Mock
  39     private RouteEntryBaseAttributes baseAttributes;
  40
  41     @Before
  42     @Override
  43     public void setUp() throws Exception {
  44         super.setUp();
  45         this.basicStatements = loadStatement("basic-statements-test");
  46         doReturn(CLUSTER).when(this.baseAttributes).getClusterId();
  47         doReturn(LOCAL_AS).when(this.baseAttributes).getLocalAs();
  48         doReturn(IPV4).when(this.baseAttributes).getOriginatorId();
  49     }
  50
  51     @Test
  52     public void testMedEq() {
  53         Statement statement = this.basicStatements.stream()
  54                 .filter(st -> st.getName().equals("med-eq-test")).findFirst().get();
  55         RouteAttributeContainer attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
  56                 .setMultiExitDisc(new MultiExitDiscBuilder().setMed(200L).build())
  57                 .build());
  58
  59         RouteAttributeContainer result = this.statementRegistry.applyExportStatement(
  60                 this.baseAttributes,
  61                 this.exportParameters,
  62                 attributeContainer,
  63                 statement);
  64         assertNotNull(result.getAttributes());
  65
  66
  67         attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
  68                 .setMultiExitDisc(new MultiExitDiscBuilder().setMed(100L).build())
  69                 .build());
  70         result = this.statementRegistry.applyExportStatement(
  71                 this.baseAttributes,
  72                 this.exportParameters,
  73                 attributeContainer,
  74                 statement);
  75         assertNull(result.getAttributes());
  76     }
  77
  78     @Test
  79     public void testOriginEq() {
  80         Statement statement = this.basicStatements.stream()
  81                 .filter(st -> st.getName().equals("origin-eq-test")).findFirst().get();
  82         RouteAttributeContainer attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
  83                 .setOrigin(new OriginBuilder().setValue(BgpOrigin.Egp).build())
  84                 .build());
  85
  86         RouteAttributeContainer result = this.statementRegistry.applyExportStatement(
  87                 this.baseAttributes,
  88                 this.exportParameters,
  89                 attributeContainer,
  90                 statement);
  91         assertNotNull(result.getAttributes());
  92
  93
  94         attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
  95                 .setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build())
  96                 .build());
  97         result = this.statementRegistry.applyExportStatement(
  98                 this.baseAttributes,
  99                 this.exportParameters,
 100                 attributeContainer,
 101                 statement);
 102         assertNull(result.getAttributes());
 103     }
 104
 105     @Test
 106     public void testNextHopIn() {
 107         Statement statement = this.basicStatements.stream()
 108                 .filter(st -> st.getName().equals("nexthop-in-test")).findFirst().get();
 109         RouteAttributeContainer attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
 110                 .setCNextHop(new Ipv6NextHopCaseBuilder().setIpv6NextHop(new Ipv6NextHopBuilder()
 111                         .setGlobal(new Ipv6Address("2001:db8::1")).build()).build())
 112                 .build());
 113
 114         RouteAttributeContainer result = this.statementRegistry.applyExportStatement(
 115                 this.baseAttributes,
 116                 this.exportParameters,
 117                 attributeContainer,
 118                 statement);
 119         assertNotNull(result.getAttributes());
 120
 121
 122         attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
 123                 .setCNextHop(new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder()
 124                         .setGlobal(new Ipv4Address("42.42.42.42")).build()).build())
 125                 .build());
 126         result = this.statementRegistry.applyExportStatement(
 127                 this.baseAttributes,
 128                 this.exportParameters,
 129                 attributeContainer,
 130                 statement);
 131         assertNull(result.getAttributes());
 132     }
 133
 134     @Test
 135     public void testLocalPref() {
 136         Statement statement = this.basicStatements.stream()
 137                 .filter(st -> st.getName().equals("local-pref-eq-test")).findFirst().get();
 138         RouteAttributeContainer attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
 139                 .setLocalPref(new LocalPrefBuilder().setPref(350L).build())
 140                 .build());
 141
 142         RouteAttributeContainer result = this.statementRegistry.applyExportStatement(
 143                 this.baseAttributes,
 144                 this.exportParameters,
 145                 attributeContainer,
 146                 statement);
 147         assertNotNull(result.getAttributes());
 148
 149
 150         attributeContainer = routeAttributeContainerFalse(new AttributesBuilder()
 151                 .setLocalPref(new LocalPrefBuilder().setPref(100L).build())
 152                 .build());
 153         result = this.statementRegistry.applyExportStatement(
 154                 this.baseAttributes,
 155                 this.exportParameters,
 156                 attributeContainer,
 157                 statement);
 158         assertNull(result.getAttributes());
 159     }
 160 }