#!/usr/bin/env python3
import json
import time
import unittest
import subprocess

import cereal.messaging as messaging
from system.hardware import TICI
from selfdrive.manager.process_config import managed_processes


class TestRawgpsd(unittest.TestCase):
  @classmethod
  def setUpClass(cls):
    if not TICI:
      raise unittest.SkipTest

  def tearDown(self):
    managed_processes['rawgpsd'].stop()

  def test_startup_time(self):
    for _ in range(5):
      sm = messaging.SubMaster(['qcomGnss'])
      managed_processes['rawgpsd'].start()

      start_time = time.monotonic()
      for __ in range(10):
        sm.update(1 * 1000)
        if sm.updated['qcomGnss']:
          break
      assert sm.rcv_frame['qcomGnss'] > 0, "rawgpsd didn't start outputting messages in time"

      et = time.monotonic() - start_time
      assert et < 5, f"rawgpsd took {et:.1f}s to start"
      managed_processes['rawgpsd'].stop()

  def test_turns_off_gnss(self):
    for s in (0.1, 0.5, 1, 5):
      managed_processes['rawgpsd'].start()
      time.sleep(s)
      managed_processes['rawgpsd'].stop()

      ls = subprocess.check_output("mmcli -m any --location-status --output-json", shell=True, encoding='utf-8')
      loc_status = json.loads(ls)
      assert set(loc_status['modem']['location']['enabled']) <= {'3gpp-lac-ci'}


if __name__ == "__main__":
  unittest.main()