diff options
Diffstat (limited to 'tests/test_sslproto.py')
| -rw-r--r-- | tests/test_sslproto.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/test_sslproto.py b/tests/test_sslproto.py index 053fefe..812dedb 100644 --- a/tests/test_sslproto.py +++ b/tests/test_sslproto.py @@ -33,6 +33,7 @@ class SslProtoHandshakeTests(test_utils.TestCase): waiter.cancel() transport = mock.Mock() sslpipe = mock.Mock() + sslpipe.shutdown.return_value = b'' sslpipe.do_handshake.side_effect = do_handshake with mock.patch('asyncio.sslproto._SSLPipe', return_value=sslpipe): ssl_proto.connection_made(transport) @@ -40,6 +41,9 @@ class SslProtoHandshakeTests(test_utils.TestCase): with test_utils.disable_logger(): self.loop.run_until_complete(handshake_fut) + # Close the transport + ssl_proto._app_transport.close() + if __name__ == '__main__': unittest.main() |