import pytest

from openpilot.starpilot.common import cpu_throttle


def test_offline_cpu_placeholders_do_not_hide_sustained_load(monkeypatch):
  monkeypatch.setattr(cpu_throttle, "_online_cpu_count", lambda: 4)

  usage = cpu_throttle._online_cpu_usage([80, 80, 80, 80, 0, 0, 0, 0])
  average = sum(usage) / len(usage)

  assert usage == [80.0, 80.0, 80.0, 80.0]
  assert cpu_throttle._compute_throttle_factor(average, 0) == pytest.approx(1.75)


@pytest.mark.parametrize(("online_spec", "expected"), (
  ("0-3", 4),
  ("0-3,6-7", 6),
))
def test_online_cpu_count_parser(monkeypatch, tmp_path, online_spec, expected):
  online_path = tmp_path / "online"
  online_path.write_text(online_spec, encoding="utf-8")
  monkeypatch.setattr(cpu_throttle, "Path", lambda _: online_path)
  cpu_throttle._online_cpu_count.cache_clear()

  assert cpu_throttle._online_cpu_count() == expected