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_members(self):
userfeatures = UserFeatures(self._agi, self._cursor, self._args)
userfeatures._set_feature_list = Mock()
userfeatures._set_caller = Mock()
userfeatures._set_line = Mock()
userfeatures._set_user = Mock()
userfeatures._set_members()
objects.ExtenFeatures = Mock()
old_user, objects.User = objects.User, Mock()
self.assertEqual(userfeatures._userid, self._variables['XIVO_USERID'])
self.assertEqual(userfeatures._dstid, self._variables['XIVO_DSTID'])
self.assertEqual(userfeatures._zone, self._variables['XIVO_CALLORIGIN'])
self.assertEqual(userfeatures._srcnum, self._variables['XIVO_SRCNUM'])
self.assertEqual(userfeatures._dstnum, self._variables['XIVO_DSTNUM'])
self.assertTrue(userfeatures._set_feature_list.called)
self.assertTrue(userfeatures._set_caller.called)
self.assertTrue(userfeatures._set_line.called)
self.assertTrue(userfeatures._set_user.called)
objects.User, old_user = old_user, None
