import importlib.util from pathlib import Path import unittest from unittest.mock import patch SPEC = importlib.util.spec_from_file_location("station_navigation_server", Path(__file__).parents[1] / "server.py") server = importlib.util.module_from_spec(SPEC) SPEC.loader.exec_module(server) class StationNavigationTest(unittest.TestCase): def test_directory_is_a_positive_allowlist_without_hydrogen_data(self): source = [{ "id": "GD-1", "name": "广州合作站", "province": "广东省", "city": "广州市", "district": "黄埔区", "address": "开源大道1号", "longitude": 113.2, "latitude": 23.1, "cooperative": True, "monthlyHydrogenKg": 88.8, "totalHydrogenKg": 999.9, "vehicleCount": 20, }] with patch.object(server, "_post_open_platform", return_value=source): result = server._load_station_directory() self.assertEqual(result["summary"], {"totalStations": 1, "cooperativeStations": 1}) station = result["stations"][0] self.assertEqual(station["name"], "广州合作站") self.assertNotIn("monthlyHydrogenKg", station) self.assertNotIn("totalHydrogenKg", station) self.assertNotIn("vehicleCount", station) def test_static_assets_are_fingerprinted(self): self.assertRegex(server.STATIC_ASSET_VERSION, r"^[0-9a-f]{16}$") def test_default_cache_window_is_two_minutes(self): self.assertEqual(server.STATION_CACHE_SECONDS, 120) def test_cache_reuses_directory_snapshot_within_window(self): calls = [] with patch.object(server, "_cache", {}): first = server._cached("station-directory", 120, lambda: calls.append("load") or {"version": 1}) second = server._cached("station-directory", 120, lambda: calls.append("load") or {"version": 2}) self.assertEqual(calls, ["load"]) self.assertEqual(first, second) if __name__ == "__main__": unittest.main()