mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-04 15:13:58 +08:00
firestarsdog
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
from openpilot.starpilot.common.maps_download_progress import (
|
|
estimate_download_bytes,
|
|
estimate_file_eta_seconds,
|
|
estimate_eta_seconds,
|
|
load_maps_storage_cache,
|
|
load_size_cache,
|
|
selection_key,
|
|
storage_bytes,
|
|
)
|
|
|
|
|
|
def test_storage_bytes_and_selection_key(tmp_path):
|
|
maps_path = tmp_path / "maps"
|
|
maps_path.mkdir()
|
|
(maps_path / "first.bin").write_bytes(b"1234")
|
|
(maps_path / "nested").mkdir()
|
|
(maps_path / "nested" / "second.bin").write_bytes(b"123456")
|
|
|
|
assert storage_bytes(maps_path) == 10
|
|
assert selection_key("us-ca,us-tx,us-ca") == "us-ca,us-tx"
|
|
|
|
|
|
def test_download_size_and_eta_estimates():
|
|
assert estimate_download_bytes(100, total_files=10, downloaded_files=2) == 500
|
|
assert estimate_download_bytes(0, total_files=10, downloaded_files=2) == 0
|
|
assert estimate_eta_seconds(500, 100, 100) == 4
|
|
assert estimate_eta_seconds(500, 500, 100) == 0
|
|
|
|
|
|
def test_load_size_cache_rejects_invalid_values():
|
|
assert load_size_cache(b'{"us-ca":{"downloadBytes":123}}')["us-ca"]["downloadBytes"] == 123
|
|
assert load_size_cache("not json") == {}
|
|
assert load_size_cache("[]") == {}
|
|
|
|
|
|
def test_storage_cache_migrates_legacy_selection_without_claiming_storage_total():
|
|
cache = load_maps_storage_cache('{"country:CA":{"downloadBytes":123,"totalFiles":4}}')
|
|
|
|
assert cache.storage_bytes is None
|
|
assert cache.maps_present is None
|
|
assert cache.selection_estimate_bytes("country:CA") == 123
|
|
assert cache.selection_total_files("country:CA") == 4
|
|
|
|
migrated = load_maps_storage_cache(cache.to_json())
|
|
assert migrated.storage_bytes is None
|
|
assert migrated.selection_estimate_bytes("country:CA") == 123
|
|
|
|
|
|
def test_storage_cache_records_selection_delta_not_aggregate_storage():
|
|
cache = load_maps_storage_cache("")
|
|
cache.reconcile(10_000)
|
|
cache.reconcile(
|
|
18_000,
|
|
selection_key="country:CA",
|
|
baseline_storage_bytes=10_000,
|
|
total_files=80,
|
|
updated_at="2026-08-31T00:00:00",
|
|
)
|
|
|
|
assert cache.storage_bytes == 18_000
|
|
assert cache.maps_present is True
|
|
assert cache.selection_estimate_bytes("country:CA") == 8_000
|
|
assert cache.selection_total_files("country:CA") == 80
|
|
|
|
cache.reconcile(
|
|
18_000,
|
|
selection_key="country:CA",
|
|
baseline_storage_bytes=18_000,
|
|
total_files=80,
|
|
updated_at="2026-09-01T00:00:00",
|
|
)
|
|
|
|
assert cache.selection_estimate_bytes("country:CA") == 0
|
|
|
|
|
|
def test_storage_cache_can_be_cleared_or_marked_unknown_without_false_presence():
|
|
cache = load_maps_storage_cache("")
|
|
cache.reconcile(128)
|
|
cache.clear()
|
|
|
|
assert cache.storage_bytes == 0
|
|
assert cache.maps_present is False
|
|
|
|
cache.mark_unknown()
|
|
|
|
assert cache.storage_bytes is None
|
|
assert cache.maps_present is None
|
|
|
|
|
|
def test_file_eta_uses_mapd_file_progress_without_storage_scans():
|
|
assert estimate_file_eta_seconds(elapsed_seconds=10, total_files=10, downloaded_files=2) == 40
|
|
assert estimate_file_eta_seconds(elapsed_seconds=10, total_files=10, downloaded_files=0) == 0
|
|
assert estimate_file_eta_seconds(elapsed_seconds=10, total_files=2, downloaded_files=2) == 0
|