mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
Danny
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from unittest import mock
|
|
|
|
from openpilot.common.hardware.comma.modem import Modem, State
|
|
|
|
|
|
def _modem(params):
|
|
m = Modem()
|
|
m._read_param = lambda key: params.get(key, "") # ty: ignore[invalid-assignment] # test double
|
|
m._publish_state = lambda **kw: m.S.update(kw) # ty: ignore[invalid-assignment] # test double
|
|
m._atv = lambda cmd, pfx: '2,1' if cmd in ("AT+CREG?", "AT+CGREG?") else None # ty: ignore[invalid-assignment] # test double
|
|
return m
|
|
|
|
|
|
def test_searching_dials_when_enabled():
|
|
assert _modem({})._do_searching() == State.CONNECTING
|
|
|
|
|
|
def test_searching_does_not_dial_when_disabled():
|
|
m = _modem({"CellularDisabled": "1"})
|
|
assert m._do_searching() == State.SEARCHING
|
|
assert m.S["registration"] == "home"
|
|
|
|
|
|
def test_connected_hangs_up_when_disabled():
|
|
params = {}
|
|
m = _modem(params)
|
|
m._ppp = mock.MagicMock(has_exited=lambda: False)
|
|
m._poll = lambda: None
|
|
m._roaming_allowed = False # as read during init
|
|
with mock.patch("os.path.exists", return_value=True):
|
|
assert m._do_connected() == State.CONNECTED
|
|
params["CellularDisabled"] = "1"
|
|
assert m._do_connected() == State.DISCONNECTING
|