59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
import pytest
|
|
from marshmallow.exceptions import ValidationError
|
|
|
|
from lemur.tests.factories import RoleFactory
|
|
|
|
|
|
def test_get_object_attribute():
|
|
from lemur.schemas import get_object_attribute
|
|
|
|
with pytest.raises(ValidationError):
|
|
get_object_attribute({})
|
|
|
|
with pytest.raises(ValidationError):
|
|
get_object_attribute([{}], many=True)
|
|
|
|
with pytest.raises(ValidationError):
|
|
get_object_attribute([{}, {'id': 1}], many=True)
|
|
|
|
with pytest.raises(ValidationError):
|
|
get_object_attribute([{}, {'name': 'test'}], many=True)
|
|
|
|
assert get_object_attribute({'name': 'test'}) == 'name'
|
|
assert get_object_attribute({'id': 1}) == 'id'
|
|
assert get_object_attribute([{'name': 'test'}], many=True) == 'name'
|
|
assert get_object_attribute([{'id': 1}], many=True) == 'id'
|
|
|
|
|
|
def test_fetch_objects(session):
|
|
from lemur.roles.models import Role
|
|
from lemur.schemas import fetch_objects
|
|
|
|
role = RoleFactory()
|
|
role1 = RoleFactory()
|
|
session.commit()
|
|
|
|
data = {'id': role.id}
|
|
found_role = fetch_objects(Role, data)
|
|
assert found_role == role
|
|
|
|
data = {'name': role.name}
|
|
found_role = fetch_objects(Role, data)
|
|
assert found_role == role
|
|
|
|
data = [{'id': role.id}, {'id': role1.id}]
|
|
found_roles = fetch_objects(Role, data, many=True)
|
|
assert found_roles == [role, role1]
|
|
|
|
data = [{'name': role.name}, {'name': role1.name}]
|
|
found_roles = fetch_objects(Role, data, many=True)
|
|
assert found_roles == [role, role1]
|
|
|
|
with pytest.raises(ValidationError):
|
|
data = [{'name': 'blah'}, {'name': role1.name}]
|
|
fetch_objects(Role, data, many=True)
|
|
|
|
with pytest.raises(ValidationError):
|
|
data = {'name': 'nah'}
|
|
fetch_objects(Role, data)
|