diff --git a/src/wormhole_transit_relay/test/common.py b/src/wormhole_transit_relay/test/common.py index 961e680..a0b0c28 100644 --- a/src/wormhole_transit_relay/test/common.py +++ b/src/wormhole_transit_relay/test/common.py @@ -83,9 +83,11 @@ class ServerBase: def connectionMade(self): self.connected = True + return Protocol.connectionMade(self) def connectionLost(self, reason): self.connected = False + return Protocol.connectionLost(self, reason) def send(self, data): self.transport.write(data)