varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 16:10:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
zixuanzh
2018-11-20 23:49:37 -05:00
parent daf9f69b62
commit 2569eac3a0
2 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
network/connection/raw_connection.py
View File

@ -9,6 +9,5 @@ class RawConnection(IRawConnection):
self.reader = reader self.reader = reader
self.writer = writer self.writer = writer
def close(self): def close(self):
self.writer.close() self.writer.close()

14
tests/network/test_connection.py
View File

@ -1,29 +1,27 @@
import pytest
import asyncio import asyncio
import pytest
from network.connection.raw_connection import RawConnection # from network.connection.raw_connection import RawConnection
async def handle_echo(reader, writer): async def handle_echo(reader, writer):
data = await reader.read(100) data = await reader.read(100)
message = data.decode()
writer.write(data) writer.write(data)
await writer.drain() await writer.drain()
writer.close() writer.close()
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_echo(): async def test_simple_echo():
server_ip = '127.0.0.1' server_ip = '127.0.0.1'
server_port = 8888 server_port = 8888
await asyncio.start_server(handle_echo, server_ip, server_port) await asyncio.start_server(handle_echo, server_ip, server_port)
reader, writer = await asyncio.open_connection(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" test_message = "hello world"
writer.write(test_message.encode()) writer.write(test_message.encode())
response = (await reader.read()).decode() response = (await reader.read()).decode()
assert response == (test_message) assert response == (test_message)