Fix server close tests

This commit is contained in:
Julien Duponchelle committed 2015-02-06 11:31:54 +01:00
1 parent 5c3969ae79
commit 571044b3e8
3 files changed
+7 -3

No files matched your search

+2 -1
View File
@@ -24,6 +24,7 @@ import shutil
from uuid import uuid4
from unittest.mock import patch
from tests.utils import asyncio_patch
from gns3server.modules.project import Project
from gns3server.modules.vpcs import VPCS, VPCSVM
@@ -183,7 +184,7 @@ def test_project_close(loop, manager):
project = Project()
vm = VPCSVM("test", "00010203-0405-0607-0809-0a0b0c0d0e0f", project, manager)
project.add_vm(vm)
with patch("gns3server.modules.vpcs.vpcs_vm.VPCSVM.close") as mock:
with asyncio_patch("gns3server.modules.vpcs.vpcs_vm.VPCSVM.close") as mock:
loop.run_until_complete(asyncio.async(project.close()))
assert mock.called