Tirsvad
08ba464ec9
Day label and time parsing, next Tuesday 6pm lookup without list-position tricks, decision on Book Class / Join Waitlist / Booked / Waitlisted, confirmed clicks, console recorder and text screenshots, unit tests and container tests against local fixtures. Task: MIL-004#1 Task: MIL-004#2 Task: MIL-004#3 Task: MIL-004#4
98 lines
3.0 KiB
Python
98 lines
3.0 KiB
Python
"""@file test_booking.py
|
|
@brief Booking a class and booking the next Tuesday class, against a fake gym.
|
|
"""
|
|
|
|
import logging
|
|
|
|
import pytest
|
|
from tests.unit.fakes import FakeGym
|
|
from tests.unit.test_models import make_card
|
|
|
|
from snack_and_lift.application.booking import book_class, book_next_tuesday
|
|
from snack_and_lift.domain.errors import GymError, UnknownButtonError
|
|
from snack_and_lift.domain.models import Outcome
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("button_text", "outcome", "message"),
|
|
[
|
|
("Book Class", Outcome.NEW_BOOKING, "✓ Successfully booked: Spin Class on"),
|
|
("Join Waitlist", Outcome.NEW_WAITLIST, "✓ Joined waitlist for: Spin Class on"),
|
|
],
|
|
)
|
|
def test_book_class_clicks_and_confirms(
|
|
button_text: str, outcome: Outcome, message: str, caplog: pytest.LogCaptureFixture
|
|
) -> None:
|
|
card = make_card(button_text=button_text)
|
|
gym = FakeGym(cards=[card])
|
|
|
|
with caplog.at_level(logging.INFO):
|
|
result = book_class(gym, card)
|
|
|
|
assert result.outcome is outcome
|
|
assert gym.pressed == ["123"]
|
|
assert message in caplog.text
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("button_text", "outcome"),
|
|
[("Booked", Outcome.ALREADY_BOOKED), ("Waitlisted", Outcome.ALREADY_WAITLISTED)],
|
|
)
|
|
def test_book_class_does_not_click_when_already_done(button_text: str, outcome: Outcome) -> None:
|
|
gym = FakeGym()
|
|
|
|
result = book_class(gym, make_card(button_text=button_text))
|
|
|
|
assert result.outcome is outcome
|
|
assert gym.pressed == []
|
|
|
|
|
|
def test_book_class_raises_when_click_is_not_confirmed() -> None:
|
|
card = make_card()
|
|
gym = FakeGym(cards=[card], press_result="Book Class")
|
|
|
|
with pytest.raises(GymError, match="click not confirmed"):
|
|
book_class(gym, card)
|
|
|
|
|
|
def test_book_class_raises_when_waitlist_click_gives_booked() -> None:
|
|
card = make_card(button_text="Join Waitlist")
|
|
|
|
with pytest.raises(GymError, match="click not confirmed"):
|
|
book_class(FakeGym(cards=[card], press_result="Booked"), card)
|
|
|
|
|
|
def test_book_class_propagates_a_site_error() -> None:
|
|
with pytest.raises(GymError, match="Network down"):
|
|
book_class(FakeGym(cards=[make_card()], press_error="Network down"), make_card())
|
|
|
|
|
|
def test_book_class_raises_for_a_busy_button() -> None:
|
|
with pytest.raises(UnknownButtonError):
|
|
book_class(FakeGym(), make_card(button_text="Booking..."))
|
|
|
|
|
|
def test_book_next_tuesday_books_the_first_tuesday_class() -> None:
|
|
cards = [
|
|
make_card("Today (Wed, Aug 13)", card_id="1"),
|
|
make_card("Tue, Aug 19", card_id="2"),
|
|
make_card("Tue, Aug 26", card_id="3"),
|
|
]
|
|
gym = FakeGym(cards=cards)
|
|
|
|
result = book_next_tuesday(gym)
|
|
|
|
assert result is not None
|
|
assert result.card.card_id == "2"
|
|
assert gym.pressed == ["2"]
|
|
|
|
|
|
def test_book_next_tuesday_logs_when_there_is_no_class(caplog: pytest.LogCaptureFixture) -> None:
|
|
gym = FakeGym(cards=[make_card("Wed, Aug 13")])
|
|
|
|
with caplog.at_level(logging.INFO):
|
|
assert book_next_tuesday(gym) is None
|
|
|
|
assert "No Tue class at 18:00 found" in caplog.text
|
|
assert gym.pressed == []
|