mirror of
https://github.com/pyro2927/SouthwestCheckin.git
synced 2026-01-06 01:33:14 +00:00
2e555d7bf7
modified: test_checkin.yml modified: test_checkin_without_passes.yml modified: test_cli.yml new file: test_generate_headers.yml modified: test_reservation_lookup.yml modified: ../../my_vcr.py
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
import json
|
|
import pytest
|
|
import requests
|
|
import southwest
|
|
import checkin
|
|
from datetime import datetime, timedelta
|
|
from .my_vcr import custom_vcr
|
|
from pytz import timezone, utc
|
|
from tzlocal import get_localzone
|
|
|
|
my_vcr = custom_vcr()
|
|
r = southwest.Reservation('XXXXXX', 'John', 'Smith')
|
|
|
|
@my_vcr.use_cassette()
|
|
def test_generate_headers():
|
|
print(southwest.Reservation.generate_headers())
|
|
|
|
@my_vcr.use_cassette()
|
|
def test_reservation_lookup():
|
|
print(r.notifications)
|
|
try:
|
|
r.lookup_existing_reservation()
|
|
except Exception:
|
|
pytest.fail("Error looking up reservation")
|
|
|
|
|
|
@my_vcr.use_cassette()
|
|
def test_checkin():
|
|
phone = southwest.Notifications.Phone('1234567890')
|
|
email = southwest.Notifications.Email('test@example.com')
|
|
r.notifications = [phone, email]
|
|
try:
|
|
r.checkin()
|
|
except Exception:
|
|
pytest.fail("Error checking in")
|
|
|
|
|
|
@my_vcr.use_cassette()
|
|
def test_checkin_without_passes():
|
|
phone = southwest.Notifications.Phone('1234567890')
|
|
email = southwest.Notifications.Email('test@example.com')
|
|
r.notifications = [phone, email]
|
|
try:
|
|
r.checkin()
|
|
except Exception:
|
|
pytest.fail("Error checking in")
|
|
|
|
|
|
@my_vcr.use_cassette()
|
|
def test_openflights_api():
|
|
assert southwest.timezone_for_airport('LAX').zone == "America/Los_Angeles"
|
|
|
|
|
|
@my_vcr.use_cassette()
|
|
def test_cli():
|
|
try:
|
|
checkin.auto_checkin('XXXXXX', 'John', 'Smith')
|
|
except Exception:
|
|
pytest.fail("cli error")
|