uva-covid19-testing-communicator/tests/services/test_notification_service.py

from datetime import datetime

import pytz

from tests.base_test import BaseTest


from communicator import app
from communicator.models import Sample
from communicator.services.notification_service import TEST_MESSAGES, NotificationService


class TestNotificationService(BaseTest):

    def test_send_notification(self):
        message_count = len(TEST_MESSAGES)
        sample = Sample(email="dan@stauntonmakerspace.com", result_code="1234")
        with NotificationService(app) as notifier:
            notifier.send_result_email(sample)
        self.assertEqual(len(TEST_MESSAGES), message_count + 1)
        self.assertEqual("UVA: BE SAFE Notification", self.decode(TEST_MESSAGES[-1]['subject']))