This commit is contained in:
Julien Duponchelle 2016-03-17 15:16:09 +01:00
parent b55969d381
commit 76a0120d3e
No known key found for this signature in database
GPG Key ID: F1E2485547D4595D
7 changed files with 8 additions and 12 deletions

View File

@ -38,4 +38,3 @@ class NotificationHandler:
notif = yield from queue.get_json(5)
ws.send_str(notif)
return ws

View File

@ -25,6 +25,7 @@ class NotificationQueue(asyncio.Queue):
"""
Queue returned by the notification manager.
"""
def __init__(self):
super().__init__()
self._first = True

View File

@ -68,6 +68,7 @@ class Query:
Return a websocket connected to the path
"""
self._session = aiohttp.ClientSession()
@asyncio.coroutine
def go_request(future):
response = yield from self._session.ws_connect(self.get_url(path))

View File

@ -164,7 +164,6 @@ def test_termination_callback_error(vm, tmpdir, async_run):
vm._termination_callback(1)
assert vm.status == "stopped"
async_run(queue.get(0)) #  Ping
(action, event, kwargs) = queue.get_nowait()

View File

@ -69,7 +69,6 @@ def test_queue_json_meta(async_run):
assert len(notifications._listeners) == 0
def test_queue_ping(async_run):
"""
If we don't send a message during a long time (0.5 seconds)

View File

@ -87,7 +87,6 @@ def test_start(loop, vm, async_run):
process = MagicMock()
process.returncode = None
with NotificationManager.instance().queue() as queue:
async_run(queue.get(0)) #  Ping
@ -147,7 +146,6 @@ def test_start_0_6_1(loop, vm):
assert vm.is_running()
def test_stop(loop, vm, async_run):
process = MagicMock()
@ -166,7 +164,6 @@ def test_stop(loop, vm, async_run):
loop.run_until_complete(asyncio.async(vm.start()))
assert vm.is_running()
with asyncio_patch("gns3server.utils.asyncio.wait_for_process_termination"):
loop.run_until_complete(asyncio.async(vm.stop()))
assert vm.is_running() is False