room_cal_ha/tests/test_sensors.py

"""Unit tests for sensor property logic.

Bypasses HA entity setup by using object.__new__ and injecting a mock
coordinator — valid because every tested property only reads self.coordinator.data.
"""
from datetime import date, datetime, time, timezone
from unittest.mock import MagicMock

import pytest

from custom_components.room_cal_ha.coordinator import CoordinatorData
from custom_components.room_cal_ha.models import ApiStatusData, ReservationData
from custom_components.room_cal_ha.sensor import (
  ApiCheckedAtSensor,
  ApiStatusSensor,
  LastRefreshSensor,
  ReservationDateTimeSensor,
  ReservationStrSensor,
  ReservationTimeSensor,
  WiFiKeySensor,
)
from custom_components.room_cal_ha.const import STORE_CURRENT, STORE_NEXT


# ---------------------------------------------------------------------------
# Helpers
# ---------------------------------------------------------------------------

def _coordinator(
  current=None,
  nxt=None,
  current_ts=None,
  next_ts=None,
  api_status=None,
):
  coord = MagicMock()
  coord.data = CoordinatorData(
    current=current or ReservationData(),
    next=nxt or ReservationData(),
    current_last_refresh=current_ts,
    next_last_refresh=next_ts,
    api_status=api_status,
  )
  return coord


def _sensor(cls, coordinator, slot=STORE_CURRENT, field="guest_name", **extra):
  """Create a sensor instance without running __init__."""
  s = object.__new__(cls)
  s.coordinator = coordinator
  s._slot = slot
  s._field = field
  for k, v in extra.items():
    setattr(s, k, v)
  return s


SAMPLE = ReservationData(
  guest_name="Alice",
  check_in_date=date(2026, 7, 1),
  check_in_time=time(14, 0),
  check_out_date=date(2026, 7, 5),
  check_out_time=time(12, 0),
  wifi_ssid="MyNet",
  wifi_key="hunter2",
  door_code="0472",
)


# ---------------------------------------------------------------------------
# ReservationStrSensor
# ---------------------------------------------------------------------------

def test_str_sensor_returns_field_value():
  coord = _coordinator(current=SAMPLE)
  s = _sensor(ReservationStrSensor, coord, field="guest_name")
  assert s.native_value == "Alice"


def test_str_sensor_reads_next_slot():
  nxt = ReservationData(guest_name="Bob")
  coord = _coordinator(nxt=nxt)
  s = _sensor(ReservationStrSensor, coord, slot=STORE_NEXT, field="guest_name")
  assert s.native_value == "Bob"


def test_str_sensor_none_when_field_is_none():
  res = ReservationData(guest_name=None)
  coord = _coordinator(current=res)
  s = _sensor(ReservationStrSensor, coord, field="guest_name")
  assert s.native_value is None


# ---------------------------------------------------------------------------
# ReservationTimeSensor
# ---------------------------------------------------------------------------

def test_time_sensor_returns_iso_string():
  coord = _coordinator(current=SAMPLE)
  s = _sensor(ReservationTimeSensor, coord, field="check_in_time")
  assert s.native_value == "14:00:00"


def test_time_sensor_none_when_time_is_none():
  res = ReservationData(check_in_time=None)
  coord = _coordinator(current=res)
  s = _sensor(ReservationTimeSensor, coord, field="check_in_time")
  assert s.native_value is None


# ---------------------------------------------------------------------------
# ReservationDateTimeSensor
# ---------------------------------------------------------------------------

def test_datetime_sensor_native_value_is_date():
  coord = _coordinator(current=SAMPLE)
  s = _sensor(ReservationDateTimeSensor, coord, field="check_in_date")
  s._time_field = "check_in_time"
  assert s.native_value == date(2026, 7, 1)


def test_datetime_sensor_attrs_contain_expected_keys():
  coord = _coordinator(current=SAMPLE)
  s = _sensor(ReservationDateTimeSensor, coord, field="check_in_date")
  s._time_field = "check_in_time"
  attrs = s.extra_state_attributes
  assert attrs is not None
  for key in ("datetime", "date", "weekday", "month", "time_abbr", "hour"):
    assert key in attrs, f"missing key: {key}"


def test_datetime_sensor_attrs_none_when_date_missing():
  res = ReservationData(check_in_date=None, check_in_time=time(14, 0))
  coord = _coordinator(current=res)
  s = _sensor(ReservationDateTimeSensor, coord, field="check_in_date")
  s._time_field = "check_in_time"
  assert s.extra_state_attributes is None


def test_datetime_sensor_attrs_none_when_time_missing():
  res = ReservationData(check_in_date=date(2026, 7, 1), check_in_time=None)
  coord = _coordinator(current=res)
  s = _sensor(ReservationDateTimeSensor, coord, field="check_in_date")
  s._time_field = "check_in_time"
  assert s.extra_state_attributes is None


# ---------------------------------------------------------------------------
# WiFiKeySensor
# ---------------------------------------------------------------------------

def test_wifi_key_sensor_native_value():
  coord = _coordinator(current=SAMPLE)
  s = _sensor(WiFiKeySensor, coord, field="wifi_key")
  assert s.native_value == "hunter2"


def test_wifi_key_sensor_qr_attribute():
  coord = _coordinator(current=SAMPLE)
  s = _sensor(WiFiKeySensor, coord, field="wifi_key")
  attrs = s.extra_state_attributes
  assert attrs is not None
  assert attrs["qr"] == "WIFI:S:MyNet;T:WPA;P:hunter2;;"


def test_wifi_key_sensor_no_attrs_when_ssid_missing():
  res = ReservationData(wifi_ssid=None, wifi_key="hunter2")
  coord = _coordinator(current=res)
  s = _sensor(WiFiKeySensor, coord, field="wifi_key")
  assert s.extra_state_attributes is None


def test_wifi_key_sensor_no_attrs_when_key_missing():
  res = ReservationData(wifi_ssid="MyNet", wifi_key=None)
  coord = _coordinator(current=res)
  s = _sensor(WiFiKeySensor, coord, field="wifi_key")
  assert s.extra_state_attributes is None


# ---------------------------------------------------------------------------
# LastRefreshSensor
# ---------------------------------------------------------------------------

def test_last_refresh_sensor_current():
  ts = datetime(2026, 6, 22, 10, 0, tzinfo=timezone.utc)
  coord = _coordinator(current_ts=ts)
  s = _sensor(LastRefreshSensor, coord)
  s._slot = STORE_CURRENT
  assert s.native_value == ts


def test_last_refresh_sensor_next():
  ts = datetime(2026, 6, 22, 12, 0, tzinfo=timezone.utc)
  coord = _coordinator(next_ts=ts)
  s = _sensor(LastRefreshSensor, coord)
  s._slot = STORE_NEXT
  assert s.native_value == ts


def test_last_refresh_sensor_none_before_any_fetch():
  coord = _coordinator()
  s = _sensor(LastRefreshSensor, coord)
  s._slot = STORE_CURRENT
  assert s.native_value is None


# ---------------------------------------------------------------------------
# ApiStatusSensor
# ---------------------------------------------------------------------------

def test_api_status_sensor_native_value():
  status = ApiStatusData(status="ok")
  coord = _coordinator(api_status=status)
  s = object.__new__(ApiStatusSensor)
  s.coordinator = coord
  assert s.native_value == "ok"


def test_api_status_sensor_none_when_no_status():
  coord = _coordinator(api_status=None)
  s = object.__new__(ApiStatusSensor)
  s.coordinator = coord
  assert s.native_value is None


def test_api_status_sensor_attrs_include_checks_and_stale():
  checks = {"caldav_connection": {"ok": True}, "calendar": {"ok": False, "reason": "not found"}}
  status = ApiStatusData(status="degraded", checks=checks, stale=True)
  coord = _coordinator(api_status=status)
  s = object.__new__(ApiStatusSensor)
  s.coordinator = coord

  attrs = s.extra_state_attributes
  assert attrs["caldav_connection"] == {"ok": True}
  assert attrs["calendar"] == {"ok": False, "reason": "not found"}
  assert attrs["stale"] is True


def test_api_status_sensor_attrs_include_checked_at_when_present():
  ts = datetime(2026, 6, 22, 10, 0, tzinfo=timezone.utc)
  status = ApiStatusData(status="ok", checked_at=ts)
  coord = _coordinator(api_status=status)
  s = object.__new__(ApiStatusSensor)
  s.coordinator = coord

  attrs = s.extra_state_attributes
  assert "checked_at" in attrs
  assert attrs["checked_at"] == ts.isoformat()


def test_api_status_sensor_attrs_none_when_no_status():
  coord = _coordinator(api_status=None)
  s = object.__new__(ApiStatusSensor)
  s.coordinator = coord
  assert s.extra_state_attributes is None


# ---------------------------------------------------------------------------
# ApiCheckedAtSensor
# ---------------------------------------------------------------------------

def test_api_checked_at_sensor_returns_timestamp():
  ts = datetime(2026, 6, 22, 10, 0, tzinfo=timezone.utc)
  status = ApiStatusData(checked_at=ts)
  coord = _coordinator(api_status=status)
  s = object.__new__(ApiCheckedAtSensor)
  s.coordinator = coord
  assert s.native_value == ts


def test_api_checked_at_sensor_none_when_no_status():
  coord = _coordinator(api_status=None)
  s = object.__new__(ApiCheckedAtSensor)
  s.coordinator = coord
  assert s.native_value is None


def test_api_checked_at_sensor_none_when_checked_at_not_set():
  status = ApiStatusData(status="starting")
  coord = _coordinator(api_status=status)
  s = object.__new__(ApiCheckedAtSensor)
  s.coordinator = coord
  assert s.native_value is None