Ejemplo n.º 1
0
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)
Ejemplo n.º 2
0
 def setUp(self):
     self.tunnel = SIOWC()
     self.ami = AMI()
     self.ami.makeConnection(FileWrapper(self.tunnel))