mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-13 13:22:22 +08:00
firestarsdog
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
import importlib.util
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
MODULE_PATH = Path(__file__).resolve().parents[1] / "lib" / "starpilot_visuals.py"
|
|
SPEC = importlib.util.spec_from_file_location("starpilot_visuals_under_test", MODULE_PATH)
|
|
MODULE = importlib.util.module_from_spec(SPEC)
|
|
assert SPEC is not None and SPEC.loader is not None
|
|
SPEC.loader.exec_module(MODULE)
|
|
lead_indicator_enabled = MODULE.lead_indicator_enabled
|
|
|
|
|
|
class FakeParams:
|
|
def __init__(self, values=None):
|
|
self.values = values or {}
|
|
|
|
def get(self, key, block=False, return_default=False, encoding=None, default=None):
|
|
if key not in self.values:
|
|
return None
|
|
v = self.values[key]
|
|
return v if isinstance(v, bytes) else str(v).encode()
|
|
|
|
def get_bool(self, key, block=False, default=False):
|
|
value = self.values.get(key, default)
|
|
if isinstance(value, bool):
|
|
return value
|
|
return str(value or "").strip() in ("1", "true", "True")
|
|
|
|
|
|
class TestStarPilotVisuals(unittest.TestCase):
|
|
def test_lead_indicator_enabled_by_default(self):
|
|
self.assertTrue(lead_indicator_enabled(FakeParams()))
|
|
|
|
def test_hide_lead_marker_disables(self):
|
|
self.assertFalse(lead_indicator_enabled(FakeParams({"HideLeadMarker": True})))
|
|
|
|
def test_hide_by_default_returns_false_when_unset(self):
|
|
self.assertFalse(lead_indicator_enabled(FakeParams(), hide_by_default=True))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|