class TestGroupFeatures(unittest.TestCase):
def setUp(self):
self._agi = Mock()
self._cursor = Mock()
self._args = Mock()
self.group_features = GroupFeatures(self._agi, self._cursor, self._args)
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_members')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_options')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_vars')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_preprocess_subroutine')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_timeout')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_dial_action')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_schedule')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._needs_rewrite_cid')
@patch('xivo_agid.handlers.groupfeatures.GroupFeatures._set_rewrite_cid')
def test_execute(self,
_set_rewrite_cid,
_needs_rewrite_cid,
_set_schedule,
_set_dial_action,
_set_timeout,
_set_preprocess_subroutine,
_set_vars,
_set_options,
_set_members):
_needs_rewrite_cid.return_value = True
self.group_features.execute()
_set_members.assert_called_once_with()
_set_options.assert_called_once_with()
_set_vars.assert_called_once_with()
_set_preprocess_subroutine.assert_called_once_with()
_set_timeout.assert_called_once_with()
_set_dial_action.assert_called_once_with()
_set_schedule.assert_called_once_with()
_set_rewrite_cid.assert_called_once_with()
def test_referer_myself_needs_rewrite_cid(self):
self.group_features._id = 3
self.group_features._referer = "group:3"
self.assertTrue(self.group_features._needs_rewrite_cid())
def test_referer_voicemenu_needs_rewrite_cid(self):
self.group_features._referer = "voicemenu:778"
self.assertTrue(self.group_features._needs_rewrite_cid())
def test_set_schedule(self):
self.group_features._id = 34
self._agi.get_variable.return_value = ''
calls = [call('XIVO_PATH', 'group'), call('XIVO_PATH_ID', 34)]
self.group_features._set_schedule()
self._agi.set_variable.assert_has_calls(calls)
self._agi.set_variable.assert_any_call('XIVO_PATH', 'group')
self._agi.set_variable.assert_any_call('XIVO_PATH_ID', 34)
def setUp(self):
self._agi = Mock()
self._cursor = Mock()
self._args = Mock()
self.group_features = GroupFeatures(self._agi, self._cursor, self._args)
def incoming_group_set_features(agi, cursor, args):
groupfeatures_handler = GroupFeatures(agi, cursor, args)
groupfeatures_handler.execute()