def test_set_pickup_info(self): context = 'foo' number = '101' userfeatures = UserFeatures(self._agi, self._cursor, self._args) self.assertEqual(None, userfeatures._pickup_context) userfeatures._master_line = {'id': '2', 'number': number, 'context': context} userfeatures._dstnum = number userfeatures._set_pickup_info() self.assertEqual(context, userfeatures._pickup_context) self.assertEqual(number, userfeatures._pickup_exten)
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)