diff --git a/seatsio/domain.py b/seatsio/domain.py index 692e63d..4065ca3 100644 --- a/seatsio/domain.py +++ b/seatsio/domain.py @@ -268,6 +268,7 @@ def __init__(self, item_data): self.book_as_a_whole = item_data.get("bookAsAWhole") self.object_type = item_data.get("objectType") self.area_type = item_data.get("areaType") + self.table_type = item_data.get("tableType") self.left_neighbour = item_data.get('leftNeighbour') self.right_neighbour = item_data.get('rightNeighbour') self.distance_to_focal_point = item_data.get('distanceToFocalPoint') @@ -323,6 +324,7 @@ def __init__(self, item_data): self.book_as_a_whole = item_data.get("bookAsAWhole") self.object_type = item_data.get("objectType") self.area_type = item_data.get("areaType") + self.table_type = item_data.get("tableType") self.extra_data = item_data.get("extraData") self.is_accessible = item_data.get("isAccessible") self.is_companion_seat = item_data.get("isCompanionSeat") diff --git a/tests/reports/charts/testChartReports.py b/tests/reports/charts/testChartReports.py index 5a08443..3142c21 100644 --- a/tests/reports/charts/testChartReports.py +++ b/tests/reports/charts/testChartReports.py @@ -99,6 +99,7 @@ def test_reportItemPropertiesForTable(self, update_chart, get_report): report_item = report.get("T1")[0] assert_that(report_item.num_seats).is_equal_to(6) assert_that(report_item.book_as_a_whole).is_false() + assert_that(report_item.table_type).is_equal_to("bookBySeat") @parameterized.expand([ [ diff --git a/tests/reports/events/testEventReports.py b/tests/reports/events/testEventReports.py index 1d8e991..8dc93f6 100644 --- a/tests/reports/events/testEventReports.py +++ b/tests/reports/events/testEventReports.py @@ -157,6 +157,7 @@ def test_reportItemPropertiesForTable(self): report_item = report.get("T1")[0] assert_that(report_item.num_seats).is_equal_to(6) assert_that(report_item.book_as_a_whole).is_false() + assert_that(report_item.table_type).is_equal_to("bookByTable") def testByStatus(self): chart_key = self.create_test_chart()