def testErrors(self): p = ident.IdentClient() L = [] d = defer.Deferred() d.addErrback(L.append) p.queries.append((d, 123, 456)) p.lineReceived('123, 456 : ERROR : UNKNOWN-ERROR') self.failUnless(L[0].value.__class__ is ident.IdentError, "%s is the wrong exception" % (L[0], )) L = [] d = defer.Deferred() d.addErrback(L.append) p.queries.append((d, 234, 456)) p.lineReceived('234, 456 : ERROR : NO-USER') self.failUnless(L[0].value.__class__ is ident.NoUser, "%s is the wrong exception" % (L[0], )) L = [] d = defer.Deferred() d.addErrback(L.append) p.queries.append((d, 345, 567)) p.lineReceived('345, 567 : ERROR : INVALID-PORT') self.failUnless(L[0].value.__class__ is ident.InvalidPort, "%s is the wrong exception" % (L[0], )) L = [] d = defer.Deferred() d.addErrback(L.append) p.queries.append((d, 567, 789)) p.lineReceived('567, 789 : ERROR : HIDDEN-USER') self.failUnless(L[0].value.__class__ is ident.HiddenUser, "%s is the wrong exception" % (L[0], ))
def testLostConnection(self): p = ident.IdentClient() L = [] d = defer.Deferred() d.addErrback(L.append) p.queries.append((d, 765, 432)) p.connectionLost(failure.Failure(error.ConnectionLost())) self.failUnless(L[0].value.__class__ is ident.IdentError)
def setUp(self): """ Create a ident client used in tests. """ self.client = ident.IdentClient()