From f08aa339b40b8c6ed7600c52dd7c8422c34c1241 Mon Sep 17 00:00:00 2001 From: Alex Stokes Date: Sat, 24 Aug 2019 23:26:26 +0200 Subject: [PATCH] Verify the channel can pass some plaintext --- tests/security/test_secio.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/security/test_secio.py b/tests/security/test_secio.py index da7e1f16..f9fdad72 100644 --- a/tests/security/test_secio.py +++ b/tests/security/test_secio.py @@ -49,7 +49,7 @@ class InMemoryConnection(IRawConnection): self.current_msg = None return next_msg - def close(self) -> None: + async def close(self) -> None: self.closed = True @@ -88,14 +88,18 @@ async def test_create_secure_session(): 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() remote_pipe_task.cancel() await local_pipe_task await remote_pipe_task - assert local_secure_conn - assert remote_secure_conn - if __name__ == "__main__": asyncio.run(test_create_secure_session())