Create queue fixture

This commit is contained in:
richard-dds 2018-10-29 12:01:45 -04:00
parent 7404cc4a59
commit f3be2d76ea
2 changed files with 10 additions and 16 deletions

View File

@ -9,8 +9,7 @@ from tempfile import TemporaryDirectory
from atst.app import make_app, make_config from atst.app import make_app, make_config
from atst.database import db as _db from atst.database import db as _db
from atst.domain.auth import logout from atst.queue import queue as atst_queue
from atst.queue import queue
import tests.factories as factories import tests.factories as factories
from tests.mocks import PDF_FILENAME from tests.mocks import PDF_FILENAME
@ -136,3 +135,10 @@ def extended_financial_verification_data(pdf_upload):
"clin_2003": "7000", "clin_2003": "7000",
"task_order": pdf_upload, "task_order": pdf_upload,
} }
@pytest.fixture(scope="function")
def queue():
_queue = atst_queue
yield _queue
_queue.get_queue().empty()

View File

@ -1,17 +1,5 @@
import pytest def test_send_mail(queue):
from atst.queue import queue
# 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():
initial = len(queue.get_queue())
queue.send_mail( queue.send_mail(
["lordvader@geocities.net"], "death start", "how is it coming along?" ["lordvader@geocities.net"], "death start", "how is it coming along?"
) )
assert len(queue.get_queue()) == initial + 1 assert len(queue.get_queue()) == 1