mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-14 01:02:09 -05:00
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
from datetime import date, datetime
|
|
|
|
import pytest
|
|
from freezegun import freeze_time
|
|
|
|
from app.enums import ServicePermissionType
|
|
from app.utils import (
|
|
get_midnight_in_utc,
|
|
get_public_notify_type_text,
|
|
midnight_n_days_ago,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"date, expected_date",
|
|
[
|
|
(datetime(2016, 1, 15, 0, 30), datetime(2016, 1, 15, 0, 0)),
|
|
(datetime(2016, 6, 15, 0, 0), datetime(2016, 6, 15, 0, 0)),
|
|
(datetime(2016, 9, 15, 11, 59), datetime(2016, 9, 15, 0, 0)),
|
|
# works for both dates and datetimes
|
|
(date(2016, 1, 15), datetime(2016, 1, 15, 0, 0)),
|
|
(date(2016, 6, 15), datetime(2016, 6, 15, 0, 0)),
|
|
],
|
|
)
|
|
def test_get_midnight_in_utc_returns_expected_date(date, expected_date):
|
|
assert get_midnight_in_utc(date) == expected_date
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"current_time, arg, expected_datetime",
|
|
[
|
|
# winter
|
|
("2018-01-10 23:59", 1, datetime(2018, 1, 9, 0, 0)),
|
|
("2018-01-11 00:00", 1, datetime(2018, 1, 10, 0, 0)),
|
|
# bst switchover at 1am 25th
|
|
("2018-03-25 10:00", 1, datetime(2018, 3, 24, 0, 0)),
|
|
("2018-03-26 10:00", 1, datetime(2018, 3, 25, 0, 0)),
|
|
("2018-03-27 10:00", 1, datetime(2018, 3, 26, 0, 0)),
|
|
# summer
|
|
("2018-06-05 10:00", 1, datetime(2018, 6, 4, 0, 0)),
|
|
# zero days ago
|
|
("2018-01-11 00:00", 0, datetime(2018, 1, 11, 0, 0)),
|
|
("2018-06-05 10:00", 0, datetime(2018, 6, 5, 0, 0)),
|
|
],
|
|
)
|
|
def test_midnight_n_days_ago(current_time, arg, expected_datetime):
|
|
with freeze_time(current_time):
|
|
assert midnight_n_days_ago(arg) == expected_datetime
|
|
|
|
|
|
def test_get_public_notify_type_text():
|
|
assert (
|
|
get_public_notify_type_text(ServicePermissionType.UPLOAD_DOCUMENT) == "document"
|
|
)
|