def test_set_xivo_redirecting_info_full_callerid(self): userfeatures = UserFeatures(self._agi, self._cursor, self._args) userfeatures._user = Mock() userfeatures._user.callerid = '"Foobar" <123>' userfeatures._dstnum = '42' userfeatures._set_xivo_redirecting_info() expected_calls = [ call('XIVO_DST_REDIRECTING_NAME', 'Foobar'), call('XIVO_DST_REDIRECTING_NUM', '123'), ] self.assertEqual(self._agi.set_variable.call_args_list, expected_calls)
def test_set_xivo_redirecting_info_no_callerid(self): userfeatures = UserFeatures(self._agi, self._cursor, self._args) userfeatures._user = Mock() userfeatures._user.firstname = 'First' userfeatures._user.lastname = 'Last' userfeatures._user.callerid = '' userfeatures._dstnum = '42' userfeatures._set_xivo_redirecting_info() expected_calls = [ call('XIVO_DST_REDIRECTING_NAME', 'First Last'), call('XIVO_DST_REDIRECTING_NUM', '42'), ] self.assertEqual(self._agi.set_variable.call_args_list, expected_calls)
def test_set_user(self): userfeatures = UserFeatures(self._agi, self._cursor, self._args) userfeatures._set_xivo_user_name = Mock() userfeatures._set_xivo_redirecting_info = Mock() userfeatures._set_user() self.assertTrue(userfeatures._user is None) self.assertEqual(userfeatures._set_xivo_user_name.call_count, 0) userfeatures._dstid = self._variables['XIVO_DSTID'] with patch.object(objects.User, '__init__') as user_init: user_init.return_value = None userfeatures._set_user() self.assertEqual(userfeatures._set_xivo_user_name.call_count, 1) self.assertEqual(userfeatures._set_xivo_redirecting_info.call_count, 1) self.assertTrue(userfeatures._user is not None) self.assertTrue(isinstance(userfeatures._user, objects.User))