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

from app.models import Notification

from app.dao.notifications_dao import (
    save_notification,
    get_notification,
    get_notifications
)


def test_save_notification(notify_db, notify_db_session, sample_template, sample_job):

    assert Notification.query.count() == 0
    to = '+44709123456'
    data = {
        'to': to,
        'job': sample_job,
        'service': sample_template.service,
        'template': sample_template
    }

    notification = Notification(**data)
    save_notification(notification)

    assert Notification.query.count() == 1
    notification_from_db = Notification.query.all()[0]
    assert notification_from_db.id
    assert data['to'] == notification_from_db.to
    assert data['job'] == notification_from_db.job
    assert data['service'] == notification_from_db.service
    assert data['template'] == notification_from_db.template
    assert 'sent' == notification_from_db.status


def test_save_notification_no_job_id(notify_db, notify_db_session, sample_template):

    assert Notification.query.count() == 0
    to = '+44709123456'
    data = {
        'to': to,
        'service': sample_template.service,
        'template': sample_template
    }

    notification = Notification(**data)
    save_notification(notification)

    assert Notification.query.count() == 1
    notification_from_db = Notification.query.all()[0]
    assert notification_from_db.id
    assert data['to'] == notification_from_db.to
    assert data['service'] == notification_from_db.service
    assert data['template'] == notification_from_db.template
    assert 'sent' == notification_from_db.status


def test_get_notification_for_job(notify_db, notify_db_session, sample_notification):
    notifcation_from_db = get_notification(sample_notification.service.id,
                                           sample_notification.job_id,
                                           sample_notification.id)
    assert sample_notification == notifcation_from_db


def test_get_all_notifications_for_job(notify_db, notify_db_session, sample_job):

    from tests.app.conftest import sample_notification
    for i in range(0, 5):
        sample_notification(notify_db,
                            notify_db_session,
                            service=sample_job.service,
                            template=sample_job.template,
                            job=sample_job)

    notifcations_from_db = get_notifications(sample_job.service.id, sample_job.id)
    assert len(notifcations_from_db) == 5


def test_update_notification(notify_db, notify_db_session, sample_notification):
    assert sample_notification.status == 'sent'

    update_dict = {
        'id': str(sample_notification.id),
        'service': str(sample_notification.service.id),
        'template': sample_notification.template.id,
        'job': str(sample_notification.job.id),
        'status': 'failed'
    }

    save_notification(sample_notification, update_dict=update_dict)
    notification_from_db = Notification.query.get(sample_notification.id)
    assert notification_from_db.status == 'failed'