Ewald
d3fcc16980
- Add support for various time units (ms, s, m, h) in generation time - Enhance parsing logic to handle different input formats and invalid values - Update database write and hydration processes to preserve generation time - Add tests for new parsing capabilities and ensure generation time is maintained across quality upgrades
128 lines
3.7 KiB
Python
128 lines
3.7 KiB
Python
import datetime
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from mjr_am_backend.routes.handlers import search_impl as s
|
|
|
|
|
|
def test_dedupe_assets_payload_total_adjusted() -> None:
|
|
payload = {
|
|
"assets": [
|
|
{"filepath": "A"},
|
|
{"filepath": "A"},
|
|
{"filepath": "B"},
|
|
{"x": 1},
|
|
],
|
|
"total": 99,
|
|
}
|
|
out = s._dedupe_result_assets_payload(payload)
|
|
assert len(out["assets"]) == 3
|
|
assert out["total"] == 99
|
|
|
|
|
|
def test_collect_hydration_paths_skips_folder_and_invalid() -> None:
|
|
assets: list[Any] = [
|
|
{"kind": "folder", "filepath": "C:/x"},
|
|
{"kind": "image", "filepath": "C:/y.png"},
|
|
"bad",
|
|
{},
|
|
]
|
|
assert s._collect_hydration_filepaths(assets) == ["C:/y.png"]
|
|
|
|
|
|
def test_index_rows_and_hydrate_asset_from_row() -> None:
|
|
rows = [
|
|
{
|
|
"filepath": "C:/a.png",
|
|
"id": 3,
|
|
"rating": 5,
|
|
"tags": '["x"]',
|
|
"has_workflow": 1,
|
|
"has_generation_data": 1,
|
|
"workflow_type": "T2I",
|
|
"positive_prompt": "cinematic portrait",
|
|
"generation_time_ms": "9100",
|
|
}
|
|
]
|
|
by = s._index_rows_by_filepath(rows)
|
|
asset = {"kind": "image", "filepath": "C:/a.png"}
|
|
s._hydrate_browser_asset_from_row(asset, by)
|
|
assert asset["id"] == 3
|
|
assert asset["rating"] == 5
|
|
assert asset["tags"] == ["x"]
|
|
assert asset["has_workflow"] == 1
|
|
assert asset["has_generation_data"] == 1
|
|
assert asset["workflow_type"] == "T2I"
|
|
assert asset["positive_prompt"] == "cinematic portrait"
|
|
assert asset["generation_time_ms"] == 9100
|
|
|
|
|
|
def test_coerce_browser_tags_variants() -> None:
|
|
assert s._coerce_browser_tags('["a",1]') == ["a"]
|
|
assert s._coerce_browser_tags(["a", 1]) == ["a"]
|
|
assert s._coerce_browser_tags("bad-json") == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_hydrate_browser_assets_from_db_no_db_returns_input() -> None:
|
|
assets = [{"filepath": "C:/a.png"}]
|
|
out = await s._hydrate_browser_assets_from_db(None, assets)
|
|
assert out is assets
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_hydrate_browser_assets_from_db_applies_rows(monkeypatch) -> None:
|
|
assets = [{"kind": "image", "filepath": "C:/a.png"}]
|
|
|
|
class _DB:
|
|
async def aquery_in(self, *_args, **_kwargs):
|
|
return type(
|
|
"R",
|
|
(),
|
|
{
|
|
"ok": True,
|
|
"data": [
|
|
{
|
|
"filepath": "C:/a.png",
|
|
"id": 9,
|
|
"rating": 4,
|
|
"tags": [],
|
|
"generation_time_ms": 7250,
|
|
}
|
|
],
|
|
},
|
|
)
|
|
|
|
out = await s._hydrate_browser_assets_from_db({"db": _DB()}, assets)
|
|
assert out[0]["id"] == 9
|
|
assert out[0]["generation_time_ms"] == 7250
|
|
|
|
|
|
def test_date_bounds_helpers() -> None:
|
|
ref = datetime.datetime(2026, 2, 25, 10, 0, tzinfo=datetime.timezone.utc)
|
|
start, end = s._date_bounds_for_range("today", reference=ref)
|
|
assert isinstance(start, int)
|
|
assert end > start
|
|
|
|
start2, end2 = s._date_bounds_for_exact("2026-02-01")
|
|
assert isinstance(start2, int)
|
|
assert end2 > start2
|
|
|
|
bad = s._date_bounds_for_exact("bad")
|
|
assert bad == (None, None)
|
|
|
|
|
|
def test_exclude_assets_under_root(tmp_path) -> None:
|
|
root = tmp_path / "root"
|
|
root.mkdir()
|
|
inside = root / "a.png"
|
|
inside.write_text("x")
|
|
outside = tmp_path / "b.png"
|
|
outside.write_text("y")
|
|
|
|
out = s._exclude_assets_under_root(
|
|
[{"filepath": str(inside)}, {"filepath": str(outside)}, {"x": 1}],
|
|
str(root),
|
|
)
|
|
assert len(out) == 2
|