Bug-5023: Simplify client re-connection strategies

[bgpcep.git] / bgp / openconfig-impl / src / test / java / org / opendaylight / protocol / bgp / openconfig / impl / moduleconfig / BGPPeerProviderTest.java

diff --git a/bgp/openconfig-impl/src/test/java/org/opendaylight/protocol/bgp/openconfig/impl/moduleconfig/BGPPeerProviderTest.java b/bgp/openconfig-impl/src/test/java/org/opendaylight/protocol/bgp/openconfig/impl/moduleconfig/BGPPeerProviderTest.java
index e14968d43d8a0620aee682b697dc2a9b7871a5f1..9745e46aa86063f5e5a37139fbc3697c7358d813 100644 (file)
--- a/bgp/openconfig-impl/src/test/java/org/opendaylight/protocol/bgp/openconfig/impl/moduleconfig/BGPPeerProviderTest.java
+++ b/bgp/openconfig-impl/src/test/java/org/opendaylight/protocol/bgp/openconfig/impl/moduleconfig/BGPPeerProviderTest.java
@@ -56,6 +56,8 @@ public class BGPPeerProviderTest {
         final ModuleKey mKey = Mockito.mock(ModuleKey.class);
         Mockito.doReturn(mKey).when(this.neighborState).getModuleKey(this.neighbor.getKey());
         Mockito.doReturn("mKey").when(mKey).toString();
+        Mockito.doReturn("mKey").when(mKey).getName();
+        Mockito.doReturn(ModuleKey.class).when(mKey).getType();
         Mockito.doReturn(Boolean.TRUE).when(this.neighborState).remove(mKey, createNeighbor());
         this.peerProvider = new BGPPeerProvider(configHolders, moduleProvider, this.dataBroker);