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):