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
