Browse Source

raw_connection echo test

pull/66/head
zixuanzh 6 years ago
parent
commit
daf9f69b62
  1. 0
      tests/network/__init__.py
  2. 29
      tests/network/test_connection.py

0
tests/network/__init__.py

29
tests/network/test_connection.py

@ -0,0 +1,29 @@
import pytest
import asyncio
from network.connection.raw_connection import RawConnection
async def handle_echo(reader, writer):
data = await reader.read(100)
message = data.decode()
writer.write(data)
await writer.drain()
writer.close()
@pytest.mark.asyncio
async def test_echo():
server_ip = '127.0.0.1'
server_port = 8888
await asyncio.start_server(handle_echo, server_ip, server_port)
reader, writer = await asyncio.open_connection(server_ip, server_port)
raw_connection = RawConnection(server_ip, server_port, reader, writer)
test_message = "hello world"
writer.write(test_message.encode())
response = (await reader.read()).decode()
assert response == (test_message)
Loading…
Cancel
Save