test_coordinator.py 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680
  1. from datetime import date, datetime, time
  2. from unittest.mock import AsyncMock, MagicMock, patch
  3. import pytest
  4. from homeassistant.core import HomeAssistant
  5. from homeassistant.exceptions import HomeAssistantError
  6. from pytest_homeassistant_custom_component.common import MockConfigEntry
  7. from custom_components.room_cal_ha.backends import BackendAuthError, BackendConnectError
  8. from custom_components.room_cal_ha.const import (
  9. CONF_API_BASE_URL,
  10. CONF_CALDAV_URL,
  11. CONF_CALENDAR_NAME,
  12. CONF_RESERVATION_DEFAULTS,
  13. CONF_SYNC_MODE,
  14. DOMAIN,
  15. STORE_CURRENT,
  16. STORE_NEXT,
  17. SYNC_MODE_API,
  18. SYNC_MODE_CALDAV,
  19. SYNC_MODE_MANUAL,
  20. )
  21. from custom_components.room_cal_ha.coordinator import RoomCalCoordinator
  22. from custom_components.room_cal_ha.models import OverrideData, ReservationData
  23. SAMPLE = ReservationData(
  24. guest_name="Alice",
  25. check_in_date=date(2026, 7, 1),
  26. check_in_time=time(14, 0),
  27. check_out_date=date(2026, 7, 5),
  28. check_out_time=time(12, 0),
  29. wifi_key="stone-angel-market",
  30. door_code="0472",
  31. )
  32. _BASE_DATA = {
  33. "room_state_name": "Room State",
  34. "current_name": "Current",
  35. "next_name": "Next",
  36. }
  37. def _make_entry(hass, options: dict) -> MockConfigEntry:
  38. entry = MockConfigEntry(domain=DOMAIN, data=_BASE_DATA, options=options)
  39. entry.add_to_hass(hass)
  40. return entry
  41. @pytest.fixture
  42. async def coordinator(hass: HomeAssistant, mock_entry):
  43. mock_entry.add_to_hass(hass)
  44. coord = RoomCalCoordinator(hass, mock_entry)
  45. await coord.async_setup()
  46. return coord
  47. # ---------------------------------------------------------------------------
  48. # Manual mode
  49. # ---------------------------------------------------------------------------
  50. async def test_manual_mode_serves_empty(coordinator: RoomCalCoordinator):
  51. await coordinator.async_refresh()
  52. assert coordinator.data.current == ReservationData()
  53. assert coordinator.data.next == ReservationData()
  54. assert coordinator.data.current_last_refresh is None
  55. assert coordinator.data.next_last_refresh is None
  56. # ---------------------------------------------------------------------------
  57. # CalDAV mode
  58. # ---------------------------------------------------------------------------
  59. async def test_caldav_fetch_writes_store_and_timestamps(hass: HomeAssistant):
  60. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_CALDAV, CONF_CALDAV_URL: "https://cal.example.com", CONF_CALENDAR_NAME: "Room"})
  61. coord = RoomCalCoordinator(hass, entry)
  62. await coord.async_setup()
  63. with patch("custom_components.room_cal_ha.coordinator.CalDavBackend") as MockBackend:
  64. MockBackend.return_value.async_fetch = AsyncMock(return_value=(SAMPLE, None))
  65. await coord.async_refresh()
  66. assert coord.data.current.guest_name == "Alice"
  67. assert coord.data.current.door_code == "0472"
  68. assert coord.data.next == ReservationData()
  69. assert coord.data.current_last_refresh is not None
  70. assert coord.data.next_last_refresh is not None
  71. async def test_caldav_empty_result_clears_store(hass: HomeAssistant):
  72. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_CALDAV, CONF_CALDAV_URL: "https://cal.example.com", CONF_CALENDAR_NAME: "Room"})
  73. coord = RoomCalCoordinator(hass, entry)
  74. await coord.async_setup()
  75. await coord.store.async_set(STORE_CURRENT, SAMPLE)
  76. with patch("custom_components.room_cal_ha.coordinator.CalDavBackend") as MockBackend:
  77. MockBackend.return_value.async_fetch = AsyncMock(return_value=(None, None))
  78. await coord.async_refresh()
  79. assert coord.data.current == ReservationData()
  80. assert coord.store.get(STORE_CURRENT) is None
  81. async def test_caldav_connect_error_preserves_store(hass: HomeAssistant):
  82. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_CALDAV, CONF_CALDAV_URL: "https://cal.example.com", CONF_CALENDAR_NAME: "Room"})
  83. coord = RoomCalCoordinator(hass, entry)
  84. await coord.async_setup()
  85. await coord.store.async_set(STORE_CURRENT, SAMPLE)
  86. with patch("custom_components.room_cal_ha.coordinator.CalDavBackend") as MockBackend:
  87. MockBackend.return_value.async_fetch = AsyncMock(side_effect=BackendConnectError("timeout"))
  88. await coord.async_refresh()
  89. assert coord.data.current.guest_name == "Alice"
  90. assert coord.data.current_last_refresh is None
  91. async def test_caldav_auth_error_triggers_reauth(hass: HomeAssistant):
  92. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_CALDAV, CONF_CALDAV_URL: "https://cal.example.com", CONF_CALENDAR_NAME: "Room"})
  93. coord = RoomCalCoordinator(hass, entry)
  94. await coord.async_setup()
  95. with (
  96. patch("custom_components.room_cal_ha.coordinator.CalDavBackend") as MockBackend,
  97. patch.object(entry, "async_start_reauth") as mock_reauth,
  98. ):
  99. MockBackend.return_value.async_fetch = AsyncMock(side_effect=BackendAuthError)
  100. await coord.async_refresh()
  101. mock_reauth.assert_called_once_with(hass)
  102. # ---------------------------------------------------------------------------
  103. # API mode
  104. # ---------------------------------------------------------------------------
  105. async def test_api_fetch_writes_store(hass: HomeAssistant):
  106. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_API, CONF_API_BASE_URL: "https://api.example.com"})
  107. coord = RoomCalCoordinator(hass, entry)
  108. await coord.async_setup()
  109. with patch("custom_components.room_cal_ha.coordinator.RoomCalApiBackend") as MockBackend:
  110. instance = MockBackend.return_value
  111. instance.async_fetch = AsyncMock(return_value=(SAMPLE, None))
  112. await coord.async_refresh()
  113. assert coord.data.current.guest_name == "Alice"
  114. assert coord.data.current_last_refresh is not None
  115. async def test_api_connect_error_preserves_store(hass: HomeAssistant):
  116. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_API, CONF_API_BASE_URL: "https://api.example.com"})
  117. coord = RoomCalCoordinator(hass, entry)
  118. await coord.async_setup()
  119. await coord.store.async_set(STORE_CURRENT, SAMPLE)
  120. with patch("custom_components.room_cal_ha.coordinator.RoomCalApiBackend") as MockBackend:
  121. instance = MockBackend.return_value
  122. instance.async_fetch = AsyncMock(side_effect=BackendConnectError("timeout"))
  123. await coord.async_refresh()
  124. assert coord.data.current.guest_name == "Alice"
  125. assert coord.data.current_last_refresh is None
  126. async def test_api_auth_error_triggers_reauth(hass: HomeAssistant):
  127. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_API, CONF_API_BASE_URL: "https://api.example.com"})
  128. coord = RoomCalCoordinator(hass, entry)
  129. await coord.async_setup()
  130. with (
  131. patch("custom_components.room_cal_ha.coordinator.RoomCalApiBackend") as MockBackend,
  132. patch.object(entry, "async_start_reauth") as mock_reauth,
  133. ):
  134. instance = MockBackend.return_value
  135. instance.async_fetch = AsyncMock(side_effect=BackendAuthError)
  136. await coord.async_refresh()
  137. mock_reauth.assert_called_once_with(hass)
  138. # ---------------------------------------------------------------------------
  139. # Thin updates
  140. # ---------------------------------------------------------------------------
  141. async def test_thin_update_overlays_fields(coordinator: RoomCalCoordinator):
  142. await coordinator.store.async_set(STORE_CURRENT, SAMPLE)
  143. await coordinator.async_refresh()
  144. await coordinator.async_apply_update(STORE_CURRENT, {"wifi_ssid": "GuestNet", "door_code": "9999"})
  145. assert coordinator.data.current.wifi_ssid == "GuestNet"
  146. assert coordinator.data.current.door_code == "9999"
  147. assert coordinator.data.current.guest_name == "Alice"
  148. assert coordinator.data.current.wifi_key == "stone-angel-market"
  149. async def test_thin_update_updates_last_refresh(coordinator: RoomCalCoordinator):
  150. await coordinator.store.async_set(STORE_CURRENT, SAMPLE)
  151. await coordinator.async_refresh()
  152. assert coordinator.data.current_last_refresh is None
  153. await coordinator.async_apply_update(STORE_CURRENT, {"door_code": "1234"})
  154. assert coordinator.data.current_last_refresh is not None
  155. # ---------------------------------------------------------------------------
  156. # Store persistence
  157. # ---------------------------------------------------------------------------
  158. async def test_store_persists_across_reload(hass: HomeAssistant, mock_entry):
  159. mock_entry.add_to_hass(hass)
  160. coord = RoomCalCoordinator(hass, mock_entry)
  161. await coord.async_setup()
  162. await coord.store.async_set(STORE_CURRENT, SAMPLE)
  163. coord2 = RoomCalCoordinator(hass, mock_entry)
  164. await coord2.async_setup()
  165. loaded = coord2.store.get(STORE_CURRENT)
  166. assert loaded is not None
  167. assert loaded.guest_name == "Alice"
  168. assert loaded.door_code == "0472"
  169. # ---------------------------------------------------------------------------
  170. # _reservation_window_state
  171. # ---------------------------------------------------------------------------
  172. DATED = ReservationData(
  173. check_in_date=date(2026, 7, 1),
  174. check_in_time=time(14, 0),
  175. check_out_date=date(2026, 7, 5),
  176. check_out_time=time(12, 0),
  177. )
  178. def _state_at(coordinator, res, dt_str):
  179. """Freeze dt_util.now() to a naive local datetime and call _reservation_window_state."""
  180. frozen = MagicMock()
  181. frozen.replace.return_value = datetime.fromisoformat(dt_str) # naive, no TZ
  182. with patch("custom_components.room_cal_ha.coordinator.dt_util.now", return_value=frozen):
  183. return coordinator._reservation_window_state(res)
  184. async def test_window_state_active(coordinator):
  185. assert _state_at(coordinator, DATED, "2026-07-03T10:00:00") == "active"
  186. async def test_window_state_pending(coordinator):
  187. assert _state_at(coordinator, DATED, "2026-06-30T10:00:00") == "pending"
  188. async def test_window_state_elapsed(coordinator):
  189. assert _state_at(coordinator, DATED, "2026-07-10T12:00:00") == "elapsed"
  190. async def test_window_state_at_check_in_is_active(coordinator):
  191. assert _state_at(coordinator, DATED, "2026-07-01T14:00:00") == "active"
  192. async def test_window_state_at_check_out_is_elapsed(coordinator):
  193. assert _state_at(coordinator, DATED, "2026-07-05T12:00:00") == "elapsed"
  194. async def test_window_state_no_dates_returns_none(coordinator):
  195. assert coordinator._reservation_window_state(ReservationData()) is None
  196. # ---------------------------------------------------------------------------
  197. # Check-in WiFi key behavior
  198. # ---------------------------------------------------------------------------
  199. _WIFI_NEXT = ReservationData(
  200. guest_name="Bob",
  201. check_in_date=date(2026, 7, 1),
  202. check_in_time=time(14, 0),
  203. check_out_date=date(2026, 7, 5),
  204. check_out_time=time(12, 0),
  205. wifi_ssid="GuestNet",
  206. wifi_key="securepass",
  207. )
  208. async def test_check_in_calls_wifi_key(hass: HomeAssistant, coordinator: RoomCalCoordinator):
  209. await coordinator.store.async_set(STORE_NEXT, _WIFI_NEXT)
  210. mock_switch = MagicMock()
  211. mock_switch.async_turn_on = AsyncMock()
  212. coordinator._guest_mode_switch = mock_switch
  213. with (
  214. patch.object(coordinator, "_is_mode_sync_on", return_value=True),
  215. patch("custom_components.room_cal_ha.wifi.async_set_wifi_key", new_callable=AsyncMock) as mock_wifi,
  216. ):
  217. await coordinator._async_do_check_in()
  218. coordinator.cancel_mode_sync()
  219. mock_wifi.assert_awaited_once_with(hass, "GuestNet", "securepass")
  220. mock_switch.async_turn_on.assert_awaited_once()
  221. async def test_check_in_skips_wifi_when_ssid_none(coordinator: RoomCalCoordinator):
  222. await coordinator.store.async_set(STORE_NEXT, ReservationData(wifi_ssid=None, wifi_key="securepass"))
  223. coordinator._guest_mode_switch = None
  224. with patch("custom_components.room_cal_ha.wifi.async_set_wifi_key", new_callable=AsyncMock) as mock_wifi:
  225. await coordinator._async_do_check_in()
  226. mock_wifi.assert_not_awaited()
  227. async def test_check_in_skips_wifi_when_key_none(coordinator: RoomCalCoordinator):
  228. await coordinator.store.async_set(STORE_NEXT, ReservationData(wifi_ssid="GuestNet", wifi_key=None))
  229. coordinator._guest_mode_switch = None
  230. with patch("custom_components.room_cal_ha.wifi.async_set_wifi_key", new_callable=AsyncMock) as mock_wifi:
  231. await coordinator._async_do_check_in()
  232. mock_wifi.assert_not_awaited()
  233. async def test_check_in_wifi_error_still_activates_guest_mode(coordinator: RoomCalCoordinator):
  234. await coordinator.store.async_set(STORE_NEXT, _WIFI_NEXT)
  235. mock_switch = MagicMock()
  236. mock_switch.async_turn_on = AsyncMock()
  237. coordinator._guest_mode_switch = mock_switch
  238. with (
  239. patch.object(coordinator, "_is_mode_sync_on", return_value=True),
  240. patch("custom_components.room_cal_ha.wifi.async_set_wifi_key", side_effect=HomeAssistantError("hub down")),
  241. ):
  242. await coordinator._async_do_check_in()
  243. coordinator.cancel_mode_sync()
  244. mock_switch.async_turn_on.assert_awaited_once()
  245. # ---------------------------------------------------------------------------
  246. # async_start_override / async_end_override
  247. # ---------------------------------------------------------------------------
  248. _OVERRIDE_RES = ReservationData(guest_name="Override Guest", door_code="1111")
  249. _WIFI_OVERRIDE = ReservationData(guest_name="WiFi Guest", wifi_ssid="GuestNet", wifi_key="secret")
  250. async def test_start_override_sets_current_in_data(coordinator: RoomCalCoordinator):
  251. coordinator._guest_mode_switch = None
  252. await coordinator.async_start_override(OverrideData(current=_OVERRIDE_RES))
  253. assert coordinator.data.current.guest_name == "Override Guest"
  254. async def test_start_override_sets_override_active_flag(coordinator: RoomCalCoordinator):
  255. coordinator._guest_mode_switch = None
  256. await coordinator.async_start_override(OverrideData(current=_OVERRIDE_RES))
  257. assert coordinator.data.override_active is True
  258. async def test_start_override_cancels_mode_sync(coordinator: RoomCalCoordinator):
  259. cancel1, cancel2 = MagicMock(), MagicMock()
  260. coordinator._mode_sync_handles = [cancel1, cancel2]
  261. coordinator._guest_mode_switch = None
  262. await coordinator.async_start_override(OverrideData(current=_OVERRIDE_RES))
  263. cancel1.assert_called_once()
  264. cancel2.assert_called_once()
  265. assert coordinator._mode_sync_handles == []
  266. async def test_start_override_turns_on_guest_mode(coordinator: RoomCalCoordinator):
  267. mock_switch = MagicMock()
  268. mock_switch.async_turn_on = AsyncMock()
  269. coordinator._guest_mode_switch = mock_switch
  270. with patch.object(coordinator, "_is_mode_sync_on", return_value=True):
  271. await coordinator.async_start_override(OverrideData(current=_OVERRIDE_RES))
  272. mock_switch.async_turn_on.assert_awaited_once()
  273. async def test_start_override_does_not_turn_on_guest_mode_when_mode_sync_off(coordinator: RoomCalCoordinator):
  274. mock_switch = MagicMock()
  275. mock_switch.async_turn_on = AsyncMock()
  276. coordinator._guest_mode_switch = mock_switch
  277. with patch.object(coordinator, "_is_mode_sync_on", return_value=False):
  278. await coordinator.async_start_override(OverrideData(current=_OVERRIDE_RES))
  279. mock_switch.async_turn_on.assert_not_awaited()
  280. async def test_start_override_calls_wifi_when_credentials_present(hass: HomeAssistant, coordinator: RoomCalCoordinator):
  281. coordinator._guest_mode_switch = MagicMock()
  282. coordinator._guest_mode_switch.async_turn_on = AsyncMock()
  283. with (
  284. patch.object(coordinator, "_is_mode_sync_on", return_value=True),
  285. patch("custom_components.room_cal_ha.wifi.async_set_wifi_key", new_callable=AsyncMock) as mock_wifi,
  286. ):
  287. await coordinator.async_start_override(OverrideData(current=_WIFI_OVERRIDE))
  288. mock_wifi.assert_awaited_once_with(hass, "GuestNet", "secret")
  289. async def test_start_override_skips_wifi_when_mode_sync_off(coordinator: RoomCalCoordinator):
  290. coordinator._guest_mode_switch = None
  291. with (
  292. patch.object(coordinator, "_is_mode_sync_on", return_value=False),
  293. patch("custom_components.room_cal_ha.wifi.async_set_wifi_key", new_callable=AsyncMock) as mock_wifi,
  294. ):
  295. await coordinator.async_start_override(OverrideData(current=_WIFI_OVERRIDE))
  296. mock_wifi.assert_not_awaited()
  297. async def test_start_override_skips_wifi_when_ssid_missing(coordinator: RoomCalCoordinator):
  298. coordinator._guest_mode_switch = None
  299. no_ssid = ReservationData(wifi_ssid=None, wifi_key="secret")
  300. with patch("custom_components.room_cal_ha.wifi.async_set_wifi_key", new_callable=AsyncMock) as mock_wifi:
  301. await coordinator.async_start_override(OverrideData(current=no_ssid))
  302. mock_wifi.assert_not_awaited()
  303. async def test_start_override_uses_store_next_when_override_next_is_none(coordinator: RoomCalCoordinator):
  304. await coordinator.store.async_set(STORE_NEXT, SAMPLE)
  305. coordinator._guest_mode_switch = None
  306. await coordinator.async_start_override(OverrideData(current=_OVERRIDE_RES))
  307. assert coordinator.data.next.guest_name == "Alice"
  308. async def test_start_override_uses_override_next_when_provided(coordinator: RoomCalCoordinator):
  309. override_next = ReservationData(guest_name="Override Next")
  310. coordinator._guest_mode_switch = None
  311. await coordinator.async_start_override(OverrideData(current=_OVERRIDE_RES, next=override_next))
  312. assert coordinator.data.next.guest_name == "Override Next"
  313. async def test_start_override_applies_defaults_to_current(hass: HomeAssistant):
  314. entry = MockConfigEntry(
  315. domain=DOMAIN,
  316. data={},
  317. options={
  318. CONF_SYNC_MODE: SYNC_MODE_MANUAL,
  319. CONF_RESERVATION_DEFAULTS: {"wifi_ssid": "DefaultSSID", "check_out_time": "11:00:00"},
  320. },
  321. )
  322. entry.add_to_hass(hass)
  323. coord = RoomCalCoordinator(hass, entry)
  324. await coord.async_setup()
  325. coord._guest_mode_switch = None
  326. partial_override = ReservationData(guest_name="Override Guest")
  327. await coord.async_start_override(OverrideData(current=partial_override))
  328. assert coord.data.current.guest_name == "Override Guest"
  329. assert coord.data.current.wifi_ssid == "DefaultSSID"
  330. assert coord.data.current.check_out_time == time(11, 0)
  331. async def test_start_override_defaults_do_not_overwrite_explicit_fields(hass: HomeAssistant):
  332. entry = MockConfigEntry(
  333. domain=DOMAIN,
  334. data={},
  335. options={
  336. CONF_SYNC_MODE: SYNC_MODE_MANUAL,
  337. CONF_RESERVATION_DEFAULTS: {"wifi_ssid": "DefaultSSID"},
  338. },
  339. )
  340. entry.add_to_hass(hass)
  341. coord = RoomCalCoordinator(hass, entry)
  342. await coord.async_setup()
  343. coord._guest_mode_switch = None
  344. explicit_override = ReservationData(guest_name="Override Guest", wifi_ssid="ExplicitSSID")
  345. await coord.async_start_override(OverrideData(current=explicit_override))
  346. assert coord.data.current.wifi_ssid == "ExplicitSSID"
  347. async def test_refresh_during_override_applies_defaults_to_current(hass: HomeAssistant):
  348. entry = MockConfigEntry(
  349. domain=DOMAIN,
  350. data={},
  351. options={
  352. CONF_SYNC_MODE: SYNC_MODE_MANUAL,
  353. CONF_RESERVATION_DEFAULTS: {"door_code": "9999"},
  354. },
  355. )
  356. entry.add_to_hass(hass)
  357. coord = RoomCalCoordinator(hass, entry)
  358. await coord.async_setup()
  359. coord._guest_mode_switch = None
  360. partial_override = ReservationData(guest_name="Override Guest")
  361. await coord.async_start_override(OverrideData(current=partial_override))
  362. await coord.async_refresh()
  363. assert coord.data.current.guest_name == "Override Guest"
  364. assert coord.data.current.door_code == "9999"
  365. async def test_start_override_applies_defaults_to_store_next(hass: HomeAssistant):
  366. entry = MockConfigEntry(
  367. domain=DOMAIN,
  368. data={},
  369. options={
  370. CONF_SYNC_MODE: SYNC_MODE_MANUAL,
  371. CONF_RESERVATION_DEFAULTS: {"wifi_ssid": "DefaultSSID"},
  372. },
  373. )
  374. entry.add_to_hass(hass)
  375. coord = RoomCalCoordinator(hass, entry)
  376. await coord.async_setup()
  377. coord._guest_mode_switch = None
  378. await coord.store.async_set(STORE_NEXT, ReservationData(guest_name="Next Guest"))
  379. await coord.async_start_override(OverrideData(current=_OVERRIDE_RES))
  380. assert coord.data.next.guest_name == "Next Guest"
  381. assert coord.data.next.wifi_ssid == "DefaultSSID"
  382. async def test_end_override_clears_override_active(coordinator: RoomCalCoordinator):
  383. coordinator._guest_mode_switch = None
  384. coordinator._override = OverrideData(current=_OVERRIDE_RES)
  385. await coordinator.async_end_override()
  386. assert coordinator.data.override_active is False
  387. async def test_end_override_restores_store_current(coordinator: RoomCalCoordinator):
  388. await coordinator.store.async_set(STORE_CURRENT, SAMPLE)
  389. coordinator._guest_mode_switch = None
  390. coordinator._override = OverrideData(current=_OVERRIDE_RES)
  391. await coordinator.async_end_override()
  392. assert coordinator.data.current.guest_name == "Alice"
  393. # ---------------------------------------------------------------------------
  394. # Thin updates during override
  395. # ---------------------------------------------------------------------------
  396. async def test_thin_update_current_during_override_updates_override_not_store(coordinator: RoomCalCoordinator):
  397. await coordinator.store.async_set(STORE_CURRENT, SAMPLE)
  398. await coordinator.async_refresh()
  399. coordinator._override = OverrideData(current=_OVERRIDE_RES)
  400. await coordinator.async_apply_update(STORE_CURRENT, {"door_code": "9999"})
  401. assert coordinator.data.current.door_code == "9999"
  402. # Store is unchanged
  403. assert coordinator.store.get(STORE_CURRENT).door_code == "0472"
  404. async def test_thin_update_current_during_override_data_still_shows_override_active(coordinator: RoomCalCoordinator):
  405. await coordinator.async_refresh()
  406. coordinator._override = OverrideData(current=_OVERRIDE_RES)
  407. await coordinator.async_apply_update(STORE_CURRENT, {"door_code": "9999"})
  408. assert coordinator.data.override_active is True
  409. async def test_thin_update_next_during_override_falls_through_to_store(coordinator: RoomCalCoordinator):
  410. coordinator._override = OverrideData(current=_OVERRIDE_RES) # override.next is None
  411. await coordinator.async_refresh()
  412. await coordinator.async_apply_update(STORE_NEXT, {"guest_name": "Next Guest"})
  413. assert coordinator.store.get(STORE_NEXT).guest_name == "Next Guest"
  414. assert coordinator.data.override_active is True
  415. assert coordinator.data.current.guest_name == "Override Guest"
  416. # ---------------------------------------------------------------------------
  417. # Backend fetch during override preserves override display
  418. # ---------------------------------------------------------------------------
  419. async def test_backend_fetch_during_override_updates_store_but_preserves_override(hass: HomeAssistant):
  420. entry = _make_entry(hass, {CONF_SYNC_MODE: SYNC_MODE_CALDAV, CONF_CALDAV_URL: "https://cal.example.com", CONF_CALENDAR_NAME: "Room"})
  421. coord = RoomCalCoordinator(hass, entry)
  422. await coord.async_setup()
  423. coord._override = OverrideData(current=_OVERRIDE_RES)
  424. backend_data = ReservationData(guest_name="Backend Alice")
  425. with patch("custom_components.room_cal_ha.coordinator.CalDavBackend") as MockBackend:
  426. MockBackend.return_value.async_fetch = AsyncMock(return_value=(backend_data, None))
  427. await coord.async_refresh()
  428. # Store was written by the backend
  429. assert coord.store.get(STORE_CURRENT).guest_name == "Backend Alice"
  430. # But displayed data still shows the override
  431. assert coord.data.current.guest_name == "Override Guest"
  432. assert coord.data.override_active is True
  433. # ---------------------------------------------------------------------------
  434. # async_reconcile_from_store (shared helper)
  435. # ---------------------------------------------------------------------------
  436. _ACTIVE = ReservationData(
  437. guest_name="Active Guest",
  438. check_in_date=date(2026, 6, 20),
  439. check_in_time=time(14, 0),
  440. check_out_date=date(2026, 6, 25),
  441. check_out_time=time(12, 0),
  442. )
  443. _ELAPSED = ReservationData(
  444. guest_name="Past Guest",
  445. check_in_date=date(2026, 6, 1),
  446. check_in_time=time(14, 0),
  447. check_out_date=date(2026, 6, 5),
  448. check_out_time=time(12, 0),
  449. )
  450. _FUTURE = ReservationData(
  451. guest_name="Future Guest",
  452. check_in_date=date(2026, 7, 1),
  453. check_in_time=time(14, 0),
  454. check_out_date=date(2026, 7, 5),
  455. check_out_time=time(12, 0),
  456. )
  457. # Freeze time at 2026-06-22 12:00 local (naive)
  458. _RECONCILE_NOW = MagicMock()
  459. _RECONCILE_NOW.replace.return_value = datetime(2026, 6, 22, 12, 0, 0)
  460. async def test_reconcile_active_current_turns_on_guest_mode(coordinator: RoomCalCoordinator):
  461. await coordinator.store.async_set(STORE_CURRENT, _ACTIVE)
  462. mock_switch = MagicMock()
  463. mock_switch.async_turn_on = AsyncMock()
  464. mock_switch.async_turn_off = AsyncMock()
  465. coordinator._guest_mode_switch = mock_switch
  466. with (
  467. patch.object(coordinator, "_is_mode_sync_on", return_value=True),
  468. patch("custom_components.room_cal_ha.coordinator.dt_util.now", return_value=_RECONCILE_NOW),
  469. ):
  470. await coordinator.async_reconcile_from_store()
  471. coordinator.cancel_mode_sync()
  472. mock_switch.async_turn_on.assert_awaited_once()
  473. mock_switch.async_turn_off.assert_not_awaited()
  474. async def test_reconcile_elapsed_current_clears_store_and_turns_off_guest_mode(coordinator: RoomCalCoordinator):
  475. await coordinator.store.async_set(STORE_CURRENT, _ELAPSED)
  476. mock_switch = MagicMock()
  477. mock_switch.async_turn_on = AsyncMock()
  478. mock_switch.async_turn_off = AsyncMock()
  479. coordinator._guest_mode_switch = mock_switch
  480. with (
  481. patch.object(coordinator, "_is_mode_sync_on", return_value=True),
  482. patch("custom_components.room_cal_ha.coordinator.dt_util.now", return_value=_RECONCILE_NOW),
  483. ):
  484. await coordinator.async_reconcile_from_store()
  485. assert coordinator.store.get(STORE_CURRENT) is None
  486. mock_switch.async_turn_off.assert_awaited_once()
  487. async def test_reconcile_promotes_next_when_check_in_elapsed(coordinator: RoomCalCoordinator):
  488. await coordinator.store.async_set(STORE_NEXT, _ACTIVE)
  489. with (
  490. patch.object(coordinator, "_is_mode_sync_on", return_value=True),
  491. patch("custom_components.room_cal_ha.coordinator.dt_util.now", return_value=_RECONCILE_NOW),
  492. ):
  493. await coordinator.async_reconcile_from_store()
  494. coordinator.cancel_mode_sync()
  495. assert coordinator.store.get(STORE_CURRENT).guest_name == "Active Guest"
  496. async def test_reconcile_does_not_touch_guest_mode_when_mode_sync_off(coordinator: RoomCalCoordinator):
  497. await coordinator.store.async_set(STORE_CURRENT, _ACTIVE)
  498. mock_switch = MagicMock()
  499. mock_switch.async_turn_on = AsyncMock()
  500. mock_switch.async_turn_off = AsyncMock()
  501. coordinator._guest_mode_switch = mock_switch
  502. with (
  503. patch.object(coordinator, "_is_mode_sync_on", return_value=False),
  504. patch("custom_components.room_cal_ha.coordinator.dt_util.now", return_value=_RECONCILE_NOW),
  505. ):
  506. await coordinator.async_reconcile_from_store()
  507. mock_switch.async_turn_on.assert_not_awaited()
  508. mock_switch.async_turn_off.assert_not_awaited()