lint
This commit is contained in:
parent
daf9f69b62
commit
2569eac3a0
|
@ -9,6 +9,5 @@ class RawConnection(IRawConnection):
|
|||
self.reader = reader
|
||||
self.writer = writer
|
||||
|
||||
|
||||
def close(self):
|
||||
self.writer.close()
|
||||
|
|
|
@ -1,29 +1,27 @@
|
|||
import pytest
|
||||
import asyncio
|
||||
import pytest
|
||||
|
||||
from network.connection.raw_connection import RawConnection
|
||||
# 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():
|
||||
async def test_simple_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)
|
||||
|
||||
# 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…
Reference in New Issue
Block a user