arno/gns3-server
1
0
Fork 0
mirror of https://github.com/GNS3/gns3-server synced 2024-11-28 11:18:11 +00:00
Code Issues Releases Wiki Activity

Fix Docker console test.

Browse Source
This commit is contained in:
grossmj 2018-11-30 18:59:02 +08:00
parent bec2270334
commit 58b2ee152d
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gns3server/compute/docker/docker_vm.py
View File

@ -671,7 +671,7 @@ class DockerVM(BaseNode):
while True:
msg = await ws.receive()
if msg.type == aiohttp.WSMsgType.text:
if msg.type == aiohttp.WSMsgType.TEXT:
out.feed_data(msg.data.encode())
elif msg.type == aiohttp.WSMsgType.BINARY:
out.feed_data(msg.data)

6
tests/compute/docker/test_docker_vm.py
View File

@ -1037,11 +1037,11 @@ def test_read_console_output_with_binary_mode(vm, loop):
async def receive(self):
if not self.sent:
self.sent = True
return MagicMock(tp=aiohttp.WSMsgType.BINARY, data=b"test")
return MagicMock(type=aiohttp.WSMsgType.BINARY, data=b"test")
else:
return MagicMock(tp=aiohttp.WSMsgType.CLOSE)
return MagicMock(type=aiohttp.WSMsgType.CLOSE)
def close(self):
async def close(self):
pass
input_stream = InputStreamMock()