If the connection between the controller and compute is closed reopen it

This commit is contained in:
Julien Duponchelle committed 2016-07-20 12:43:23 +02:00
1 parent 73d5066392
commit 7c4c03cf17
2 files changed
+19 -12

No files matched your search

+4 -4
View File
@@ -185,8 +185,8 @@ def test_connectNotification(compute, async_run):
return response
compute._controller._notification = MagicMock()
compute._session = AsyncioMagicMock(return_value=ws_mock)
compute._session.ws_connect = AsyncioMagicMock(return_value=ws_mock)
compute._http_session = AsyncioMagicMock(return_value=ws_mock)
compute._http_session.ws_connect = AsyncioMagicMock(return_value=ws_mock)
ws_mock.receive = receive
async_run(compute._connect_notification())
@@ -218,8 +218,8 @@ def test_connectNotificationPing(compute, async_run):
return response
compute._controller._notification = MagicMock()
compute._session = AsyncioMagicMock(return_value=ws_mock)
compute._session.ws_connect = AsyncioMagicMock(return_value=ws_mock)
compute._http_session = AsyncioMagicMock(return_value=ws_mock)
compute._http_session.ws_connect = AsyncioMagicMock(return_value=ws_mock)
ws_mock.receive = receive
async_run(compute._connect_notification())