mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-14 23:08:07 -04:00
Localize notification_utils to the admin
This changeset pulls in all of the notification_utils code directly into the admin and removes it as an external dependency. We are doing this to cut down on operational maintenance of the project and will begin removing parts of it no longer needed for the admin. Signed-off-by: Carlo Costino <carlo.costino@gsa.gov>
This commit is contained in:
Carlo Costino
269
tests/notifications_utils/test_letter_timings.py
Normal file
269
tests/notifications_utils/test_letter_timings.py
Normal file
@@ -0,0 +1,269 @@
|
||||
from datetime import datetime
|
||||
|
||||
import pytest
|
||||
import pytz
|
||||
from freezegun import freeze_time
|
||||
|
||||
from notifications_utils.letter_timings import (
|
||||
get_letter_timings,
|
||||
letter_can_be_cancelled,
|
||||
)
|
||||
|
||||
|
||||
@freeze_time("2017-07-14 13:59:59") # Friday, before print deadline (3PM EST)
|
||||
@pytest.mark.parametrize(
|
||||
"upload_time, expected_print_time, is_printed, first_class, expected_earliest, expected_latest",
|
||||
[
|
||||
# EST
|
||||
# ==================================================================
|
||||
# First thing Monday
|
||||
(
|
||||
"Monday 2017-07-10 00:00:01",
|
||||
"Tuesday 2017-07-11 15:00",
|
||||
True,
|
||||
"Wednesday 2017-07-12 16:00",
|
||||
"Thursday 2017-07-13 16:00",
|
||||
"Friday 2017-07-14 16:00",
|
||||
),
|
||||
# Monday at 17:29 EST (sent on monday)
|
||||
(
|
||||
"Monday 2017-07-10 16:29:59",
|
||||
"Tuesday 2017-07-11 15:00",
|
||||
True,
|
||||
"Wednesday 2017-07-12 16:00",
|
||||
"Thursday 2017-07-13 16:00",
|
||||
"Friday 2017-07-14 16:00",
|
||||
),
|
||||
# Monday at 17:30 EST (sent on tuesday)
|
||||
(
|
||||
"Monday 2017-07-10 16:30:01",
|
||||
"Wednesday 2017-07-12 15:00",
|
||||
True,
|
||||
"Thursday 2017-07-13 16:00",
|
||||
"Friday 2017-07-14 16:00",
|
||||
"Saturday 2017-07-15 16:00",
|
||||
),
|
||||
# Tuesday before 17:30 EST
|
||||
(
|
||||
"Tuesday 2017-07-11 12:00:00",
|
||||
"Wednesday 2017-07-12 15:00",
|
||||
True,
|
||||
"Thursday 2017-07-13 16:00",
|
||||
"Friday 2017-07-14 16:00",
|
||||
"Saturday 2017-07-15 16:00",
|
||||
),
|
||||
# Wednesday before 17:30 EST
|
||||
(
|
||||
"Wednesday 2017-07-12 12:00:00",
|
||||
"Thursday 2017-07-13 15:00",
|
||||
True,
|
||||
"Friday 2017-07-14 16:00",
|
||||
"Saturday 2017-07-15 16:00",
|
||||
"Monday 2017-07-17 16:00",
|
||||
),
|
||||
# Thursday before 17:30 EST
|
||||
(
|
||||
"Thursday 2017-07-13 12:00:00",
|
||||
"Friday 2017-07-14 15:00",
|
||||
False,
|
||||
"Saturday 2017-07-15 16:00",
|
||||
"Monday 2017-07-17 16:00",
|
||||
"Tuesday 2017-07-18 16:00",
|
||||
),
|
||||
# Friday anytime
|
||||
(
|
||||
"Friday 2017-07-14 00:00:00",
|
||||
"Monday 2017-07-17 15:00",
|
||||
False,
|
||||
"Tuesday 2017-07-18 16:00",
|
||||
"Wednesday 2017-07-19 16:00",
|
||||
"Thursday 2017-07-20 16:00",
|
||||
),
|
||||
(
|
||||
"Friday 2017-07-14 12:00:00",
|
||||
"Monday 2017-07-17 15:00",
|
||||
False,
|
||||
"Tuesday 2017-07-18 16:00",
|
||||
"Wednesday 2017-07-19 16:00",
|
||||
"Thursday 2017-07-20 16:00",
|
||||
),
|
||||
(
|
||||
"Friday 2017-07-14 22:00:00",
|
||||
"Monday 2017-07-17 15:00",
|
||||
False,
|
||||
"Tuesday 2017-07-18 16:00",
|
||||
"Wednesday 2017-07-19 16:00",
|
||||
"Thursday 2017-07-20 16:00",
|
||||
),
|
||||
# Saturday anytime
|
||||
(
|
||||
"Saturday 2017-07-14 12:00:00",
|
||||
"Monday 2017-07-17 15:00",
|
||||
False,
|
||||
"Tuesday 2017-07-18 16:00",
|
||||
"Wednesday 2017-07-19 16:00",
|
||||
"Thursday 2017-07-20 16:00",
|
||||
),
|
||||
# Sunday before 1730 EST
|
||||
(
|
||||
"Sunday 2017-07-15 15:59:59",
|
||||
"Monday 2017-07-17 15:00",
|
||||
False,
|
||||
"Tuesday 2017-07-18 16:00",
|
||||
"Wednesday 2017-07-19 16:00",
|
||||
"Thursday 2017-07-20 16:00",
|
||||
),
|
||||
# Sunday after 17:30 EST
|
||||
(
|
||||
"Sunday 2017-07-16 16:30:01",
|
||||
"Tuesday 2017-07-18 15:00",
|
||||
False,
|
||||
"Wednesday 2017-07-19 16:00",
|
||||
"Thursday 2017-07-20 16:00",
|
||||
"Friday 2017-07-21 16:00",
|
||||
),
|
||||
# GMT
|
||||
# ==================================================================
|
||||
# Monday at 17:29 GMT
|
||||
(
|
||||
"Monday 2017-01-02 17:29:59",
|
||||
"Tuesday 2017-01-03 15:00",
|
||||
True,
|
||||
"Wednesday 2017-01-04 16:00",
|
||||
"Thursday 2017-01-05 16:00",
|
||||
"Friday 2017-01-06 16:00",
|
||||
),
|
||||
# Monday at 17:00 GMT
|
||||
(
|
||||
"Monday 2017-01-02 17:30:01",
|
||||
"Wednesday 2017-01-04 15:00",
|
||||
True,
|
||||
"Thursday 2017-01-05 16:00",
|
||||
"Friday 2017-01-06 16:00",
|
||||
"Saturday 2017-01-07 16:00",
|
||||
),
|
||||
],
|
||||
)
|
||||
@pytest.mark.skip(reason="Letters being developed later")
|
||||
def test_get_estimated_delivery_date_for_letter(
|
||||
upload_time,
|
||||
expected_print_time,
|
||||
is_printed,
|
||||
first_class,
|
||||
expected_earliest,
|
||||
expected_latest,
|
||||
):
|
||||
# remove the day string from the upload_time, which is purely informational
|
||||
|
||||
def format_dt(x):
|
||||
return x.astimezone(pytz.timezone("America/New_York")).strftime(
|
||||
"%A %Y-%m-%d %H:%M"
|
||||
)
|
||||
|
||||
upload_time = upload_time.split(" ", 1)[1]
|
||||
|
||||
timings = get_letter_timings(upload_time, postage="second")
|
||||
|
||||
assert format_dt(timings.printed_by) == expected_print_time
|
||||
assert timings.is_printed == is_printed
|
||||
assert format_dt(timings.earliest_delivery) == expected_earliest
|
||||
assert format_dt(timings.latest_delivery) == expected_latest
|
||||
|
||||
first_class_timings = get_letter_timings(upload_time, postage="first")
|
||||
|
||||
assert format_dt(first_class_timings.printed_by) == expected_print_time
|
||||
assert first_class_timings.is_printed == is_printed
|
||||
assert format_dt(first_class_timings.earliest_delivery) == first_class
|
||||
assert format_dt(first_class_timings.latest_delivery) == first_class
|
||||
|
||||
|
||||
@pytest.mark.parametrize("status", ["sending", "pending"])
|
||||
def test_letter_cannot_be_cancelled_if_letter_status_is_not_created_or_pending_virus_check(
|
||||
status,
|
||||
):
|
||||
notification_created_at = datetime.utcnow()
|
||||
|
||||
assert not letter_can_be_cancelled(status, notification_created_at)
|
||||
|
||||
|
||||
@freeze_time("2018-7-7 16:00:00")
|
||||
@pytest.mark.parametrize(
|
||||
"notification_created_at",
|
||||
[
|
||||
datetime(2018, 7, 6, 18, 0), # created yesterday after 1730
|
||||
datetime(2018, 7, 7, 12, 0), # created today
|
||||
],
|
||||
)
|
||||
@pytest.mark.skip(reason="Letters not part of release")
|
||||
def test_letter_can_be_cancelled_if_before_1730_and_letter_created_before_1730(
|
||||
notification_created_at,
|
||||
):
|
||||
notification_status = "pending-virus-check"
|
||||
|
||||
assert letter_can_be_cancelled(notification_status, notification_created_at)
|
||||
|
||||
|
||||
@freeze_time("2017-12-12 17:30:00")
|
||||
@pytest.mark.parametrize(
|
||||
"notification_created_at",
|
||||
[
|
||||
datetime(2017, 12, 12, 17, 0),
|
||||
datetime(2017, 12, 12, 17, 30),
|
||||
],
|
||||
)
|
||||
@pytest.mark.skip(reason="Letters not part of release")
|
||||
def test_letter_cannot_be_cancelled_if_1730_exactly_and_letter_created_at_or_before_1730(
|
||||
notification_created_at,
|
||||
):
|
||||
notification_status = "pending-virus-check"
|
||||
|
||||
assert not letter_can_be_cancelled(notification_status, notification_created_at)
|
||||
|
||||
|
||||
@freeze_time("2018-7-7 19:00:00")
|
||||
@pytest.mark.parametrize(
|
||||
"notification_created_at",
|
||||
[
|
||||
datetime(2018, 7, 6, 18, 0), # created yesterday after 1730
|
||||
datetime(2018, 7, 7, 12, 0), # created today before 1730
|
||||
],
|
||||
)
|
||||
@pytest.mark.skip(reason="Letters not part of release")
|
||||
def test_letter_cannot_be_cancelled_if_after_1730_and_letter_created_before_1730(
|
||||
notification_created_at,
|
||||
):
|
||||
notification_status = "created"
|
||||
|
||||
assert not letter_can_be_cancelled(notification_status, notification_created_at)
|
||||
|
||||
|
||||
@freeze_time("2018-7-7 15:00:00")
|
||||
@pytest.mark.skip(reason="Letters not part of release")
|
||||
def test_letter_cannot_be_cancelled_if_before_1730_and_letter_created_before_1730_yesterday():
|
||||
notification_status = "created"
|
||||
|
||||
assert not letter_can_be_cancelled(notification_status, datetime(2018, 7, 6, 14, 0))
|
||||
|
||||
|
||||
@freeze_time("2018-7-7 15:00:00")
|
||||
@pytest.mark.skip(reason="Letters not part of release")
|
||||
def test_letter_cannot_be_cancelled_if_before_1730_and_letter_created_after_1730_two_days_ago():
|
||||
notification_status = "created"
|
||||
|
||||
assert not letter_can_be_cancelled(notification_status, datetime(2018, 7, 5, 19, 0))
|
||||
|
||||
|
||||
@freeze_time("2018-7-7 19:00:00")
|
||||
@pytest.mark.parametrize(
|
||||
"notification_created_at",
|
||||
[
|
||||
datetime(2018, 7, 7, 18, 30),
|
||||
datetime(2018, 7, 7, 19, 0),
|
||||
],
|
||||
)
|
||||
def test_letter_can_be_cancelled_if_after_1730_and_letter_created_at_1730_today_or_later(
|
||||
notification_created_at,
|
||||
):
|
||||
notification_status = "created"
|
||||
|
||||
assert letter_can_be_cancelled(notification_status, notification_created_at)
|
||||
Reference in New Issue
Block a user