from openpilot.system.hardware import usb


def test_chestnut_present(tmp_path, monkeypatch):
  monkeypatch.setattr(usb, "USB_DEVICES_PATH", tmp_path)
  device = tmp_path / "1-1"
  device.mkdir()
  (device / "idVendor").write_text("add1\n")
  (device / "idProduct").write_text("0001\n")

  assert usb.chestnut_present()


def test_chestnut_absent_for_other_usb_device(tmp_path, monkeypatch):
  monkeypatch.setattr(usb, "USB_DEVICES_PATH", tmp_path)
  device = tmp_path / "1-1"
  device.mkdir()
  (device / "idVendor").write_text("18d1\n")
  (device / "idProduct").write_text("4ee7\n")

  assert not usb.chestnut_present()