Fix Docker console test.

pull/1481/head
grossmj 6 years ago
parent bec2270334
commit 58b2ee152d

@ -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)

@ -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()

Loading…
Cancel
Save