def test_set_line(self):
userfeatures = UserFeatures(self._agi, self._cursor, self._args)
mocked_line = Mock()
mocked_line.number = '1001'
userfeatures._set_line()
self.assertEqual(userfeatures._line, None)
userfeatures._dstid = self._variables['XIVO_DSTID']
with patch('xivo_agid.objects.Line') as lines_cls:
lines_cls.return_value = mocked_line
userfeatures._set_line()
lines_cls.assert_called_with(int(self._variables['XIVO_DSTID']))
self.assertEqual(mocked_line, userfeatures._line)
def test_set_user(self):
userfeatures = UserFeatures(self._agi, self._cursor, self._args)
userfeatures._set_xivo_user_name = 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.assertTrue(userfeatures._user is not None)
self.assertTrue(isinstance(userfeatures._user, objects.User))
