84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from booking_ingestion.excel import BookingExcelError
|
|
from booking_ingestion.excel_review import (
|
|
BookingDraftItem,
|
|
BookingDraftPage,
|
|
BookingDraftSummary,
|
|
validate_draft_id,
|
|
validate_review_item_id,
|
|
validate_review_item_ids,
|
|
validate_review_quantity,
|
|
validate_review_room_type,
|
|
)
|
|
|
|
|
|
class BookingExcelReviewContractTests(unittest.TestCase):
|
|
def test_review_inputs_are_strict_and_normalized(self) -> None:
|
|
draft_id = "bookingdraft-" + "a" * 32
|
|
self.assertEqual(validate_draft_id(draft_id), draft_id)
|
|
self.assertEqual(validate_review_item_id(7), 7)
|
|
self.assertEqual(validate_review_item_ids([7, 8]), (7, 8))
|
|
self.assertEqual(validate_review_room_type(" U-TWN\nVIP "), "U-TWN VIP")
|
|
self.assertEqual(validate_review_quantity(2), 2)
|
|
|
|
for invalid in ("draft-1", "", None):
|
|
with self.subTest(draft_id=invalid), self.assertRaises(BookingExcelError):
|
|
validate_draft_id(invalid)
|
|
for invalid in (True, 0, -1, "2"):
|
|
with self.subTest(item_id=invalid), self.assertRaises(BookingExcelError):
|
|
validate_review_item_id(invalid)
|
|
for invalid in (None, [], [1, 1], [True], list(range(1, 52))):
|
|
with self.subTest(item_ids=invalid), self.assertRaises(BookingExcelError):
|
|
validate_review_item_ids(invalid)
|
|
for invalid in (True, 0, 10_000, 1.5, "2"):
|
|
with self.subTest(quantity=invalid), self.assertRaises(BookingExcelError):
|
|
validate_review_quantity(invalid)
|
|
|
|
def test_page_contract_exposes_review_state_without_hotel_detail(self) -> None:
|
|
draft_id = "bookingdraft-" + "b" * 32
|
|
summary = BookingDraftSummary(
|
|
draft_id=draft_id,
|
|
filename="Booking.xlsx",
|
|
status="reviewing",
|
|
source_rows=1,
|
|
worksheet_count=1,
|
|
distinct_group_codes=1,
|
|
extracted_items=1,
|
|
confirmed_items=0,
|
|
pending_items=1,
|
|
deleted_items=0,
|
|
confirmed_room_quantity=0,
|
|
created_at=None,
|
|
updated_at=None,
|
|
)
|
|
item = BookingDraftItem(
|
|
item_id=3,
|
|
draft_id=draft_id,
|
|
worksheet="Booking",
|
|
row_no=9,
|
|
item_no=1,
|
|
tour_code="TOUR-1",
|
|
room_type_raw="拼จU-4ค24",
|
|
room_type_code=None,
|
|
quantity=2,
|
|
review_status="pending",
|
|
automatic=False,
|
|
source_fragment="【拼จU-4ค24】 2",
|
|
updated_at=None,
|
|
)
|
|
|
|
payload = BookingDraftPage(summary, (item,), 1, 50, 0).to_dict()
|
|
|
|
self.assertEqual(payload["summary"]["pending_items"], 1)
|
|
self.assertEqual(payload["items"][0]["tour_code"], "TOUR-1")
|
|
self.assertIsNone(payload["items"][0]["room_type"])
|
|
self.assertEqual(payload["items"][0]["quantity"], 2)
|
|
self.assertNotIn("hotel_raw", payload["items"][0])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(verbosity=2)
|