From f34cf2b84b10973e1186e0410b7709fde9440a49 Mon Sep 17 00:00:00 2001 From: Benjamin Dauvergne Date: Sat, 24 Mar 2018 02:09:33 +0100 Subject: [PATCH 3/5] tests: use freezegun in test_notification (#22732) --- tests/test_notification.py | 30 ++++++++++++------------------ 1 file changed, 12 insertions(+), 18 deletions(-) diff --git a/tests/test_notification.py b/tests/test_notification.py index 0dbb1b5..2f62b56 100644 --- a/tests/test_notification.py +++ b/tests/test_notification.py @@ -255,13 +255,9 @@ def test_notification_id_and_origin(john_doe): @mock.patch('combo.apps.lingo.models.requests.get') -def test_notify_remote_items(mock_get, app, john_doe, jane_doe, regie): - - datetime_format = '%Y-%m-%dT%H:%M:%S' - invoice_now = now() - creation_date = (invoice_now - timedelta(days=1)).strftime(datetime_format) - pay_limit_date = (invoice_now + timedelta(days=20)).strftime(datetime_format) - new_pay_limit_date = (invoice_now + timedelta(days=5)).strftime(datetime_format) +@pytest.mark.freeze_time('2016-01-02') +def test_notify_remote_items(mock_get, app, john_doe, jane_doe, regie, freezer): + invoice_now = freezer() FAKE_PENDING_INVOICES = { "data": { @@ -272,8 +268,8 @@ def test_notify_remote_items(mock_get, app, john_doe, jane_doe, regie): 'label': '010101', 'total_amount': '10', 'amount': '10', - 'created': creation_date, - 'pay_limit_date': pay_limit_date, + 'created': '2016-01-01T09:00:00', + 'pay_limit_date': '2016-01-20T09:00:00', 'has_pdf': False, }, { @@ -281,8 +277,8 @@ def test_notify_remote_items(mock_get, app, john_doe, jane_doe, regie): 'label': '0101011', 'total_amount': '1.5', 'amount': '1.5', - 'created': creation_date, - 'pay_limit_date': pay_limit_date, + 'created': '2016-01-01T09:00:00', + 'pay_limit_date': '2016-01-20T09:00:00', 'has_pdf': False, } ] @@ -294,8 +290,8 @@ def test_notify_remote_items(mock_get, app, john_doe, jane_doe, regie): 'label': '020202', 'total_amount': '2.0', 'amount': '2.0', - 'created': creation_date, - 'pay_limit_date': pay_limit_date, + 'created': '2016-01-01T09:00:00', + 'pay_limit_date': '2016-01-20T09:00:00', 'has_pdf': False, } ] @@ -308,8 +304,8 @@ def test_notify_remote_items(mock_get, app, john_doe, jane_doe, regie): 'label': '030303', 'total_amount': '42', 'amount': '42', - 'created': creation_date, - 'pay_limit_date': pay_limit_date, + 'created': '2016-01-01T09:00:00', + 'pay_limit_date': '2016-01-20T09:00:00', 'has_pdf': False, } ] @@ -336,9 +332,7 @@ def test_notify_remote_items(mock_get, app, john_doe, jane_doe, regie): cell = ActiveItems(page=page, placeholder='content', order=0) cell.save() - for john_doe in FAKE_PENDING_INVOICES['data']: - for invoice in FAKE_PENDING_INVOICES['data'][john_doe]['invoices']: - invoice['pay_limit_date'] = new_pay_limit_date + freezer.move_to(invoice_now + timedelta(days=16)) # create remind notifications regie.notify_new_remote_invoices() -- 2.14.2