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()