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

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

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

@ -262,7 +262,7 @@ def test_list_files(tmpdir, loop):
def test_emit(async_run):
with NotificationManager.instance().queue() as queue:
(action, event, context) = async_run(queue.get(0.5)) # Ping
(action, event, context) = async_run(queue.get(0.5)) #  Ping
project = Project(project_id=str(uuid4()))
project.emit("test", {})

View File

@ -87,9 +87,8 @@ def test_start(loop, vm, async_run):
process = MagicMock()
process.returncode = None
with NotificationManager.instance().queue() as queue:
async_run(queue.get(0)) # Ping
async_run(queue.get(0)) #  Ping
with asyncio_patch("gns3server.hypervisor.vpcs.vpcs_vm.VPCSVM._check_requirements", return_value=True):
with asyncio_patch("asyncio.create_subprocess_exec", return_value=process) as mock_exec:
@ -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
@ -176,8 +173,8 @@ def test_stop(loop, vm, async_run):
else:
process.terminate.assert_called_with()
async_run(queue.get(0)) # Ping
async_run(queue.get(0)) # Started
async_run(queue.get(0)) #  Ping
async_run(queue.get(0)) #  Started
(action, event, kwargs) = async_run(queue.get(0))
assert action == "vm.stopped"