Skip to content

Commit

Permalink
update reversations tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Der-Henning committed May 6, 2023
1 parent d221472 commit f1d7a9c
Showing 1 changed file with 11 additions and 17 deletions.
28 changes: 11 additions & 17 deletions tests/test_reservations.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,52 +13,46 @@ def reservations():
return Reservations(mock_client)


@pytest.fixture
def item():
return Item({
"item_id": "123",
"display_name": "Test Item",
"items_available": 1
})


def test_reserve(reservations):
def test_reserve(reservations: Reservations):
reservations.reserve("123", "Test Item")
assert len(reservations.reservation_query) == 1


def test_make_orders(reservations, item):
def test_make_orders(reservations: Reservations, tgtg_item: dict):
callback_mock = MagicMock()
reservations.reserve("123", "Test Item")
reservations.make_orders({"123": item}, callback_mock)
reservations.make_orders({"123": Item(tgtg_item)}, callback_mock)
assert len(reservations.active_orders) == 1
assert len(reservations.reservation_query) == 0
callback_mock.assert_called_once_with(
Reservation("123", 1, "Test Item")
)


def test_update_active_orders(reservations):
def test_update_active_orders(reservations: Reservations):
order = Order("1", "123", 1, "Test Item")
reservations.active_orders = [order]
reservations.update_active_orders()
assert len(reservations.active_orders) == 1
reservations.client.get_order_status.return_value = {"state": "CANELLED"}
reservations.update_active_orders()
assert len(reservations.active_orders) == 0


def test_cancel_order(reservations):
def test_cancel_order(reservations: Reservations):
order = Order("1", "123", 1, "Test Item")
reservations.active_orders = [order]
reservations.cancel_order(order)


def test_cancel_all_orders(reservations):
def test_cancel_all_orders(reservations: Reservations):
order1 = Order("1", "123", 1, "Test Item 1")
order2 = Order("2", "123", 2, "Test Item 2")
reservations.active_orders = [order1, order2]
reservations.cancel_all_orders()


def test_get_favorites(reservations, tgtg_item):
def test_get_favorites(reservations: Reservations, tgtg_item: dict):
reservations.client.get_favorites.return_value = [tgtg_item]
favorites = reservations.get_favorites()
assert len(favorites) == 1
Expand All @@ -67,7 +61,7 @@ def test_get_favorites(reservations, tgtg_item):
assert favorites[0].items_available == tgtg_item.get("items_available")


def test_create_order(reservations):
def test_create_order(reservations: Reservations):
reservations.client.create_order.return_value = {"id": "1"}
reservation = Reservation("123", 1, "Test Item")
reservations._create_order(reservation)
Expand Down

0 comments on commit f1d7a9c

Please sign in to comment.