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