jgaunt/SouthwestCheckin
1
0
Fork 0
mirror of https://github.com/pyro2927/SouthwestCheckin.git synced 2025-12-06 01:13:19 +00:00
Code Issues Releases Wiki Activity

Adding tests around notifications

Browse Source
This commit is contained in:
Joseph Pintozzi
2019-02-02 20:02:24 -08:00
parent 875ceef805
commit 27122855ee
2 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
checkin_test.py
View File

@@ -3,7 +3,8 @@ import json
import pytest
import requests
from datetime import datetime, timedelta
from pytz import timezone
from pytz import timezone, utc
from tzlocal import get_localzone
def template_time(file_name):
in_string = open(file_name, 'r').read()
@@ -12,7 +13,6 @@ def template_time(file_name):
in_string = in_string.replace('TIME_HERE', t.strftime('%H:%M'))
return in_string
def test_checkin(requests_mock):
requests_mock.register_uri('GET', '/api/mobile-misc/v1/mobile-misc/page/view-reservation/XXXX?first-name=John&last-name=Smith', text=template_time('fixtures/view-reservation.json'))
requests_mock.register_uri('GET', '/api/mobile-air-operations/v1/mobile-air-operations/page/check-in/XXXX?first-name=John&last-name=Smith', text=template_time('fixtures/checkin-get.json'))
@@ -38,3 +38,23 @@ def test_openflights_api():
tzresult = requests.post("https://openflights.org/php/apsearch.php", tzrequest)
airport_tz = timezone(json.loads(tzresult.text)['airports'][0]['tz_id'])
assert airport_tz.zone == "America/Los_Angeles"
def test_notifications(requests_mock, mocker):
requests_mock.register_uri('GET', '/api/mobile-air-operations/v1/mobile-air-operations/page/check-in/XXXX?first-name=John&last-name=Smith', text=template_time('fixtures/checkin-get.json'))
data = template_time('fixtures/checkin-post.json')
requests_mock.register_uri('POST', '/api/mobile-air-operations/v1/mobile-air-operations/page/check-in', text=data)
data = json.loads(data)
requests_mock.register_uri('POST', '/php/apsearch.php', text=template_time('fixtures/openflights.json'))
mocked_checkin = mocker.patch('checkin.send_notification')
t = datetime.now(utc).astimezone(get_localzone()) + timedelta(minutes=5)
try:
checkin.schedule_checkin(t, 'XXXX', 'John', 'Smith', None, None)
checkin.send_notification.assert_not_called()
checkin.schedule_checkin(t, 'XXXX', 'John', 'Smith', 'test@example.com', None)
checkin.send_notification.assert_called_once_with(data['checkInConfirmationPage'], emailaddr='test@example.com')
checkin.send_notification.reset_mock()
checkin.schedule_checkin(t, 'XXXX', 'John', 'Smith', None, '1234567890')
checkin.send_notification.assert_called_once_with(data['checkInConfirmationPage'], mobilenum='1234567890')
except:
pytest.fail("Error checking in")

1
requirements.txt
View File

@@ -1,6 +1,7 @@
datetime
docopts
pytest
pytest-mock
python-dateutil
pytz
requests