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)