feat(map): add cached public station navigation

This commit is contained in:
lingniu
2026-08-12 10:50:06 +08:00
parent 357dd490e9
commit 6ef98d03c2
13 changed files with 780 additions and 4 deletions
+73
View File
@@ -0,0 +1,73 @@
import assert from 'node:assert/strict';
import fs from 'node:fs';
import vm from 'node:vm';
const source = fs.readFileSync(new URL('../app.js', import.meta.url), 'utf8');
const sandbox = {
console,
document: { addEventListener() {}, getElementById() { return null; } },
window: {},
pinyinPro: { pinyin(value) { return value === '广州合作站' ? 'guang zhou he zuo zhan' : value; } },
navigator: {},
};
sandbox.globalThis = sandbox;
vm.runInNewContext(source, sandbox, { filename: 'app.js' });
assert.equal(sandbox.stationSearchScore({ name: '广州合作站' }, 'gz'), 2);
assert.equal(sandbox.stationSearchScore({ address: '广东省广州市黄埔区' }, '黄埔'), 3);
assert.equal(sandbox.stationSearchScore({ name: '广州合作站' }, '北京'), 0);
assert.equal(sandbox.hasCoordinates({ longitude: 113.2, latitude: 23.1 }), true);
assert.equal(sandbox.hasCoordinates({ longitude: 'bad', latitude: 23.1 }), false);
vm.runInNewContext(`
state.stations = [
{ id: 'partner', name: '广州合作站', cooperative: true },
{ id: 'external', name: '广州外部站', cooperative: false }
];
globalThis.allStationIds = filteredStations().map(station => station.id);
state.stationFilter = 'partner';
globalThis.partnerStationIds = filteredStations().map(station => station.id);
`, sandbox);
assert.equal(sandbox.allStationIds.join(','), 'partner,external');
assert.equal(sandbox.partnerStationIds.join(','), 'partner');
vm.runInNewContext(`
state.stations = [
{ id: 'gd-hp', name: '黄埔合作站', province: '广东省', city: '广州市', district: '黄埔区', longitude: 113.3, latitude: 23.1, cooperative: true },
{ id: 'gd-nh', name: '南海合作站', province: '广东省', city: '佛山市', district: '南海区', longitude: 113.1, latitude: 23.0, cooperative: true },
{ id: 'zj-jx', name: '嘉兴合作站', province: '浙江省', city: '嘉兴市', district: '秀洲区', longitude: 120.7, latitude: 30.7, cooperative: true }
];
globalThis.hierarchy = [stationHierarchyLevel(4.8), stationHierarchyLevel(8), stationHierarchyLevel(13)];
globalThis.provinceNodeCount = buildStationNodes(filteredStations(), 'province').length;
globalThis.cityNodeCount = buildStationNodes(filteredStations(), 'city').length;
globalThis.stationNodeCount = buildStationNodes(filteredStations(), 'station').length;
globalThis.locationOptionCount = stationLocationOptions().length;
`, sandbox);
assert.equal(sandbox.hierarchy.join(','), 'province,city,station');
assert.equal(sandbox.provinceNodeCount, 2);
assert.equal(sandbox.cityNodeCount, 3);
assert.equal(sandbox.stationNodeCount, 3);
assert.equal(sandbox.locationOptionCount, 8);
vm.runInNewContext(`
globalThis.inferredDistrict = stationDistrictName({ province: '浙江省', city: '嘉兴市', address: '嘉兴市海盐县海盐经济开发区' });
state.locationFilter = { province: '', city: '', district: '' };
state.query = '广州';
globalThis.queryMatches = stationMatchesQuery({ name: '广州合作站' });
globalThis.queryMisses = stationMatchesQuery({ name: '杭州合作站' });
`, sandbox);
assert.equal(sandbox.inferredDistrict, '海盐县');
assert.equal(sandbox.queryMatches, true);
assert.equal(sandbox.queryMisses, false);
assert.ok(Math.abs(sandbox.distanceKm([113, 23], [114, 23]) - 102.4) < 1);
assert.match(source, /zoom < 11\) return 'city'/);
assert.match(source, /state\.userLocation && left\.kind === 'station'/);
assert.doesNotMatch(source, /rankSecondaryTab|setRank\(/);
assert.match(source, /stationSearchSuggestions/);
assert.match(source, /handleStationSearchInput/);
assert.match(source, /window\.setTimeout\(\(\) => \{ state\.query = value/);
assert.match(source, /focusSearchResults\(\)/);
assert.match(source, /visibleOnly && !normalize\(state\.query\)/);
assert.match(source, /station-list-partner-tag/);
assert.match(source, /node\.cooperative \? 'H₂ · 合作' : ''/);
assert.doesNotMatch(source, /在线 \$\{formatNumber\(node\.online\)\}/);
assert.doesNotMatch(source, /monthlyHydrogenKg|totalHydrogenKg|vehicle/);
assert.match(source, /uri\.amap\.com\/navigation/);
console.log('station navigation app tests: ok');
+46
View File
@@ -0,0 +1,46 @@
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()