BUG-6647 Increase code coverage and clean up II

[bgpcep.git] / bgp / bmp-mock / src / main / java / org / opendaylight / protocol / bmp / mock / BmpMockUtil.java

diff --git a/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockUtil.java b/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockUtil.java
index 254b91a9983d9801dccd1ec40a97015c3003330f..6ee8c512edd96c033fcad8b0307659c500add372 100755 (executable)
--- a/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockUtil.java
+++ b/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockUtil.java
@@ -94,7 +94,7 @@ final class BmpMockUtil {
     private static PeerHeader createPeerHeader(final Ipv4Address bgpId, final AdjRibInType ribType) {
         return new PeerHeaderBuilder()
             .setAddress(new IpAddress(bgpId))
-            .setAdjRibInType(AdjRibInType.PrePolicy)
+            .setAdjRibInType(ribType)
             .setAs(new AsNumber(ASN))
             .setBgpId(bgpId)
             .setIpv4(true)