cleanup

src/wormhole_transit_relay/test/test_transit_server.py

@@ -396,6 +396,7 @@ class TransitWebSockets(_Transit, ServerBase, unittest.TestCase):
         # should be an immediate error)
         with self.assertRaises(Disconnected):
             p1.send(b"more message")
+            self.flush()
 
     def new_protocol(self):
         ws_factory = WebSocketServerFactory("ws://localhost:4002")

src/wormhole_transit_relay/transit_server.py

@@ -261,11 +261,7 @@ class WebSocketTransitConnection(WebSocketServerProtocol):
             if token is None:
                 self._state.bad_token()
         else:
-            try:
-                self._state.got_bytes(payload)
-            except Exception as e:
-                log.err("Failed to send to partner: {}".format(e))
-                self.sendClose(3000, "send to partner failed")
+            self._state.got_bytes(payload)
 
     def onClose(self, wasClean, code, reason):
         """