diff --git a/tests/test_provisioning.py b/tests/test_provisioning.py index 68bed6c..9bdf34d 100644 --- a/tests/test_provisioning.py +++ b/tests/test_provisioning.py @@ -1553,7 +1553,11 @@ def test_provisioning_vaultwarden_rate_limited(monkeypatch) -> None: conn = DummyConn(row) manager._ensure_vaultwarden_invite(conn, ctx) - inserts = [params for query, params in conn.executed if "access_request_tasks" in query] + inserts = [ + params + for query, params in conn.executed + if "INSERT INTO access_request_tasks" in query and isinstance(params, tuple) and len(params) >= 4 + ] assert any(params[2] == "pending" and "rate limited until" in (params[3] or "") for params in inserts) @@ -1607,7 +1611,11 @@ def test_provisioning_vaultwarden_grandfathered(monkeypatch) -> None: conn = DummyConn(row) manager._ensure_vaultwarden_invite(conn, ctx) - inserts = [params for query, params in conn.executed if "access_request_tasks" in query] + inserts = [ + params + for query, params in conn.executed + if "INSERT INTO access_request_tasks" in query and isinstance(params, tuple) and len(params) >= 4 + ] assert any(params[2] == "ok" and params[3] == "grandfathered" for params in inserts)