def test_in_registry():
from field_permissions.registry import FieldPermissionsModelRegistry
field_permissions = FieldPermissionsModelRegistry()
field_permissions.register(Dummy)
tm = Dummy.objects.create(field1="test")
assert field_permissions.in_registry(tm)
def test_get_model_fields_exclude_some_fields():
from field_permissions.registry import FieldPermissionsModelRegistry
field_permissions = FieldPermissionsModelRegistry()
field_permissions.register(Dummy, exclude_fields=["field1", "dummysub"])
model_field_names = [
field.name for field in field_permissions.get_model_fields(Dummy)
]
assert model_field_names == ["id", "field2", "field3", "second_sub"]
def test_get_model_fields_include_some_fields():
from field_permissions.registry import FieldPermissionsModelRegistry
field_permissions = FieldPermissionsModelRegistry()
field_permissions.register(Dummy, include_fields=["id", "field1"])
model_field_names = [
field.name for field in field_permissions.get_model_fields(Dummy)
]
assert model_field_names == ["id", "field1"]
def test_get_model_fields_exclude_some_fields():
from field_permissions.registry import FieldPermissionsModelRegistry
field_permissions = FieldPermissionsModelRegistry()
field_permissions.register(Dummy, exclude_fields=['field1', 'dummysub'])
model_field_names = [
field.name for field in field_permissions.get_model_fields(Dummy)
]
assert model_field_names == ['id', 'field2', 'field3', 'second_sub']
def test_get_field_permissions_for_model():
from field_permissions.registry import FieldPermissionsModelRegistry
field_permissions = FieldPermissionsModelRegistry()
field_permissions.register(Dummy)
perms = field_permissions.get_field_permissions_for_model(Dummy)
perm_codenames = [codename for (codename, name) in perms]
assert perm_codenames == [
'view_dummy_dummysub_set', 'change_dummy_dummysub_set',
'view_dummy_id', 'change_dummy_id', 'view_dummy_field1',
'change_dummy_field1', 'view_dummy_field2', 'change_dummy_field2',
'view_dummy_field3', 'change_dummy_field3', 'view_dummy_second_sub',
'change_dummy_second_sub'
]
def test_not_in_registry():
from field_permissions.registry import FieldPermissionsModelRegistry
field_permissions = FieldPermissionsModelRegistry()
assert not field_permissions.in_registry(Dummy)