darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 07:35:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add date helper methods to be used as date range filters for notification counts

Browse Source
This commit is contained in:
Imdad Ahad
2017-01-27 12:21:28 +00:00
parent ed52f41039
commit 781bef557a
2 changed files with 37 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/app/test_utils.py Normal file
View File

@@ -0,0 +1,25 @@
from datetime import datetime
import pytest
from app.utils import (
get_midnight_for_date,
get_midnight_for_day_before
)
@pytest.mark.parametrize('date, expected_date', [
(datetime(2016, 1, 15, 0, 30), datetime(2016, 1, 15, 0, 0)),
(datetime(2016, 1, 15, 0, 0), datetime(2016, 1, 15, 0, 0)),
(datetime(2016, 1, 15, 11, 59), datetime(2016, 1, 15, 0, 0)),
])
def test_get_midnight_for_today_returns_expected_date(date, expected_date):
assert get_midnight_for_date(date) == expected_date
@pytest.mark.parametrize('date, expected_date', [
(datetime(2016, 1, 15, 0, 30), datetime(2016, 1, 14, 0, 0)),
(datetime(2016, 1, 15, 0, 0), datetime(2016, 1, 14, 0, 0)),
(datetime(2016, 1, 15, 11, 59), datetime(2016, 1, 14, 0, 0)),
])
def test_get_midnight_for_day_before_returns_expected_date(date, expected_date):
assert get_midnight_for_day_before(date) == expected_date