77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
from zotero_kb.cards import CardBuilder
|
|
from zotero_kb.config import AppConfig
|
|
from zotero_kb.projects import ProjectService
|
|
from zotero_kb.workspace import Workspace
|
|
from zotero_kb.zotero_reader import ZoteroItemRecord
|
|
|
|
|
|
class FakeLlmClient:
|
|
def generate_card(self, source_bundle: dict[str, object]) -> dict[str, object]:
|
|
title = str(source_bundle["title"])
|
|
if "Scoped" in title:
|
|
summary = "Project-scoped retrieval improves drafting."
|
|
claims = ["Project scoping improves citation precision."]
|
|
else:
|
|
summary = "Unrelated retrieval baseline."
|
|
claims = ["Baseline retrieval is broad."]
|
|
return {
|
|
"summary": summary,
|
|
"core_claims": claims,
|
|
"methods": ["Method details."],
|
|
"evidence": ["Evidence details."],
|
|
"quotable_passages": claims,
|
|
"writing_hints": [summary],
|
|
"keywords": ["retrieval"],
|
|
}
|
|
|
|
|
|
def _build_item(item_key: str, title: str) -> ZoteroItemRecord:
|
|
return ZoteroItemRecord(
|
|
item_key=item_key,
|
|
title=title,
|
|
creators=["Alice Smith"],
|
|
year="2024",
|
|
item_type="journalArticle",
|
|
abstract=title,
|
|
tags=["retrieval"],
|
|
collection_paths=[["Theory", "Drafting"]],
|
|
notes=[title],
|
|
attachment_texts=[title],
|
|
)
|
|
|
|
|
|
def test_add_item_to_project_updates_selected_items_and_project_index(tmp_path: Path) -> None:
|
|
config = AppConfig(workspace_dir=tmp_path / "workspace", zotero_data_dir=tmp_path / "zotero")
|
|
workspace = Workspace(config)
|
|
workspace.create_project("thesis-ch2", "Thesis Chapter 2", "openai", "gpt-5-mini")
|
|
builder = CardBuilder(config.workspace_dir, FakeLlmClient())
|
|
builder.build_or_update(_build_item("PAPER0001", "Scoped Retrieval for Drafting"))
|
|
|
|
service = ProjectService(config.workspace_dir)
|
|
payload = service.add_items("thesis-ch2", ["PAPER0001"])
|
|
|
|
assert payload["selected_items"] == ["PAPER0001"]
|
|
assert payload["cards"][0]["item_key"] == "PAPER0001"
|
|
|
|
selected_items = json.loads(
|
|
(config.workspace_dir / "projects" / "thesis-ch2" / "selected-items.json").read_text(encoding="utf-8")
|
|
)
|
|
assert selected_items == ["PAPER0001"]
|
|
|
|
|
|
def test_remove_item_from_project_updates_selected_items(tmp_path: Path) -> None:
|
|
config = AppConfig(workspace_dir=tmp_path / "workspace", zotero_data_dir=tmp_path / "zotero")
|
|
workspace = Workspace(config)
|
|
workspace.create_project("thesis-ch2", "Thesis Chapter 2", "openai", "gpt-5-mini")
|
|
builder = CardBuilder(config.workspace_dir, FakeLlmClient())
|
|
builder.build_or_update(_build_item("PAPER0001", "Scoped Retrieval for Drafting"))
|
|
|
|
service = ProjectService(config.workspace_dir)
|
|
service.add_items("thesis-ch2", ["PAPER0001"])
|
|
payload = service.remove_item("thesis-ch2", "PAPER0001")
|
|
|
|
assert payload["selected_items"] == []
|