50 lines
2.1 KiB
Python
50 lines
2.1 KiB
Python
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_static_assets_resolve_from_the_release_directory(self):
|
|
self.assertIn('os.chdir(str(ROOT))', Path(__file__).parents[1].joinpath('server.py').read_text(encoding='utf-8'))
|
|
|
|
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()
|