Verify the channel can pass some plaintext
This commit is contained in:
parent
737195f461
commit
f08aa339b4
|
@ -49,7 +49,7 @@ class InMemoryConnection(IRawConnection):
|
||||||
self.current_msg = None
|
self.current_msg = None
|
||||||
return next_msg
|
return next_msg
|
||||||
|
|
||||||
def close(self) -> None:
|
async def close(self) -> None:
|
||||||
self.closed = True
|
self.closed = True
|
||||||
|
|
||||||
|
|
||||||
|
@ -88,14 +88,18 @@ async def test_create_secure_session():
|
||||||
local_session_builder, remote_session_builder
|
local_session_builder, remote_session_builder
|
||||||
)
|
)
|
||||||
|
|
||||||
|
msg = b"abc"
|
||||||
|
await local_secure_conn.write(msg)
|
||||||
|
received_msg = await remote_secure_conn.read()
|
||||||
|
assert received_msg == msg
|
||||||
|
|
||||||
|
await asyncio.gather(local_secure_conn.close(), remote_secure_conn.close())
|
||||||
|
|
||||||
local_pipe_task.cancel()
|
local_pipe_task.cancel()
|
||||||
remote_pipe_task.cancel()
|
remote_pipe_task.cancel()
|
||||||
await local_pipe_task
|
await local_pipe_task
|
||||||
await remote_pipe_task
|
await remote_pipe_task
|
||||||
|
|
||||||
assert local_secure_conn
|
|
||||||
assert remote_secure_conn
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
asyncio.run(test_create_secure_session())
|
asyncio.run(test_create_secure_session())
|
||||||
|
|
Loading…
Reference in New Issue
Block a user