Fix context chain closing
[openflowplugin.git] / openflowplugin-impl / src / test / java / org / opendaylight / openflowplugin / impl / device / DeviceManagerImplTest.java
index 2b2a50b8af8e56612b8663af252258829de70c16..431e7b2f1c279c04aa08f4792cf0cde93d7974be 100644 (file)
@@ -23,7 +23,6 @@ import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mock;
-import org.mockito.Mockito;
 import org.mockito.runners.MockitoJUnitRunner;
 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
@@ -124,8 +123,7 @@ public class DeviceManagerImplTest {
 
         deviceManager.close();
 
-        verify(deviceContext).shutdownConnection();
-        verify(deviceContext, Mockito.never()).close();
+        verify(deviceContext).close();
     }
 
     private static ConcurrentHashMap<DeviceInfo, DeviceContext> getContextsCollection(final DeviceManagerImpl deviceManager) throws NoSuchFieldException, IllegalAccessException {