chore: commit local changes
This commit is contained in:
1 parent
c1336ad5b3
commit
e6e0832c8a
15 files changed
+291
-24
No files matched your search
@@ -217,6 +217,29 @@ async def test_sql_store_filters_errors_only(sql_store):
|
||||
assert only_errors[0]["error_type"] == "APITimeoutError"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_sql_store_filters_by_user_ids(sql_store):
|
||||
"""``user_ids`` (set by a fuzzy username search) matches several accounts."""
|
||||
base = datetime.now(UTC)
|
||||
for uid in ("alice", "alicia", "bob"):
|
||||
await sql_store.record(
|
||||
LlmMetricRecord(
|
||||
id=uuid.uuid4().hex,
|
||||
created_at=base,
|
||||
user_id=uid,
|
||||
duration_ms=1,
|
||||
status="success",
|
||||
)
|
||||
)
|
||||
# Fuzzy "ali" → {alice, alicia}; bob must be excluded.
|
||||
rows = await sql_store.list(LlmMetricsQuery(user_ids=["alice", "alicia"], limit=10))
|
||||
assert {r["user_id"] for r in rows} == {"alice", "alicia"}
|
||||
assert await sql_store.count(LlmMetricsQuery(user_ids=["alice", "alicia"])) == 2
|
||||
# user_ids takes precedence over a single user_id.
|
||||
rows = await sql_store.list(LlmMetricsQuery(user_id="bob", user_ids=["alice"], limit=10))
|
||||
assert {r["user_id"] for r in rows} == {"alice"}
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_sql_store_iter_all_paginates(sql_store):
|
||||
base = datetime.now(UTC)
|
||||
|
||||
Reference in new issue
Block a user