def test_set_members(self): userfeatures = UserFeatures(self._agi, self._cursor, self._args) userfeatures._set_feature_list = Mock() userfeatures._set_caller = Mock() userfeatures._set_lines = 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._lineid, self._variables['XIVO_LINEID']) self.assertEqual(userfeatures._zone, self._variables['XIVO_CALLORIGIN']) self.assertEqual(userfeatures._bypass_filter, self._variables['XIVO_CALLFILTER_BYPASS']) 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_lines.called) self.assertTrue(userfeatures._set_user.called) objects.User, old_user = old_user, None
def test_set_lines(self): userfeatures = UserFeatures(self._agi, self._cursor, self._args) mocked_line = Mock() mocked_line.lines = [{'number': '1001'}] userfeatures._set_lines() self.assertEqual(userfeatures._lines, None) userfeatures._dstid = self._variables['XIVO_DSTID'] with patch('xivo_agid.objects.Lines') as lines_cls: lines_cls.return_value = mocked_line userfeatures._set_lines() lines_cls.assert_called_with(self._agi, self._cursor, int(self._variables['XIVO_DSTID'])) self.assertEqual(mocked_line, userfeatures._lines) self.assertEqual(mocked_line.lines[0], userfeatures._master_line)