mirror of
https://github.com/GSA/notifications-api.git
synced 2026-05-05 16:48:31 -04:00
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
import pytest
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
from app.dao.organisations_dao import (
|
|
dao_create_organisation,
|
|
dao_get_organisations,
|
|
dao_get_organisation_by_id, dao_update_organisation
|
|
)
|
|
from app.models import Organisation
|
|
|
|
from tests.app.db import create_organisation
|
|
|
|
|
|
def test_create_organisation(notify_db, notify_db_session):
|
|
organisation = create_organisation()
|
|
|
|
assert Organisation.query.count() == 1
|
|
organisation_from_db = Organisation.query.first()
|
|
assert organisation == organisation_from_db
|
|
|
|
|
|
def test_create_organisation_without_name_or_colour_is_valid(notify_db, notify_db_session):
|
|
organisation = create_organisation(name=None, colour=None)
|
|
|
|
assert Organisation.query.count() == 1
|
|
organisation_from_db = Organisation.query.first()
|
|
assert organisation == organisation_from_db
|
|
|
|
|
|
def test_create_organisation_without_logo_raises_error(notify_db, notify_db_session):
|
|
with pytest.raises(IntegrityError) as excinfo:
|
|
create_organisation(logo=None)
|
|
assert 'column "logo" violates not-null constraint' in str(excinfo.value)
|
|
assert Organisation.query.count() == 0
|
|
|
|
|
|
def test_get_organisations_gets_all_organisations(notify_db, notify_db_session):
|
|
org_1 = create_organisation(name='test_org_1')
|
|
org_2 = create_organisation(name='test_org_2')
|
|
|
|
organisations = dao_get_organisations()
|
|
|
|
assert len(organisations) == 2
|
|
assert org_1 == organisations[0]
|
|
assert org_2 == organisations[1]
|
|
|
|
|
|
def test_get_organisation_by_id_gets_correct_organisation(notify_db, notify_db_session):
|
|
organisation = create_organisation()
|
|
|
|
organisation_from_db = dao_get_organisation_by_id(organisation.id)
|
|
|
|
assert organisation_from_db == organisation
|
|
|
|
|
|
def test_update_organisation(notify_db, notify_db_session):
|
|
updated_name = 'new name'
|
|
organisation = create_organisation()
|
|
|
|
organisation_from_db = Organisation.query.first()
|
|
assert organisation_from_db.name != updated_name
|
|
|
|
organisation.name = updated_name
|
|
|
|
dao_update_organisation(organisation)
|
|
organisation_from_db = Organisation.query.first()
|
|
|
|
assert organisation_from_db.name == updated_name
|