369 lines
13 KiB
Python
369 lines
13 KiB
Python
"""Acceptance tests for the optional Home Assistant integration.
|
|
|
|
Groups:
|
|
1. HA disabled — regression guard (uses default client/seeded_client fixtures)
|
|
2. HomeAssistantClient unit tests — mock requests.get, no Flask required
|
|
3. Poller integration — uses ha_app fixture, calls _poll_once() directly
|
|
4. Dashboard/device/battery UI — checks rendered HTML with HA enabled
|
|
"""
|
|
|
|
import os
|
|
import tempfile
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# HA-enabled app fixture
|
|
# ---------------------------------------------------------------------------
|
|
|
|
@pytest.fixture()
|
|
def ha_app():
|
|
"""Flask app with Home Assistant enabled (HOMEASSISTANT_URL set).
|
|
|
|
The poller is started but its first fire is after 300 s — it never fires
|
|
during a test run.
|
|
"""
|
|
db_fd, db_path = tempfile.mkstemp(suffix=".db")
|
|
os.close(db_fd)
|
|
|
|
class HaTestConfig:
|
|
SQLALCHEMY_DATABASE_URI = f"sqlite:///{db_path}"
|
|
SECRET_KEY = "test-secret"
|
|
SQLALCHEMY_TRACK_MODIFICATIONS = False
|
|
TESTING = True
|
|
HOMEASSISTANT_URL = "http://ha.test:8123"
|
|
HOMEASSISTANT_API_KEY = "fake-token"
|
|
HOMEASSISTANT_POLL_INTERVAL = 300
|
|
|
|
from app import create_app
|
|
flask_app = create_app(HaTestConfig)
|
|
yield flask_app
|
|
os.unlink(db_path)
|
|
|
|
|
|
@pytest.fixture()
|
|
def ha_client_f(ha_app):
|
|
"""Test client bound to the HA-enabled app."""
|
|
with ha_app.test_client() as c:
|
|
yield c
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Group 1 — HA disabled (regression guard)
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_ha_disabled_dashboard_no_ha_column(client):
|
|
resp = client.get("/")
|
|
assert resp.status_code == 200
|
|
assert b"ha-pct" not in resp.data
|
|
|
|
|
|
def test_ha_disabled_device_no_ha_field(client):
|
|
client.post("/device/add", data={"name": "Dev A", "battery_slots": "1"})
|
|
resp = client.get("/device/1")
|
|
assert resp.status_code == 200
|
|
assert b"ha_entity_id" not in resp.data
|
|
|
|
|
|
def test_ha_disabled_battery_detail_still_has_percentage_field(client):
|
|
"""battery_percentage edit field is always shown regardless of HA config."""
|
|
client.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
resp = client.get("/battery/1")
|
|
assert b"battery_percentage" in resp.data
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Group 2 — HomeAssistantClient unit tests
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_ha_client_disabled_when_no_url():
|
|
from ha_client import HomeAssistantClient
|
|
c = HomeAssistantClient(None, None)
|
|
assert c.enabled is False
|
|
assert c.get_state("sensor.foo") is None
|
|
|
|
|
|
def test_ha_client_disabled_when_no_key():
|
|
from ha_client import HomeAssistantClient
|
|
c = HomeAssistantClient("http://ha.local", None)
|
|
assert c.enabled is False
|
|
|
|
|
|
def test_ha_client_returns_integer_state():
|
|
from ha_client import HomeAssistantClient
|
|
mock_resp = MagicMock()
|
|
mock_resp.json.return_value = {"state": "85", "entity_id": "sensor.foo"}
|
|
mock_resp.raise_for_status.return_value = None
|
|
|
|
with patch("ha_client.requests.get", return_value=mock_resp) as mock_get:
|
|
c = HomeAssistantClient("http://ha.local:8123", "tok")
|
|
result = c.get_state("sensor.foo")
|
|
|
|
assert result == 85
|
|
mock_get.assert_called_once_with(
|
|
"http://ha.local:8123/api/states/sensor.foo",
|
|
headers={"Authorization": "Bearer tok"},
|
|
timeout=10,
|
|
)
|
|
|
|
|
|
def test_ha_client_handles_float_state():
|
|
from ha_client import HomeAssistantClient
|
|
mock_resp = MagicMock()
|
|
mock_resp.json.return_value = {"state": "72.6"}
|
|
mock_resp.raise_for_status.return_value = None
|
|
|
|
with patch("ha_client.requests.get", return_value=mock_resp):
|
|
c = HomeAssistantClient("http://ha.local", "tok")
|
|
assert c.get_state("sensor.x") == 72
|
|
|
|
|
|
def test_ha_client_returns_none_on_network_error():
|
|
from ha_client import HomeAssistantClient
|
|
import requests as req
|
|
with patch("ha_client.requests.get", side_effect=req.RequestException("timeout")):
|
|
c = HomeAssistantClient("http://ha.local", "tok")
|
|
assert c.get_state("sensor.x") is None
|
|
|
|
|
|
def test_ha_client_returns_none_on_non_numeric_state():
|
|
from ha_client import HomeAssistantClient
|
|
mock_resp = MagicMock()
|
|
mock_resp.json.return_value = {"state": "unavailable"}
|
|
mock_resp.raise_for_status.return_value = None
|
|
|
|
with patch("ha_client.requests.get", return_value=mock_resp):
|
|
c = HomeAssistantClient("http://ha.local", "tok")
|
|
assert c.get_state("sensor.x") is None
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Group 3 — Poller integration
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def _make_session_factory(ha_app):
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
engine = create_engine(ha_app.config["SQLALCHEMY_DATABASE_URI"])
|
|
return sessionmaker(bind=engine)
|
|
|
|
|
|
def test_poll_updates_installed_batteries(ha_app, ha_client_f):
|
|
"""Installed battery in a device with ha_entity_id gets battery_percentage updated."""
|
|
ha_client_f.post("/device/add", data={"name": "Dev A", "battery_slots": "2"})
|
|
ha_client_f.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
# Assign battery to device
|
|
ha_client_f.post("/battery/1/assign", data={"device_id": "1"})
|
|
# Set ha_entity_id on device
|
|
ha_client_f.post("/device/1/edit", data={
|
|
"name": "Dev A", "battery_slots": "2", "ha_entity_id": "sensor.dev_a_battery"
|
|
})
|
|
|
|
from ha_client import HomeAssistantClient
|
|
from ha_poller import HaPoller
|
|
from models import Battery
|
|
|
|
mock_ha = MagicMock(spec=HomeAssistantClient)
|
|
mock_ha.enabled = True
|
|
mock_ha.get_state.return_value = 42
|
|
|
|
Session = _make_session_factory(ha_app)
|
|
poller = HaPoller(mock_ha, Session, interval=300)
|
|
poller._poll_once()
|
|
|
|
s = Session()
|
|
b = s.get(Battery, 1)
|
|
assert b.battery_percentage == 42
|
|
s.close()
|
|
|
|
mock_ha.get_state.assert_called_once_with("sensor.dev_a_battery")
|
|
|
|
|
|
def test_poll_skips_uninstalled_batteries(ha_app, ha_client_f):
|
|
"""Batteries that are available (not installed) are not updated by the poller."""
|
|
ha_client_f.post("/device/add", data={"name": "Dev B", "battery_slots": "1"})
|
|
ha_client_f.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
# Set entity on device but do NOT install the battery
|
|
ha_client_f.post("/device/1/edit", data={
|
|
"name": "Dev B", "battery_slots": "1", "ha_entity_id": "sensor.dev_b_battery"
|
|
})
|
|
|
|
from ha_client import HomeAssistantClient
|
|
from ha_poller import HaPoller
|
|
from models import Battery
|
|
|
|
mock_ha = MagicMock(spec=HomeAssistantClient)
|
|
mock_ha.enabled = True
|
|
mock_ha.get_state.return_value = 55
|
|
|
|
Session = _make_session_factory(ha_app)
|
|
poller = HaPoller(mock_ha, Session, interval=300)
|
|
poller._poll_once()
|
|
|
|
s = Session()
|
|
b = s.get(Battery, 1)
|
|
assert b.battery_percentage is None # uninstalled — not touched
|
|
s.close()
|
|
|
|
|
|
def test_poll_skips_devices_without_entity_id(ha_app, ha_client_f):
|
|
"""Devices with no ha_entity_id must never reach get_state."""
|
|
ha_client_f.post("/device/add", data={"name": "Dev C", "battery_slots": "1"})
|
|
|
|
from ha_client import HomeAssistantClient
|
|
from ha_poller import HaPoller
|
|
|
|
mock_ha = MagicMock(spec=HomeAssistantClient)
|
|
mock_ha.enabled = True
|
|
|
|
Session = _make_session_factory(ha_app)
|
|
poller = HaPoller(mock_ha, Session, interval=300)
|
|
poller._poll_once()
|
|
|
|
mock_ha.get_state.assert_not_called()
|
|
|
|
|
|
def test_poll_handles_api_error_gracefully(ha_app, ha_client_f):
|
|
"""When get_state returns None, no exception is raised and percentage stays None."""
|
|
ha_client_f.post("/device/add", data={"name": "Dev D", "battery_slots": "1"})
|
|
ha_client_f.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
ha_client_f.post("/battery/1/assign", data={"device_id": "1"})
|
|
ha_client_f.post("/device/1/edit", data={
|
|
"name": "Dev D", "battery_slots": "1", "ha_entity_id": "sensor.dev_d"
|
|
})
|
|
|
|
from ha_client import HomeAssistantClient
|
|
from ha_poller import HaPoller
|
|
from models import Battery
|
|
|
|
mock_ha = MagicMock(spec=HomeAssistantClient)
|
|
mock_ha.enabled = True
|
|
mock_ha.get_state.return_value = None # simulate API failure
|
|
|
|
Session = _make_session_factory(ha_app)
|
|
poller = HaPoller(mock_ha, Session, interval=300)
|
|
poller._poll_once() # must not raise
|
|
|
|
s = Session()
|
|
b = s.get(Battery, 1)
|
|
assert b.battery_percentage is None
|
|
s.close()
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Group 4 — Dashboard / device / battery UI with HA enabled
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_dashboard_shows_ha_column_when_enabled(ha_client_f):
|
|
resp = ha_client_f.get("/")
|
|
assert resp.status_code == 200
|
|
assert b"ha-pct" in resp.data
|
|
|
|
|
|
def test_dashboard_shows_low_battery_warning(ha_app, ha_client_f):
|
|
ha_client_f.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
|
|
# Set percentage directly in DB
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from models import Battery
|
|
engine = create_engine(ha_app.config["SQLALCHEMY_DATABASE_URI"])
|
|
s = sessionmaker(bind=engine)()
|
|
b = s.get(Battery, 1)
|
|
b.battery_percentage = 15
|
|
s.commit()
|
|
s.close()
|
|
|
|
resp = ha_client_f.get("/")
|
|
assert b"badge-warning" in resp.data
|
|
assert b"15%" in resp.data
|
|
|
|
|
|
def test_dashboard_no_warning_for_high_percentage(ha_app, ha_client_f):
|
|
ha_client_f.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from models import Battery
|
|
engine = create_engine(ha_app.config["SQLALCHEMY_DATABASE_URI"])
|
|
s = sessionmaker(bind=engine)()
|
|
b = s.get(Battery, 1)
|
|
b.battery_percentage = 85
|
|
s.commit()
|
|
s.close()
|
|
|
|
resp = ha_client_f.get("/")
|
|
assert b"85%" in resp.data
|
|
# badge-warning should NOT appear for this battery's percentage
|
|
# (may still appear in page for other reasons, so check row contains 85% but not warning badge near it)
|
|
assert b"col-ha-pct" in resp.data # column rendered
|
|
|
|
|
|
def test_device_detail_shows_ha_field(ha_client_f):
|
|
ha_client_f.post("/device/add", data={"name": "Dev E", "battery_slots": "1"})
|
|
resp = ha_client_f.get("/device/1")
|
|
assert resp.status_code == 200
|
|
assert b"ha_entity_id" in resp.data
|
|
|
|
|
|
def test_edit_device_ha_entity_id_saves(ha_client_f):
|
|
ha_client_f.post("/device/add", data={"name": "Dev F", "battery_slots": "1"})
|
|
ha_client_f.post("/device/1/edit", data={
|
|
"name": "Dev F", "battery_slots": "1", "ha_entity_id": "sensor.my_remote"
|
|
})
|
|
resp = ha_client_f.get("/device/1")
|
|
assert b"sensor.my_remote" in resp.data
|
|
|
|
|
|
def test_edit_device_ha_entity_id_clear(ha_client_f):
|
|
ha_client_f.post("/device/add", data={"name": "Dev G", "battery_slots": "1"})
|
|
ha_client_f.post("/device/1/edit", data={
|
|
"name": "Dev G", "battery_slots": "1", "ha_entity_id": "sensor.foo"
|
|
})
|
|
# Now clear it
|
|
ha_client_f.post("/device/1/edit", data={
|
|
"name": "Dev G", "battery_slots": "1", "ha_entity_id": ""
|
|
})
|
|
resp = ha_client_f.get("/device/1")
|
|
assert b"sensor.foo" not in resp.data
|
|
|
|
|
|
def test_charge_log_resets_percentage_to_100(ha_app, ha_client_f):
|
|
ha_client_f.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
|
|
# Set low percentage
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from models import Battery
|
|
engine = create_engine(ha_app.config["SQLALCHEMY_DATABASE_URI"])
|
|
s = sessionmaker(bind=engine)()
|
|
b = s.get(Battery, 1)
|
|
b.battery_percentage = 30
|
|
s.commit()
|
|
s.close()
|
|
|
|
# Log a charge
|
|
ha_client_f.post("/battery/1/charge-log/add", data={"charged_date": "2026-04-13"})
|
|
|
|
s2 = sessionmaker(bind=engine)()
|
|
b2 = s2.get(Battery, 1)
|
|
assert b2.battery_percentage == 100
|
|
s2.close()
|
|
|
|
|
|
def test_manual_battery_percentage_edit(ha_client_f):
|
|
ha_client_f.post("/battery/add", data={"brand": "X", "count": "1"})
|
|
|
|
# Set via edit form
|
|
ha_client_f.post("/battery/1/edit-details", data={"battery_percentage": "55"})
|
|
resp = ha_client_f.get("/battery/1")
|
|
assert b"55%" in resp.data
|
|
|
|
# Clear via edit form
|
|
ha_client_f.post("/battery/1/edit-details", data={"battery_percentage": ""})
|
|
resp2 = ha_client_f.get("/battery/1")
|
|
# percentage display row should not appear when None
|
|
assert b"Battery %" not in resp2.data or b"55%" not in resp2.data
|