first

backend/tests/test_actions.py

@@ -0,0 +1,80 @@
+from datetime import datetime
+
+from sqlmodel import Session
+
+from app.models import EntityCache, EntityFlag, AuditLog
+
+
+def _seed_entity(session: Session):
+    session.add(EntityCache(
+        entity_id="light.test",
+        domain="light",
+        friendly_name="Test",
+        state="on",
+        fetched_at=datetime.utcnow(),
+    ))
+    session.commit()
+
+
+def test_favorite_entity(client, session):
+    _seed_entity(session)
+    resp = client.post("/api/entities/actions", json={
+        "action": "favorite",
+        "entity_ids": ["light.test"],
+    })
+    assert resp.status_code == 200
+    data = resp.json()
+    assert data["action"] == "favorite"
+    assert data["results"][0]["ok"] is True
+
+    # Vérifier le flag
+    flag = session.get(EntityFlag, "light.test")
+    assert flag is not None
+    assert flag.favorite is True
+
+
+def test_unfavorite_entity(client, session):
+    _seed_entity(session)
+    # D'abord favori
+    client.post("/api/entities/actions", json={
+        "action": "favorite",
+        "entity_ids": ["light.test"],
+    })
+    # Puis défavori
+    resp = client.post("/api/entities/actions", json={
+        "action": "unfavorite",
+        "entity_ids": ["light.test"],
+    })
+    assert resp.status_code == 200
+    flag = session.get(EntityFlag, "light.test")
+    assert flag.favorite is False
+
+
+def test_ignore_entity(client, session):
+    _seed_entity(session)
+    resp = client.post("/api/entities/actions", json={
+        "action": "ignore",
+        "entity_ids": ["light.test"],
+    })
+    assert resp.status_code == 200
+    flag = session.get(EntityFlag, "light.test")
+    assert flag.ignored_local is True
+
+
+def test_bulk_action(client, session):
+    session.add(EntityCache(entity_id="light.a", domain="light", state="on", fetched_at=datetime.utcnow()))
+    session.add(EntityCache(entity_id="light.b", domain="light", state="off", fetched_at=datetime.utcnow()))
+    session.commit()
+
+    resp = client.post("/api/entities/actions", json={
+        "action": "favorite",
+        "entity_ids": ["light.a", "light.b"],
+    })
+    data = resp.json()
+    assert len(data["results"]) == 2
+
+    # Vérifier audit_log
+    from sqlmodel import select
+    logs = session.exec(select(AuditLog)).all()
+    assert len(logs) >= 1
+    assert logs[-1].action == "favorite"