Compare commits

...

4 Commits

4 changed files with 32 additions and 22 deletions

View File

@@ -1,5 +1,8 @@
from __future__ import annotations from __future__ import annotations
from dataclasses import dataclass, field
from typing import Any, cast
from auditui.app import Auditui from auditui.app import Auditui
from auditui.search_utils import build_search_text, filter_items from auditui.search_utils import build_search_text, filter_items
@@ -12,16 +15,17 @@ class StubLibrary:
return item.get("authors", "") return item.get("authors", "")
def test_get_search_text_is_cached() -> None: @dataclass(slots=True)
class Dummy: class Dummy:
def __init__(self) -> None: _search_text_cache: dict[int, str] = field(default_factory=dict)
self._search_text_cache: dict[int, str] = {} library_client: StubLibrary = field(default_factory=StubLibrary)
self.library_client = StubLibrary()
def test_get_search_text_is_cached() -> None:
item = {"title": "Title", "authors": "Author"} item = {"title": "Title", "authors": "Author"}
dummy = Dummy() dummy = Dummy()
first = Auditui._get_search_text(dummy, item) first = Auditui._get_search_text(cast(Auditui, dummy), item)
second = Auditui._get_search_text(dummy, item) second = Auditui._get_search_text(cast(Auditui, dummy), item)
assert first == "title author" assert first == "title author"
assert first == second assert first == second
assert len(dummy._search_text_cache) == 1 assert len(dummy._search_text_cache) == 1
@@ -38,7 +42,7 @@ def test_filter_items_uses_cache() -> None:
def cached(item: dict) -> str: def cached(item: dict) -> str:
cache_key = id(item) cache_key = id(item)
if cache_key not in cache: if cache_key not in cache:
cache[cache_key] = build_search_text(item, library) cache[cache_key] = build_search_text(item, cast(Any, library))
return cache[cache_key] return cache[cache_key]
result = filter_items(items, "beta", cached) result = filter_items(items, "beta", cached)

View File

@@ -1,12 +1,14 @@
from dataclasses import dataclass, field
from auditui.library import LibraryClient from auditui.library import LibraryClient
@dataclass(slots=True)
class MockClient: class MockClient:
def __init__(self) -> None: put_calls: list[tuple[str, dict]] = field(default_factory=list)
self.put_calls: list[tuple[str, dict]] = [] post_calls: list[tuple[str, dict]] = field(default_factory=list)
self.post_calls: list[tuple[str, dict]] = [] _post_response: dict = field(default_factory=dict)
self._post_response: dict = {} raise_on_put: bool = False
self.raise_on_put = False
def put(self, path: str, body: dict) -> dict: def put(self, path: str, body: dict) -> dict:
if self.raise_on_put: if self.raise_on_put:

View File

@@ -1,3 +1,6 @@
from dataclasses import dataclass
from typing import Any, cast
from auditui import table_utils from auditui import table_utils
@@ -25,9 +28,9 @@ class StubLibrary:
return item.get("asin") return item.get("asin")
@dataclass(slots=True)
class StubDownloads: class StubDownloads:
def __init__(self, cached: set[str]) -> None: _cached: set[str]
self._cached = cached
def is_cached(self, asin: str) -> bool: def is_cached(self, asin: str) -> bool:
return asin in self._cached return asin in self._cached
@@ -63,7 +66,7 @@ def test_format_item_as_row_with_downloaded() -> None:
"asin": "ASIN123", "asin": "ASIN123",
} }
title, author, runtime, progress, downloaded = table_utils.format_item_as_row( title, author, runtime, progress, downloaded = table_utils.format_item_as_row(
item, library, downloads item, library, cast(Any, downloads)
) )
assert title == "Title" assert title == "Title"
assert author == "Author One" assert author == "Author One"

View File

@@ -1,4 +1,5 @@
import json import json
from dataclasses import dataclass, field
from pathlib import Path from pathlib import Path
import pytest import pytest
@@ -6,13 +7,13 @@ import pytest
from auditui import ui from auditui import ui
@dataclass(slots=True)
class DummyApp: class DummyApp:
def __init__(self) -> None: client: object | None = None
self.client = None auth: object | None = None
self.auth = None library_client: object | None = None
self.library_client = None all_items: list[dict] = field(default_factory=list)
self.all_items = [] BINDINGS: list[tuple[str, str, str]] = field(default_factory=list)
self.BINDINGS = []
@pytest.fixture @pytest.fixture