class AMITestCase(unittest.TestCase): def setUp(self): self.tunnel = SIOWC() self.ami = AMI() self.ami.makeConnection(FileWrapper(self.tunnel)) def getCommandDF(self, command, args, kw): #prepare command if hasattr(self.ami, method): df = getattr(self.ami, method)(*args, **kwargs) # call lineReceived return df else: self.fail('Unknown protocol method: %r' % method) def assertResponse(self, cmd, *args, **kw): result = NotImplemented return self.getCommandDF(cmd, args, kw ).addCallback(self.assertEqual, result) def assertException(self, cmd, result_line, forErrors=(AMIFailure,), *args, **kw): def onError(failure, forErrors): if failure.trap(*forErrors): pass else: self.fail('expected: %s' % forErrors) return self.getCommandDF(cmd, args, kw ).addErrback(onError, forErrors)
def setUp(self): self.tunnel = SIOWC() self.ami = AMI() self.ami.makeConnection(FileWrapper(self.tunnel))