ensure the app context is included in every single flask command

This commit is contained in:
Leo Hemsted
2017-11-24 12:01:28 +00:00
parent fcf3932bb4
commit 4c14e3279f
2 changed files with 41 additions and 20 deletions

View File

@@ -3,10 +3,12 @@ from datetime import datetime
from app.commands import backfill_processing_time
def test_backfill_processing_time_works_for_correct_dates(mocker):
def test_backfill_processing_time_works_for_correct_dates(mocker, notify_api):
send_mock = mocker.patch('app.commands.send_processing_time_for_start_and_end')
backfill_processing_time.callback('2017-08-01', '2017-08-03')
# backfill_processing_time is a click.Command object - if you try invoking the callback on its own, it
# throws a `RuntimeError: There is no active click context.` - so get at the original function using __wrapped__
backfill_processing_time.callback.__wrapped__('2017-08-01', '2017-08-03')
assert send_mock.call_count == 3
send_mock.assert_any_call(datetime(2017, 7, 31, 23, 0), datetime(2017, 8, 1, 23, 0))