From 0db8ed3225256f556eef6f1bd42183c1cccb5b1e Mon Sep 17 00:00:00 2001 From: meejah Date: Wed, 14 Apr 2021 15:11:38 -0600 Subject: [PATCH] even more explicit --- src/wormhole_transit_relay/test/test_transit_server.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/wormhole_transit_relay/test/test_transit_server.py b/src/wormhole_transit_relay/test/test_transit_server.py index 5937c3b..f05e3e4 100644 --- a/src/wormhole_transit_relay/test/test_transit_server.py +++ b/src/wormhole_transit_relay/test/test_transit_server.py @@ -34,9 +34,6 @@ def handshake(token, side=None): return hs class _Transit: - def new_protocol(self): - return self.new_protocol_tcp() - def count(self): return sum([ len(potentials) @@ -359,10 +356,16 @@ class _Transit: class TransitWithLogs(_Transit, ServerBase, unittest.TestCase): log_requests = True + def new_protocol(self): + return self.new_protocol_tcp() + class TransitWithoutLogs(_Transit, ServerBase, unittest.TestCase): log_requests = False + def new_protocol(self): + return self.new_protocol_tcp() + class TransitWebSockets(_Transit, ServerBase, unittest.TestCase):