From ef96af2a80497b9f21a81ca2a1735584c9adea2b Mon Sep 17 00:00:00 2001
From: meejah <meejah@meejah.ca>
Date: Sat, 10 Apr 2021 18:45:20 -0600
Subject: [PATCH] websocket tests already use the interface

---
 src/wormhole_transit_relay/test/test_transit_server.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/wormhole_transit_relay/test/test_transit_server.py b/src/wormhole_transit_relay/test/test_transit_server.py
index fee418a..826093d 100644
--- a/src/wormhole_transit_relay/test/test_transit_server.py
+++ b/src/wormhole_transit_relay/test/test_transit_server.py
@@ -12,7 +12,11 @@ from autobahn.twisted.testing import (
     MemoryReactorClockResolver,
 )
 from autobahn.exception import Disconnected
-from .common import ServerBase
+from zope.interface import implementer
+from .common import (
+    ServerBase,
+    IRelayTestClient,
+)
 from ..server_state import (
     MemoryUsageRecorder,
     blur_size,
@@ -400,6 +404,7 @@ class TransitWebSockets(_Transit, ServerBase, unittest.TestCase):
         ws_factory.transit = self._transit_server
         ws_protocol = ws_factory.buildProtocol(('127.0.0.1', 0))
 
+        @implementer(IRelayTestClient)
         class TransitWebSocketClientProtocol(WebSocketClientProtocol):
             _received = b""
             connected = False