mirror of https://github.com/libp2p/py-libp2p.git
zixuanzh
6 years ago
2 changed files with 6 additions and 9 deletions
@ -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