class FingerFactoryTests(unittest.TestCase): def setUp(self): self.factory = FingerFactory() self.proto = self.factory.buildProtocol(("127.0.0.1", 0)) self.tr = proto_helpers.StringTransport() self.proto.makeConnection(self.tr) def test_getUser(self): """Do we get the result for a valid user?""" expected = "Good Day!" d = self.factory.getUser("sparky") d.debug = True def onError(err): return err d.addErrback(onError) def writeResponse(message): self.assertEquals(message, expected) d.addCallback(writeResponse) return d def test_getUserBad(self): """Do we get the NSU message?""" expected = "No such user" d = self.factory.getUser("parky") d.addCallback(self.assertEquals, expected) return d
def setUp(self): self.factory = FingerFactory() self.proto = self.factory.buildProtocol(("127.0.0.1", 0)) self.tr = proto_helpers.StringTransport() self.proto.makeConnection(self.tr)