diff --git a/twistednoise.py b/twistednoise.py index e51fd3d..298e893 100644 --- a/twistednoise.py +++ b/twistednoise.py @@ -192,9 +192,12 @@ class TNServerTest(unittest.TestCase): # Feed it into the protocol self.proto.dataReceived(encmsg) + # wait to pass it through d = yield task.deferLater(reactor, .1, bool, 1) + # fetch it out clientend = self.protos[0] self.assertEqual(clientend.data, ptmsg) + # clean up connection clientend.transport.loseConnection()