feat: sync latest ARR implementation
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user