athenad test: use helper function (#29217)

pull/29225/head
Shane Smiskol 2 years ago committed by GitHub
parent 24bfb2b44b
commit b53bf00555
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 8
      selfdrive/athena/tests/test_athenad_ping.py

@ -9,6 +9,7 @@ from unittest.mock import MagicMock
from common.params import Params from common.params import Params
from common.timeout import Timeout from common.timeout import Timeout
from selfdrive.athena import athenad from selfdrive.athena import athenad
from selfdrive.manager.helpers import write_onroad_params
from system.hardware import TICI from system.hardware import TICI
@ -37,9 +38,6 @@ class TestAthenadPing(unittest.TestCase):
def _received_ping(self) -> bool: def _received_ping(self) -> bool:
return self._get_ping_time() is not None return self._get_ping_time() is not None
def _set_onroad(self, onroad: bool) -> None:
self.params.put_bool("IsOnroad", onroad)
@classmethod @classmethod
def setUpClass(cls) -> None: def setUpClass(cls) -> None:
cls.params = Params() cls.params = Params()
@ -101,12 +99,12 @@ class TestAthenadPing(unittest.TestCase):
@unittest.skipIf(not TICI, "only run on desk") @unittest.skipIf(not TICI, "only run on desk")
def test_offroad(self) -> None: def test_offroad(self) -> None:
self._set_onroad(False) write_onroad_params(False, self.params)
self.assertTimeout(100) # expect approx 90s self.assertTimeout(100) # expect approx 90s
@unittest.skipIf(not TICI, "only run on desk") @unittest.skipIf(not TICI, "only run on desk")
def test_onroad(self) -> None: def test_onroad(self) -> None:
self._set_onroad(True) write_onroad_params(True, self.params)
self.assertTimeout(30) # expect 20-30s self.assertTimeout(30) # expect 20-30s

Loading…
Cancel
Save