130 lines
4.5 KiB
Python
130 lines
4.5 KiB
Python
from uuid import UUID
|
|
|
|
import pytest
|
|
from sqlalchemy_utils import PhoneNumber
|
|
from teal.db import UniqueViolation
|
|
from teal.enums import Country
|
|
|
|
from ereuse_devicehub.config import DevicehubConfig
|
|
from ereuse_devicehub.db import db
|
|
from ereuse_devicehub.devicehub import Devicehub
|
|
from ereuse_devicehub.resources.agent import OrganizationDef, models, schemas
|
|
from ereuse_devicehub.resources.agent.models import Membership, Organization, Person, System
|
|
from tests.conftest import app_context, create_user
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_agent():
|
|
"""Tests creating an person."""
|
|
person = Person(name='Timmy',
|
|
tax_id='XYZ',
|
|
country=Country.ES,
|
|
telephone=PhoneNumber('+34666666666'),
|
|
email='foo@bar.com')
|
|
db.session.add(person)
|
|
db.session.commit()
|
|
|
|
p = schemas.Person().dump(person)
|
|
assert p['name'] == person.name == 'Timmy'
|
|
assert p['taxId'] == person.tax_id == 'XYZ'
|
|
assert p['country'] == person.country.name == 'ES'
|
|
assert p['telephone'] == person.telephone.international == '+34 666 66 66 66'
|
|
assert p['email'] == person.email == 'foo@bar.com'
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_system():
|
|
"""Tests creating a system."""
|
|
system = System(name='Workbench',
|
|
email='hello@ereuse.org')
|
|
db.session.add(system)
|
|
db.session.commit()
|
|
|
|
s = schemas.System().dump(system)
|
|
assert s['name'] == system.name == 'Workbench'
|
|
assert s['email'] == system.email == 'hello@ereuse.org'
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_organization():
|
|
"""Tests creating an organization."""
|
|
org = Organization(name='ACME',
|
|
tax_id='XYZ',
|
|
country=Country.ES,
|
|
email='contact@acme.com')
|
|
db.session.add(org)
|
|
db.session.commit()
|
|
|
|
o = schemas.Organization().dump(org)
|
|
assert o['name'] == org.name == 'ACME'
|
|
assert o['taxId'] == org.tax_id == 'XYZ'
|
|
assert org.country.name == o['country'] == 'ES'
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_membership():
|
|
"""Tests assigning an Individual to an Organization."""
|
|
person = Person(name='Timmy')
|
|
org = Organization(name='ACME')
|
|
person.member_of.add(Membership(org, person, id='acme-1'))
|
|
db.session.add(person)
|
|
db.session.flush()
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_membership_repeated():
|
|
person = Person(name='Timmy')
|
|
org = Organization(name='ACME')
|
|
person.member_of.add(Membership(org, person, id='acme-1'))
|
|
db.session.add(person)
|
|
|
|
person.member_of.add(Membership(org, person))
|
|
with pytest.raises(UniqueViolation):
|
|
db.session.flush()
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_membership_repeating_id():
|
|
person = Person(name='Timmy')
|
|
org = Organization(name='ACME')
|
|
person.member_of.add(Membership(org, person, id='acme-1'))
|
|
db.session.add(person)
|
|
db.session.flush()
|
|
|
|
person2 = Person(name='Tommy')
|
|
person2.member_of.add(Membership(org, person2, id='acme-1'))
|
|
db.session.add(person2)
|
|
with pytest.raises(UniqueViolation) as e:
|
|
db.session.flush()
|
|
assert 'One member id per organization' in str(e)
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_default_org_exists(config: DevicehubConfig):
|
|
"""
|
|
Ensures that the default organization is created on app
|
|
initialization and that is accessible for the method
|
|
:meth:`ereuse_devicehub.resources.user.Organization.get_default_org`.
|
|
"""
|
|
assert models.Organization.query.filter_by(name=config.ORGANIZATION_NAME,
|
|
tax_id=config.ORGANIZATION_TAX_ID).one()
|
|
assert isinstance(models.Organization.get_default_org_id(), UUID)
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_assign_individual_user():
|
|
"""Tests assigning an individual to an user."""
|
|
user = create_user()
|
|
assert len(user.individuals) == 1
|
|
assert next(iter(user.individuals)).name == 'Timmy'
|
|
|
|
|
|
@pytest.mark.usefixtures(app_context.__name__)
|
|
def test_create_organization_main_method(app: Devicehub):
|
|
org_def = app.resources[models.Organization.t] # type: OrganizationDef
|
|
o = org_def.create_org('ACME', tax_id='FOO', country='ES')
|
|
org = models.Agent.query.filter_by(id=o['id']).one() # type: Organization
|
|
assert org.name == o['name'] == 'ACME'
|
|
assert org.tax_id == o['taxId'] == 'FOO'
|
|
assert org.country.name == o['country'] == 'ES'
|