Bug 4827 - BGP add-path unit tests

[bgpcep.git] / bgp / rib-impl / src / main / java / org / opendaylight / protocol / bgp / rib / impl / ToInternalExportPolicy.java

diff --git a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/ToInternalExportPolicy.java b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/ToInternalExportPolicy.java
index 07e982dcd287c1740bdc04a7759ff8dccfb9a26b..3de72121f833eb73007a3861a6e47ac756a588bf 100644 (file)
--- a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/ToInternalExportPolicy.java
+++ b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/ToInternalExportPolicy.java
@@ -7,7 +7,7 @@
  */
 package org.opendaylight.protocol.bgp.rib.impl;
 
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ClusterIdentifier;
 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
@@ -32,6 +32,9 @@ final class ToInternalExportPolicy extends AbstractReflectingExportPolicy {
         case RrClient:
             // Client iBGP -> Non-Client iBGP, reflect
             return reflectedAttributes(attributes);
+        case Internal:
+            // Internal iBGP -> Non-Client iBGP, reflect
+            return reflectedFromInternalAttributes(attributes);
         default:
             throw new IllegalArgumentException("Unhandled source role " + sourceRole);
         }