notifications-api/tests/app/dao/test_templates_dao.py

from app.dao.templates_dao import (
    dao_create_template,
    dao_get_template_by_id_and_service_id,
    dao_get_all_templates_for_service,
    dao_update_template
)
from tests.app.conftest import sample_template as create_sample_template
from app.models import Template


def test_create_template(sample_service):
    data = {
        'name': 'Sample Template',
        'template_type': "sms",
        'content': "Template content",
        'service': sample_service
    }
    template = Template(**data)
    dao_create_template(template)

    assert Template.query.count() == 1
    assert len(dao_get_all_templates_for_service(sample_service.id)) == 1
    assert dao_get_all_templates_for_service(sample_service.id)[0].name == 'Sample Template'


def test_update_template(sample_service):
    data = {
        'name': 'Sample Template',
        'template_type': "sms",
        'content': "Template content",
        'service': sample_service
    }
    template = Template(**data)
    dao_create_template(template)
    created = dao_get_all_templates_for_service(sample_service.id)[0]
    assert created.name == 'Sample Template'

    created.name = 'new name'
    dao_update_template(created)
    assert dao_get_all_templates_for_service(sample_service.id)[0].name == 'new name'


def test_get_all_templates_for_service(service_factory):
    service_1 = service_factory.get('service 1')
    service_2 = service_factory.get('service 2')

    assert Template.query.count() == 2
    assert len(dao_get_all_templates_for_service(service_1.id)) == 1
    assert len(dao_get_all_templates_for_service(service_2.id)) == 1

    template_1 = Template(
        name='Sample Template 1',
        template_type="sms",
        content="Template content",
        service=service_1
    )
    template_2 = Template(
        name='Sample Template 2',
        template_type="sms",
        content="Template content",
        service=service_1
    )
    template_3 = Template(
        name='Sample Template 3',
        template_type="sms",
        content="Template content",
        service=service_2
    )
    dao_create_template(template_1)
    dao_create_template(template_2)
    dao_create_template(template_3)

    assert Template.query.count() == 5
    assert len(dao_get_all_templates_for_service(service_1.id)) == 3
    assert len(dao_get_all_templates_for_service(service_2.id)) == 2


def test_get_all_templates_for_service_in_created_order(sample_service):
    template_1 = Template(
        name='Sample Template 1',
        template_type="sms",
        content="Template content",
        service=sample_service
    )
    template_2 = Template(
        name='Sample Template 2',
        template_type="sms",
        content="Template content",
        service=sample_service
    )
    template_3 = Template(
        name='Sample Template 3',
        template_type="sms",
        content="Template content",
        service=sample_service
    )
    dao_create_template(template_1)
    dao_create_template(template_2)
    dao_create_template(template_3)

    assert Template.query.count() == 3
    assert dao_get_all_templates_for_service(sample_service.id)[0].name == 'Sample Template 1'
    assert dao_get_all_templates_for_service(sample_service.id)[1].name == 'Sample Template 2'
    assert dao_get_all_templates_for_service(sample_service.id)[2].name == 'Sample Template 3'


def test_get_all_returns_empty_list_if_no_templates(sample_service):
    assert Template.query.count() == 0
    assert len(dao_get_all_templates_for_service(sample_service.id)) == 0


def test_get_template_by_id_and_service(notify_db, notify_db_session, sample_service):
    sample_template = create_sample_template(
        notify_db,
        notify_db_session,
        template_name='Test Template',
        service=sample_service)
    assert dao_get_template_by_id_and_service_id(
        template_id=sample_template.id,
        service_id=sample_service.id).name == 'Test Template'
    assert Template.query.count() == 1


def test_get_template_by_id_and_service_returns_none_if_no_template(sample_service):
    assert not dao_get_template_by_id_and_service_id(template_id=999, service_id=sample_service.id)