reformat files, add more tests

This commit is contained in:
dandds 2018-10-12 10:00:03 -04:00
parent ae7c644d91
commit df35725430
3 changed files with 21 additions and 0 deletions

View File

@ -10,6 +10,7 @@ from tempfile import TemporaryDirectory
from atst.app import make_app, make_config
from atst.database import db as _db
from atst.domain.auth import logout
from atst.queue import queue
import tests.factories as factories
from tests.mocks import PDF_FILENAME
@ -22,6 +23,7 @@ def app(request):
config = make_config()
config.update({"STORAGE_CONTAINER": upload_dir.name})
config.update({"RQ_QUEUES": ["test"]})
_app = make_app(config)

16
tests/test_queue.py Normal file
View File

@ -0,0 +1,16 @@
import pytest
from atst.queue import queue, send_mail
# ensure queue is always empty for unit testing
@pytest.fixture(scope="function", autouse=True)
def reset_queue():
queue.get_queue().empty()
yield
queue.get_queue().empty()
def test_send_mail():
assert len(queue.get_queue()) == 0
send_mail.queue(
["lordvader@geocities.net"], "death start", "how is it coming along?"
)
assert len(queue.get_queue()) == 1

View File

@ -47,3 +47,6 @@ def test_redis_mailer_can_save_messages(app):
mailer.send(**message_data)
assert len(mailer.messages) == 1
message = mailer.messages[0]
assert message_data["recipients"][0] in message
assert message_data["subject"] in message
assert message_data["body"] in message