Refactor SwitchConnectionProvider a bit

[openflowplugin.git] / openflowplugin-impl / src / test / java / org / opendaylight / openflowplugin / impl / OpenFlowPluginProviderImplTest.java

diff --git a/openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/OpenFlowPluginProviderImplTest.java b/openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/OpenFlowPluginProviderImplTest.java
index 9480822833d06a9d39026b669edf9eab528a39db..6c7bae1855ab93f7765bf1d69ab549af60d0e2b7 100644 (file)
--- a/openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/OpenFlowPluginProviderImplTest.java
+++ b/openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/OpenFlowPluginProviderImplTest.java
@@ -77,7 +77,7 @@ public class OpenFlowPluginProviderImplTest {
         when(dataBroker.newWriteOnlyTransaction()).thenReturn(writeTransaction);
         doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
         when(entityOwnershipService.registerListener(any(), any())).thenReturn(entityOwnershipListenerRegistration);
-        when(switchConnectionProvider.startup()).thenReturn(Futures.immediateFuture(null));
+        when(switchConnectionProvider.startup(any())).thenReturn(Futures.immediateFuture(null));
         when(switchConnectionProvider.shutdown()).thenReturn(Futures.immediateFuture(true));
         when(configurationService.getProperty(eq(ConfigurationProperty.USE_SINGLE_LAYER_SERIALIZATION.toString()),
                 any())).thenReturn(USE_SINGLE_LAYER_SERIALIZATION);
@@ -102,7 +102,7 @@ public class OpenFlowPluginProviderImplTest {
                 ofPluginDiagstatusProvider, systemReadyMonitor)) {
             // Calling the onSystemBootReady() callback
             provider.onSystemBootReady();
-            verify(switchConnectionProvider).startup();
+            verify(switchConnectionProvider).startup(any());
         }
         verify(switchConnectionProvider).shutdown();
     }