feat: sync latest ARR implementation

This commit is contained in:
Wyndham ARR
2026-07-31 15:11:42 +08:00
parent d6f8a747fa
commit bf7939dd1a
185 changed files with 17527 additions and 2260 deletions

View File

@@ -5,6 +5,7 @@ import os
import shutil
import tempfile
import unittest
from dataclasses import replace
from datetime import date
from pathlib import Path
@@ -37,6 +38,62 @@ def node_binary() -> str:
@unittest.skipUnless(artifact_tool_available(), "artifact-tool dependency is not installed")
class CompanyReportIntegrationTests(unittest.TestCase):
def test_missing_and_unmatched_group_codes_export_blank_booking_room(self):
node = node_binary()
if not node:
self.skipTest("Node.js is unavailable")
class CapturingBuilder:
def __init__(self, delegate):
self.delegate = delegate
self.reports = []
def build(self, report, work_dir):
self.reports.append(report)
return self.delegate.build(report, work_dir)
with tempfile.TemporaryDirectory(prefix="company-report-integration-") as temp_dir:
project_root = Path(temp_dir) / "project"
output_root = project_root / "outputs" / "company_reports"
project_root.mkdir(parents=True)
repository = FakeRepository(
replace(
synthetic_snapshot(include_invalid_hana=True),
room_items=(),
group_parse_versions={},
)
)
builder = CapturingBuilder(
ArtifactToolBuilder(BUILDER_SCRIPT, node_binary=node, timeout_seconds=90)
)
publisher = AtomicReportPublisher(project_root, output_root)
service = CompanyReportService(
repository,
builder,
publisher,
output_root / ".staging",
)
result = service.run(
RunRequest(2026, 7, date(2026, 7, 10), ("QBD", "HanaTour"))
)
self.assertEqual(result.status, "success")
self.assertEqual([item.row_count for item in result.companies], [1, 1])
self.assertTrue(all(item.errors == () for item in result.companies))
self.assertEqual(len(builder.reports), 2)
qbd_report, hana_report = builder.reports
self.assertEqual(qbd_report.periods[0].rows[0].res_comment, "SYN-GROUP-A")
self.assertEqual(qbd_report.periods[0].rows[0].booking_room, "")
self.assertEqual(hana_report.periods[0].rows[0].res_comment, "")
self.assertEqual(hana_report.periods[0].rows[0].booking_room, "")
self.assertTrue(
(output_root / "2026" / "07" / "QBD-July-2026.xlsx").is_file()
)
self.assertTrue(
(output_root / "2026" / "07" / "HanaTour-July-2026.xlsx").is_file()
)
def test_same_snapshot_rerun_rebuilds_full_month_with_equal_semantics(self):
node = node_binary()
if not node: