mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-06-28 10:02:06 +08:00
firestar5683
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
from openpilot.starpilot.common import connect_server as cs
|
|
|
|
|
|
class FakeParams:
|
|
def __init__(self, values=None):
|
|
self.values = dict(values or {})
|
|
|
|
def get(self, key):
|
|
return self.values.get(key)
|
|
|
|
def put(self, key, value):
|
|
self.values[key] = value
|
|
|
|
def put_bool(self, key, value):
|
|
self.values[key] = b"1" if value else b"0"
|
|
|
|
def remove(self, key):
|
|
self.values.pop(key, None)
|
|
|
|
|
|
def test_sync_konik_dongle_id_preserves_stock_id_before_switching(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(cs.Paths, "persist_root", staticmethod(lambda: str(tmp_path)))
|
|
monkeypatch.setattr(cs, "use_konik_server", lambda: True)
|
|
monkeypatch.setattr(cs, "register", lambda **kwargs: "konik-dongle")
|
|
|
|
params = FakeParams({"DongleId": "stock-dongle"})
|
|
|
|
cs.sync_konik_dongle_id(params)
|
|
|
|
assert params.get("StockDongleId") == "stock-dongle"
|
|
assert params.get("KonikDongleId") == "konik-dongle"
|
|
assert params.get("DongleId") == "konik-dongle"
|
|
|
|
|
|
def test_sync_konik_dongle_id_restores_stock_id_from_persist(monkeypatch, tmp_path):
|
|
persist_root = tmp_path / "persist"
|
|
persisted_dongle_id_path = persist_root / "comma" / "dongle_id"
|
|
persisted_dongle_id_path.parent.mkdir(parents=True, exist_ok=True)
|
|
persisted_dongle_id_path.write_text("stock-dongle")
|
|
|
|
monkeypatch.setattr(cs.Paths, "persist_root", staticmethod(lambda: str(persist_root)))
|
|
monkeypatch.setattr(cs, "use_konik_server", lambda: False)
|
|
|
|
params = FakeParams({
|
|
"DongleId": "konik-dongle",
|
|
"KonikDongleId": "konik-dongle",
|
|
})
|
|
|
|
cs.sync_konik_dongle_id(params)
|
|
|
|
assert params.get("StockDongleId") == "stock-dongle"
|
|
assert params.get("DongleId") == "stock-dongle"
|
|
|
|
|
|
def test_sync_konik_dongle_id_skips_missing_stock_backup(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(cs.Paths, "persist_root", staticmethod(lambda: str(tmp_path)))
|
|
monkeypatch.setattr(cs, "use_konik_server", lambda: False)
|
|
|
|
params = FakeParams({
|
|
"DongleId": "konik-dongle",
|
|
"KonikDongleId": "konik-dongle",
|
|
})
|
|
|
|
cs.sync_konik_dongle_id(params)
|
|
|
|
assert params.get("DongleId") == "konik-dongle"
|
|
assert params.get("StockDongleId") is None
|
|
|
|
|
|
def test_prepare_konik_server_switch_clears_cached_konik_id():
|
|
params = FakeParams({"KonikDongleId": "konik-dongle"})
|
|
params_cache = FakeParams({"KonikDongleId": "konik-dongle"})
|
|
|
|
cs.prepare_konik_server_switch(True, params, params_cache)
|
|
|
|
assert params.get("UseKonikServer") == b"1"
|
|
assert params.get("KonikDongleId") is None
|
|
assert params_cache.get("KonikDongleId") is None
|
|
|
|
|
|
def test_prepare_konik_server_switch_clears_cached_stock_id():
|
|
params = FakeParams({"DongleId": "konik-dongle"})
|
|
params_cache = FakeParams({"DongleId": "konik-dongle"})
|
|
|
|
cs.prepare_konik_server_switch(False, params, params_cache)
|
|
|
|
assert params.get("UseKonikServer") == b"0"
|
|
assert params.get("DongleId") is None
|
|
assert params_cache.get("DongleId") is None
|