d7db714562
* change get_restricted_data and get_projector_data to always use a list * Add typings to all get_restricted_data and get_projector_data methods * Replace CollectionElementList with a real list * Fixed arguments of inform_deleted_data * Moved CollectionElementCache to cache.py and refactored it * Run tests with cache enabled (using fakeredis)
99 lines
4.2 KiB
Python
99 lines
4.2 KiB
Python
from unittest import TestCase
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from openslides.core.models import Projector
|
|
from openslides.utils import collection
|
|
|
|
|
|
class TestGetModelFromCollectionString(TestCase):
|
|
def test_known_app(self):
|
|
projector_model = collection.get_model_from_collection_string('core/projector')
|
|
|
|
self.assertEqual(projector_model, Projector)
|
|
|
|
def test_unknown_app(self):
|
|
with self.assertRaises(ValueError):
|
|
collection.get_model_from_collection_string('invalid/model')
|
|
|
|
|
|
class TestCollectionElement(TestCase):
|
|
def test_from_values(self):
|
|
with patch.object(collection.CollectionElement, 'get_full_data'):
|
|
collection_element = collection.CollectionElement.from_values('testmodule/model', 42)
|
|
|
|
self.assertEqual(collection_element.collection_string, 'testmodule/model')
|
|
self.assertEqual(collection_element.id, 42)
|
|
|
|
def test_channel_message(self):
|
|
"""
|
|
Test that to_channel_message works together with from_channel_message.
|
|
"""
|
|
collection_element = collection.CollectionElement.from_values(
|
|
'testmodule/model',
|
|
42,
|
|
full_data={'data': 'value'},
|
|
information={'some': 'information'})
|
|
|
|
created_collection_element = collection.from_channel_message(
|
|
collection.to_channel_message([collection_element]))[0]
|
|
|
|
self.assertEqual(
|
|
collection_element,
|
|
created_collection_element)
|
|
self.assertEqual(created_collection_element.full_data, {'data': 'value'})
|
|
self.assertEqual(created_collection_element.information, {'some': 'information'})
|
|
|
|
def test_as_autoupdate_for_user(self):
|
|
with patch.object(collection.CollectionElement, 'get_full_data'):
|
|
collection_element = collection.CollectionElement.from_values('testmodule/model', 42)
|
|
fake_user = MagicMock()
|
|
collection_element.get_access_permissions = MagicMock()
|
|
collection_element.get_access_permissions().get_restricted_data.return_value = ['restricted_data']
|
|
collection_element.get_full_data = MagicMock()
|
|
|
|
self.assertEqual(
|
|
collection_element.as_autoupdate_for_user(fake_user),
|
|
{'collection': 'testmodule/model',
|
|
'id': 42,
|
|
'action': 'changed',
|
|
'data': 'restricted_data'})
|
|
|
|
def test_as_autoupdate_for_user_no_permission(self):
|
|
with patch.object(collection.CollectionElement, 'get_full_data'):
|
|
collection_element = collection.CollectionElement.from_values('testmodule/model', 42)
|
|
fake_user = MagicMock()
|
|
collection_element.get_access_permissions = MagicMock()
|
|
collection_element.get_access_permissions().get_restricted_data.return_value = None
|
|
collection_element.get_full_data = MagicMock()
|
|
|
|
self.assertEqual(
|
|
collection_element.as_autoupdate_for_user(fake_user),
|
|
{'collection': 'testmodule/model',
|
|
'id': 42,
|
|
'action': 'deleted'})
|
|
|
|
def test_as_autoupdate_for_user_deleted(self):
|
|
collection_element = collection.CollectionElement.from_values('testmodule/model', 42, deleted=True)
|
|
fake_user = MagicMock()
|
|
|
|
self.assertEqual(
|
|
collection_element.as_autoupdate_for_user(fake_user),
|
|
{'collection': 'testmodule/model',
|
|
'id': 42,
|
|
'action': 'deleted'})
|
|
|
|
@patch.object(collection.CollectionElement, 'get_full_data')
|
|
def test_equal(self, mock_get_full_data):
|
|
self.assertEqual(
|
|
collection.CollectionElement.from_values('testmodule/model', 1),
|
|
collection.CollectionElement.from_values('testmodule/model', 1))
|
|
self.assertEqual(
|
|
collection.CollectionElement.from_values('testmodule/model', 1),
|
|
collection.CollectionElement.from_values('testmodule/model', 1, deleted=True))
|
|
self.assertNotEqual(
|
|
collection.CollectionElement.from_values('testmodule/model', 1),
|
|
collection.CollectionElement.from_values('testmodule/model', 2))
|
|
self.assertNotEqual(
|
|
collection.CollectionElement.from_values('testmodule/model', 1),
|
|
collection.CollectionElement.from_values('testmodule/other_model', 1))
|