bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/FromExternalImportPolicyTest.java

   1 package org.opendaylight.protocol.bgp.rib.impl;
   2
   3 import static org.junit.Assert.assertEquals;
   4
   5 import org.junit.Test;
   6 import org.opendaylight.yangtools.yang.common.QName;
   7 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
   8 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
   9 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  10 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  11 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
  12 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  13 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  14 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeAttrBuilder;
  15 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
  16 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeSchemaAwareBuilder;
  17 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafNodeBuilder;
  18 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetEntryNodeBuilder;
  19 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetNodeBuilder;
  20 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableUnkeyedListNodeBuilder;
  21
  22 /**
  23  * Created by cgasparini on 10.6.2015.
  24  */
  25 public class FromExternalImportPolicyTest {
  26
  27     private static final QName DATA_QNAME = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "attributes"));
  28     private static final QName LOCALPREF = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "local-pref"));
  29     private static final QName CLUSTERID = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "cluster-id"));
  30     private static final QName CLUSTER = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "cluster"));
  31     private static final QName ASPATH = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "as-path"));
  32     private static final QName SEGMENT = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "segments"));
  33     private static final QName UNRECOGNIZED = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "unrecognized-attributes"));
  34     private static final QName NEXTHOP = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "c-next-hop"));
  35     private static final QName IPV4NH = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "ipv4-next-hop"));
  36     private static final QName ORIGINATOR = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "originator-id"));
  37     private static final QName MED = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "multi-exit-disc"));
  38     private static final QName ORIGIN = QName.cachedReference(QName.create("urn:opendaylight:params:xml:ns:yang:bgp-inet", "2013-09-19", "origin"));
  39
  40     @Test
  41     public void testEffectiveAttributes() {
  42         DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContBuilder = createContBuilder(this.DATA_QNAME);
  43         // local pref
  44         dataContBuilder.addChild(createContBuilder(LOCALPREF).addChild(createValueBuilder(100L, LOCALPREF, "pref").build()).build());
  45
  46         // cluster pref
  47         String s = "404.40.40.40";
  48         LeafSetEntryNode<Object> entry1 = ImmutableLeafSetEntryNodeBuilder.create().withNodeIdentifier(
  49             new NodeWithValue(CLUSTER, s)).withValue(s).build();
  50
  51         dataContBuilder.addChild(createContBuilder(CLUSTERID).addChild(ImmutableLeafSetNodeBuilder.create().withNodeIdentifier(
  52             new NodeIdentifier(QName.create(CLUSTER, "cluster"))).withChild(entry1).build()).build());
  53
  54         // as-path pref
  55         final ContainerNode asPath = createContBuilder(ASPATH).addChild(ImmutableUnkeyedListNodeBuilder.create()
  56             .withNodeIdentifier(new NodeIdentifier(SEGMENT)).build()).build();
  57         dataContBuilder.addChild(asPath);
  58
  59         // unrecognized
  60         dataContBuilder.addChild(ImmutableNodes.mapNodeBuilder(UNRECOGNIZED).build());
  61
  62         // c-next-hop pref
  63         final DataContainerNodeBuilder<NodeIdentifier, ChoiceNode> nextHop = Builders.choiceBuilder();
  64         nextHop.withNodeIdentifier(new NodeIdentifier(NEXTHOP));
  65         final ContainerNode cNextHop = createContBuilder(IPV4NH).addChild(createValueBuilder("199.20.160.41", IPV4NH, "global").build()).build();
  66         final ChoiceNode resultNextHop = nextHop.addChild(cNextHop).build();
  67         dataContBuilder.addChild(resultNextHop);
  68
  69         // originator pref
  70         dataContBuilder.addChild(createContBuilder(ORIGINATOR).addChild(createValueBuilder("41.41.41.41", ORIGINATOR, "originator").build()).build());
  71
  72         // origin pref
  73         final ContainerNode origin = createContBuilder(ORIGIN).addChild(createValueBuilder("igp", ORIGIN, "value").build()).build();
  74         dataContBuilder.addChild(origin);
  75
  76         // multi-exit-disc pref
  77         dataContBuilder.addChild(createContBuilder(MED).addChild(createValueBuilder("0", MED, "med").build()).build());
  78         FromExternalImportPolicy importPol = new FromExternalImportPolicy();
  79         final ContainerNode result = importPol.effectiveAttributes(dataContBuilder.build());
  80
  81         DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> dataContExpected = createContBuilder(this.DATA_QNAME);
  82
  83         dataContExpected.addChild(asPath);
  84         dataContExpected.addChild(resultNextHop);
  85         dataContExpected.addChild(origin);
  86
  87         assertEquals(dataContExpected.build(), result);
  88     }
  89
  90     private DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> createContBuilder(final QName qname) {
  91         return ImmutableContainerNodeSchemaAwareBuilder.create().withNodeIdentifier(new NodeIdentifier(qname));
  92     }
  93
  94     private <T> ImmutableLeafNodeBuilder<T> createValueBuilder(final T value, final QName qname, final String localName) {
  95         final ImmutableLeafNodeBuilder<T> valueBuilder = new ImmutableLeafNodeBuilder<>();
  96         valueBuilder.withNodeIdentifier(new NodeIdentifier(QName.create(qname, localName))).withValue(value);
  97         return valueBuilder;
  98     }
  99
 100 }