def test_lazy_model_immutable_after_load(mock_plugin, model_data, model_type,
mock_entity_create_id):
from attr.exceptions import FrozenInstanceError
from coalaip.models import Model, LazyLoadableModel
mock_plugin.load.return_value = model_data
model = LazyLoadableModel(ld_type=model_type)
model.load(mock_entity_create_id, plugin=mock_plugin)
with raises(FrozenInstanceError):
model.loaded_model = Model(data={'other': 'other'}, ld_type='other_type')
def test_lazy_model_load_raises_on_model_validation(mock_plugin, work_jsonld,
right_jsonld_factory,
model_type,
mock_entity_create_id):
from coalaip.models import LazyLoadableModel
from coalaip.model_validators import is_creation_model
from coalaip.exceptions import ModelDataError
mock_plugin.load.return_value = right_jsonld_factory()
model = LazyLoadableModel(ld_type=model_type, validator=is_creation_model)
with raises(ModelDataError):
model.load(mock_entity_create_id, plugin=mock_plugin)
def test_lazy_model_load_raises_on_type_validation(mock_plugin, model_type,
bad_type_data,
mock_entity_create_id):
from coalaip.models import LazyLoadableModel
from coalaip.exceptions import ModelDataError
mock_plugin.load.return_value = bad_type_data
assert model_type != (bad_type_data.get('@type') or
bad_type_data.get('type'))
model = LazyLoadableModel(ld_type=model_type)
with raises(ModelDataError):
model.load(mock_entity_create_id, plugin=mock_plugin)
def test_lazy_model_data_immutable_after_load(mock_plugin, model_data,
model_type,
mock_entity_create_id):
from coalaip.models import LazyLoadableModel
mock_plugin.load.return_value = model_data
model = LazyLoadableModel(ld_type=model_type)
model.load(mock_entity_create_id, plugin=mock_plugin)
with raises(TypeError):
model.data['new_data'] = 'new_data'
assert model.data == model_data
def test_lazy_model_load_raises_on_context_validation(mock_plugin,
context_urls_all,
model_type,
mock_entity_create_id):
from coalaip.models import LazyLoadableModel
from coalaip.exceptions import ModelDataError
bad_context_data = {'@context': 'other_context'}
mock_plugin.load.return_value = bad_context_data
assert context_urls_all != bad_context_data.get('@context')
model = LazyLoadableModel(ld_type=model_type, ld_context=context_urls_all)
with raises(ModelDataError):
model.load(mock_entity_create_id, plugin=mock_plugin)
def test_lazy_model_ld_context_immutable(model_type, ld_context):
from collections import Mapping
from coalaip.models import LazyLoadableModel
model = LazyLoadableModel(ld_type=model_type, ld_context=ld_context)
if isinstance(ld_context, str):
with raises(TypeError):
model.ld_context[0] = 'a'
elif isinstance(ld_context, Mapping):
with raises(TypeError):
model.ld_context['new_key'] = 'new_data'
else:
with raises(TypeError):
model.ld_context[0] = 'new_context'
def test_lazy_model_init_with_data(mock_plugin, model_data, model_type,
mock_entity_create_id):
from coalaip.models import Model, LazyLoadableModel
model = LazyLoadableModel(data=model_data, ld_type=model_type)
assert model.data == model_data
assert isinstance(model.loaded_model, Model)
assert model.loaded_model.data == model_data
assert model.loaded_model.ld_type == model.ld_type
assert model.loaded_model.ld_context == model.ld_context
assert model.loaded_model.validator == model.validator
# If initialized with data, load() becomes a noop
model.load(mock_entity_create_id, plugin=mock_plugin)
mock_plugin.load.assert_not_called()
def test_lazy_model_init_defaults(context_urls_all, model_type):
from coalaip.models import LazyLoadableModel
model = LazyLoadableModel(ld_type=model_type)
assert model.loaded_model is None
assert model.ld_type == model_type
assert set(model.ld_context) == set(context_urls_all)
assert callable(model.validator)
def test_lazy_model_init(model_type):
from attr import validators
from coalaip.models import LazyLoadableModel
ld_context = 'ld_context'
validator = validators.instance_of(dict)
model = LazyLoadableModel(ld_type=model_type, ld_context=ld_context,
validator=validator)
assert model.loaded_model is None
assert model.ld_type == model_type
assert model.ld_context == ld_context
assert model.validator == validator
def test_lazy_model_immutable(model_data, model_type):
from attr import validators
from attr.exceptions import FrozenInstanceError
from coalaip.models import Model, LazyLoadableModel
model = LazyLoadableModel(data=model_data, ld_type=model_type)
with raises(FrozenInstanceError):
model.loaded_model = Model(data={'other': 'other'}, ld_type='other_type')
with raises(FrozenInstanceError):
model.ld_type = 'other_type'
with raises(FrozenInstanceError):
model.ld_context = 'other_context'
with raises(FrozenInstanceError):
model.validator = validators.instance_of(str)
def test_lazy_model_load_jsonld(mock_plugin, model_data, model_jsonld,
model_type, model_context,
mock_entity_create_id):
from coalaip.models import Model, LazyLoadableModel
mock_plugin.load.return_value = model_jsonld
model = LazyLoadableModel(ld_type=model_type, ld_context=model_context)
model.load(mock_entity_create_id, plugin=mock_plugin)
mock_plugin.load.assert_called_with(mock_entity_create_id)
assert model.data == model_data
assert isinstance(model.loaded_model, Model)
assert model.loaded_model.data == model_data
assert model.loaded_model.ld_type == model.ld_type
assert model.loaded_model.ld_id == model.ld_id
assert model.loaded_model.ld_context == model.ld_context
assert model.loaded_model.validator == model.validator
# If initialized with data, load() becomes a noop
mock_plugin.reset_mock()
model.load(mock_entity_create_id, plugin=mock_plugin)
mock_plugin.load.assert_not_called()
def test_lazy_model_raises_on_ld_id_access_before_load(model_type):
from coalaip.models import LazyLoadableModel
from coalaip.exceptions import ModelNotYetLoadedError
model = LazyLoadableModel(ld_type=model_type)
with raises(ModelNotYetLoadedError):
model.ld_id