def test_checkGroupCount(self): self.agi.env = ENV max_group_count = 2 real_group_count = 2 def getResult(result): self.assertEqual(max_group_count, result) df = checkGroupCount(self.agi, max_group_count) df.addCallback(getResult) tell = self.agi.pendingMessages.pop(0) tell.callback(COMMANDS["SET VARIABLE"][SUCCESS]) tell = self.agi.pendingMessages.pop(0) tell.callback(COMMANDS["GET FULL VARIABLE"][SUCCESS] % real_group_count) return df
def test_checkGroupCountFailure(self): self.agi.env = ENV max_group_count = 2 real_group_count = 2 def getError(reason): if reason.trap(MaxGroupCount): pass else: return defer.fail("MaxGroupCount exception is required") df = checkGroupCount(self.agi, max_group_count) df.addErrback(getError) tell = self.agi.pendingMessages.pop(0) tell.callback(COMMANDS["SET VARIABLE"][SUCCESS]) tell = self.agi.pendingMessages.pop(0) tell.callback(COMMANDS["GET FULL VARIABLE"][SUCCESS] % real_group_count) return df