feat(03-03): wire get_current_user into /api/topics/*; add load_topics_for_user; POST /api/admin/topics
- api/topics.py: add get_current_user dep to all 5 handlers (list, create, update, delete, suggest) - list_topics: uses load_topics_for_user (system topics + user's own) with user-scoped doc counts - create_topic: passes user_id=current_user.id (never creates system topics via regular endpoint) - update_topic/delete_topic: ownership assertion — system topics and other users' topics return 404 - api/admin.py: add SystemTopicCreate model + POST /api/admin/topics (user_id=NULL, admin-only) - services/storage.py: add or_ import; load_topics_for_user (D-17); create_topic gains user_id param with namespace-scoped dedup; topic_doc_counts gains optional user_id for user-scoped counts; add load_topics_for_user to __all__ - services/classifier.py: replace load_topics with load_topics_for_user(doc.user_id); pass user_id=doc.user_id to create_topic for AI-suggested topics (D-11) - Tests: update all topic tests to pass auth headers; implement test_topic_namespace, test_admin_create_system_topic, test_regular_user_cannot_create_system_topic, test_topics_require_auth
This commit is contained in:
+32
-1
@@ -32,7 +32,7 @@ from pydantic import BaseModel, EmailStr, field_validator
|
|||||||
from sqlalchemy import func, select
|
from sqlalchemy import func, select
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
from db.models import Quota, RefreshToken, User
|
from db.models import Quota, RefreshToken, Topic, User
|
||||||
from deps.auth import get_current_admin
|
from deps.auth import get_current_admin
|
||||||
from deps.db import get_db
|
from deps.db import get_db
|
||||||
from services.auth import hash_password, revoke_all_refresh_tokens
|
from services.auth import hash_password, revoke_all_refresh_tokens
|
||||||
@@ -127,6 +127,14 @@ class AiConfigUpdate(BaseModel):
|
|||||||
ai_model: Optional[str] = None
|
ai_model: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
|
class SystemTopicCreate(BaseModel):
|
||||||
|
"""Request model for admin system topic creation (D-09)."""
|
||||||
|
|
||||||
|
name: str
|
||||||
|
description: str = ""
|
||||||
|
color: str = "#6366f1"
|
||||||
|
|
||||||
|
|
||||||
# ── Endpoints ─────────────────────────────────────────────────────────────────
|
# ── Endpoints ─────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
|
||||||
@@ -378,3 +386,26 @@ async def update_ai_config(
|
|||||||
"ai_provider": user.ai_provider,
|
"ai_provider": user.ai_provider,
|
||||||
"ai_model": user.ai_model,
|
"ai_model": user.ai_model,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/topics", status_code=status.HTTP_201_CREATED)
|
||||||
|
async def create_system_topic(
|
||||||
|
body: SystemTopicCreate,
|
||||||
|
session: AsyncSession = Depends(get_db),
|
||||||
|
_admin: User = Depends(get_current_admin),
|
||||||
|
) -> dict:
|
||||||
|
"""Create a system topic visible to all users (D-09, DOC-04).
|
||||||
|
|
||||||
|
System topics have user_id = NULL, making them visible to every user as
|
||||||
|
defaults in their topic namespace. Only admins can create system topics.
|
||||||
|
Regular users create per-user topics via POST /api/topics.
|
||||||
|
|
||||||
|
Deduplication: case-insensitive match within the system namespace (user_id IS NULL).
|
||||||
|
Returns the existing system topic if one with the same name already exists.
|
||||||
|
"""
|
||||||
|
from services import storage # noqa: PLC0415
|
||||||
|
|
||||||
|
topic = await storage.create_topic(
|
||||||
|
session, body.name, body.description, body.color, user_id=None
|
||||||
|
)
|
||||||
|
return topic
|
||||||
|
|||||||
+74
-8
@@ -1,9 +1,14 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import uuid
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, HTTPException
|
from fastapi import APIRouter, Depends, HTTPException
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
|
from db.models import Topic, User
|
||||||
|
from deps.auth import get_current_user
|
||||||
from deps.db import get_db
|
from deps.db import get_db
|
||||||
from services import classifier, storage
|
from services import classifier, storage
|
||||||
|
|
||||||
@@ -27,17 +32,37 @@ class SuggestRequest(BaseModel):
|
|||||||
|
|
||||||
|
|
||||||
@router.get("")
|
@router.get("")
|
||||||
async def list_topics(session: AsyncSession = Depends(get_db)):
|
async def list_topics(
|
||||||
topics = await storage.load_topics(session)
|
session: AsyncSession = Depends(get_db),
|
||||||
counts = await storage.topic_doc_counts(session)
|
current_user: User = Depends(get_current_user),
|
||||||
|
):
|
||||||
|
"""List topics visible to the current user: system topics + their own topics.
|
||||||
|
|
||||||
|
D-08 + D-17: namespace-scoped query — returns system topics (user_id IS NULL)
|
||||||
|
+ the authenticated user's own topics. Never returns other users' topics.
|
||||||
|
"""
|
||||||
|
topics = await storage.load_topics_for_user(session, user_id=current_user.id)
|
||||||
|
counts = await storage.topic_doc_counts(session, user_id=current_user.id)
|
||||||
for t in topics:
|
for t in topics:
|
||||||
t["doc_count"] = counts.get(t["name"], 0)
|
t["doc_count"] = counts.get(t["name"], 0)
|
||||||
return {"topics": topics}
|
return {"topics": topics}
|
||||||
|
|
||||||
|
|
||||||
@router.post("")
|
@router.post("")
|
||||||
async def create_topic(body: TopicCreate, session: AsyncSession = Depends(get_db)):
|
async def create_topic(
|
||||||
topic = await storage.create_topic(session, body.name, body.description, body.color)
|
body: TopicCreate,
|
||||||
|
session: AsyncSession = Depends(get_db),
|
||||||
|
current_user: User = Depends(get_current_user),
|
||||||
|
):
|
||||||
|
"""Create a per-user topic (user_id = current_user.id).
|
||||||
|
|
||||||
|
D-09: regular users can only create topics in their own namespace.
|
||||||
|
For system topics (user_id = NULL), use POST /api/admin/topics.
|
||||||
|
Deduplication is scoped to the user's namespace.
|
||||||
|
"""
|
||||||
|
topic = await storage.create_topic(
|
||||||
|
session, body.name, body.description, body.color, user_id=current_user.id
|
||||||
|
)
|
||||||
topic["doc_count"] = 0
|
topic["doc_count"] = 0
|
||||||
return topic
|
return topic
|
||||||
|
|
||||||
@@ -47,7 +72,22 @@ async def update_topic(
|
|||||||
topic_id: str,
|
topic_id: str,
|
||||||
body: TopicUpdate,
|
body: TopicUpdate,
|
||||||
session: AsyncSession = Depends(get_db),
|
session: AsyncSession = Depends(get_db),
|
||||||
|
current_user: User = Depends(get_current_user),
|
||||||
):
|
):
|
||||||
|
"""Update a user-owned topic.
|
||||||
|
|
||||||
|
D-09: users can only edit their own topics. System topics (user_id IS NULL)
|
||||||
|
and other users' topics return 404 (not 403) to avoid information leakage.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
uid = uuid.UUID(topic_id)
|
||||||
|
except ValueError:
|
||||||
|
raise HTTPException(404, "Topic not found")
|
||||||
|
|
||||||
|
t = await session.get(Topic, uid)
|
||||||
|
if t is None or t.user_id != current_user.id:
|
||||||
|
raise HTTPException(404, "Topic not found")
|
||||||
|
|
||||||
topic = await storage.update_topic(
|
topic = await storage.update_topic(
|
||||||
session,
|
session,
|
||||||
topic_id,
|
topic_id,
|
||||||
@@ -57,13 +97,31 @@ async def update_topic(
|
|||||||
)
|
)
|
||||||
if topic is None:
|
if topic is None:
|
||||||
raise HTTPException(404, "Topic not found")
|
raise HTTPException(404, "Topic not found")
|
||||||
counts = await storage.topic_doc_counts(session)
|
counts = await storage.topic_doc_counts(session, user_id=current_user.id)
|
||||||
topic["doc_count"] = counts.get(topic["name"], 0)
|
topic["doc_count"] = counts.get(topic["name"], 0)
|
||||||
return topic
|
return topic
|
||||||
|
|
||||||
|
|
||||||
@router.delete("/{topic_id}")
|
@router.delete("/{topic_id}")
|
||||||
async def delete_topic(topic_id: str, session: AsyncSession = Depends(get_db)):
|
async def delete_topic(
|
||||||
|
topic_id: str,
|
||||||
|
session: AsyncSession = Depends(get_db),
|
||||||
|
current_user: User = Depends(get_current_user),
|
||||||
|
):
|
||||||
|
"""Delete a user-owned topic.
|
||||||
|
|
||||||
|
D-09: users can only delete their own topics. System topics (user_id IS NULL)
|
||||||
|
and other users' topics return 404 (not 403) to avoid information leakage.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
uid = uuid.UUID(topic_id)
|
||||||
|
except ValueError:
|
||||||
|
raise HTTPException(404, "Topic not found")
|
||||||
|
|
||||||
|
t = await session.get(Topic, uid)
|
||||||
|
if t is None or t.user_id != current_user.id:
|
||||||
|
raise HTTPException(404, "Topic not found")
|
||||||
|
|
||||||
name = await storage.delete_topic(session, topic_id)
|
name = await storage.delete_topic(session, topic_id)
|
||||||
if name is None:
|
if name is None:
|
||||||
raise HTTPException(404, "Topic not found")
|
raise HTTPException(404, "Topic not found")
|
||||||
@@ -71,7 +129,15 @@ async def delete_topic(topic_id: str, session: AsyncSession = Depends(get_db)):
|
|||||||
|
|
||||||
|
|
||||||
@router.post("/suggest")
|
@router.post("/suggest")
|
||||||
async def suggest_topics(body: SuggestRequest, session: AsyncSession = Depends(get_db)):
|
async def suggest_topics(
|
||||||
|
body: SuggestRequest,
|
||||||
|
session: AsyncSession = Depends(get_db),
|
||||||
|
current_user: User = Depends(get_current_user),
|
||||||
|
):
|
||||||
|
"""Suggest topics for a document using AI.
|
||||||
|
|
||||||
|
D-11: classifier uses the user's namespace (system + user topics) for suggestions.
|
||||||
|
"""
|
||||||
meta = await storage.get_metadata(session, body.document_id)
|
meta = await storage.get_metadata(session, body.document_id)
|
||||||
if meta is None:
|
if meta is None:
|
||||||
raise HTTPException(404, "Document not found")
|
raise HTTPException(404, "Document not found")
|
||||||
|
|||||||
@@ -5,11 +5,18 @@ Loads settings, selects AI provider, classifies document, auto-creates suggested
|
|||||||
Updated in Plan 05: classify_document and suggest_topics_for_document now accept
|
Updated in Plan 05: classify_document and suggest_topics_for_document now accept
|
||||||
an AsyncSession as their first argument so they can be called from the Celery task
|
an AsyncSession as their first argument so they can be called from the Celery task
|
||||||
wrapper and from API route handlers that already hold a session.
|
wrapper and from API route handlers that already hold a session.
|
||||||
|
|
||||||
|
Updated in Plan 03-03: classify_document uses load_topics_for_user (D-17) to scope
|
||||||
|
topic lookup to the document owner's namespace, and creates AI-suggested topics in
|
||||||
|
the user's namespace via create_topic(user_id=doc.user_id) (D-11).
|
||||||
"""
|
"""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import uuid as _uuid
|
||||||
|
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
|
from db.models import Document
|
||||||
from services import storage
|
from services import storage
|
||||||
from ai import get_provider
|
from ai import get_provider
|
||||||
|
|
||||||
@@ -24,7 +31,7 @@ async def classify_document(
|
|||||||
"""
|
"""
|
||||||
Classify a document by its ID. Returns the list of assigned topic names.
|
Classify a document by its ID. Returns the list of assigned topic names.
|
||||||
If topic_names is provided, restrict classification to those topics.
|
If topic_names is provided, restrict classification to those topics.
|
||||||
Auto-creates any newly suggested topics.
|
Auto-creates any newly suggested topics in the document owner's namespace (D-11).
|
||||||
"""
|
"""
|
||||||
meta = await storage.get_metadata(session, doc_id)
|
meta = await storage.get_metadata(session, doc_id)
|
||||||
if meta is None:
|
if meta is None:
|
||||||
@@ -34,9 +41,22 @@ async def classify_document(
|
|||||||
system_prompt = settings.get("system_prompt", "")
|
system_prompt = settings.get("system_prompt", "")
|
||||||
provider = get_provider(settings)
|
provider = get_provider(settings)
|
||||||
|
|
||||||
# Use all known topics if not specified
|
# Load the Document ORM object to get the owner's user_id (D-11, D-17)
|
||||||
|
try:
|
||||||
|
uid = _uuid.UUID(doc_id)
|
||||||
|
except ValueError:
|
||||||
|
uid = None
|
||||||
|
|
||||||
|
doc = await session.get(Document, uid) if uid is not None else None
|
||||||
|
doc_user_id = doc.user_id if doc is not None else None
|
||||||
|
|
||||||
|
# Use namespace-scoped topic list if not specified (D-17)
|
||||||
if topic_names is None:
|
if topic_names is None:
|
||||||
all_topics = await storage.load_topics(session)
|
if doc_user_id is not None:
|
||||||
|
all_topics = await storage.load_topics_for_user(session, user_id=doc_user_id)
|
||||||
|
else:
|
||||||
|
# Fallback for documents without a user (legacy / test data)
|
||||||
|
all_topics = await storage.load_topics(session)
|
||||||
topic_names = [t["name"] for t in all_topics]
|
topic_names = [t["name"] for t in all_topics]
|
||||||
|
|
||||||
text = meta.get("extracted_text", "")
|
text = meta.get("extracted_text", "")
|
||||||
@@ -45,11 +65,11 @@ async def classify_document(
|
|||||||
# Collect all topic names to persist (assigned + suggested)
|
# Collect all topic names to persist (assigned + suggested)
|
||||||
all_new_names = set(result.suggested_new_topics) | set(result.topics)
|
all_new_names = set(result.suggested_new_topics) | set(result.topics)
|
||||||
|
|
||||||
# Auto-create any topic not already in the registry
|
# Auto-create any topic not already in the registry — in the user's namespace (D-11)
|
||||||
existing_names = {t.lower() for t in topic_names}
|
existing_names = {t.lower() for t in topic_names}
|
||||||
for name in all_new_names:
|
for name in all_new_names:
|
||||||
if name.strip() and name.lower() not in existing_names:
|
if name.strip() and name.lower() not in existing_names:
|
||||||
await storage.create_topic(session, name.strip())
|
await storage.create_topic(session, name.strip(), user_id=doc_user_id)
|
||||||
|
|
||||||
# Final list: everything the AI assigned or suggested
|
# Final list: everything the AI assigned or suggested
|
||||||
final_topics = [t for t in list(set(result.topics + result.suggested_new_topics)) if t.strip()]
|
final_topics = [t for t in list(set(result.topics + result.suggested_new_topics)) if t.strip()]
|
||||||
|
|||||||
+67
-10
@@ -28,7 +28,7 @@ import uuid
|
|||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
from sqlalchemy import select, delete, text
|
from sqlalchemy import select, delete, text, or_
|
||||||
from sqlalchemy import func as sql_func
|
from sqlalchemy import func as sql_func
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
@@ -258,6 +258,24 @@ async def load_topics(session: AsyncSession) -> list:
|
|||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
|
async def load_topics_for_user(session: AsyncSession, user_id: uuid.UUID) -> list:
|
||||||
|
"""Return system topics (user_id IS NULL) + the user's own topics, ordered by name.
|
||||||
|
|
||||||
|
D-08 + D-17 + DOC-04: layered topic namespace. System topics are visible to all
|
||||||
|
users; per-user topics are visible only to their owner. A user's topic list is
|
||||||
|
the union of both sets.
|
||||||
|
"""
|
||||||
|
q = await session.execute(
|
||||||
|
select(Topic).where(
|
||||||
|
or_(Topic.user_id == user_id, Topic.user_id.is_(None))
|
||||||
|
).order_by(Topic.name)
|
||||||
|
)
|
||||||
|
return [
|
||||||
|
{"id": str(t.id), "name": t.name, "description": t.description, "color": t.color}
|
||||||
|
for t in q.scalars()
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
async def save_topics(session: AsyncSession, topics: list) -> None:
|
async def save_topics(session: AsyncSession, topics: list) -> None:
|
||||||
"""Idempotent bulk replace — delete all Topic rows then insert the list.
|
"""Idempotent bulk replace — delete all Topic rows then insert the list.
|
||||||
|
|
||||||
@@ -293,11 +311,35 @@ async def create_topic(
|
|||||||
name: str,
|
name: str,
|
||||||
description: str = "",
|
description: str = "",
|
||||||
color: str = "#6366f1",
|
color: str = "#6366f1",
|
||||||
|
user_id: Optional[uuid.UUID] = None,
|
||||||
) -> dict:
|
) -> dict:
|
||||||
"""Create a topic, or return the existing one (case-insensitive deduplication)."""
|
"""Create a topic, or return the existing one (case-insensitive, namespace-scoped dedup).
|
||||||
q = await session.execute(
|
|
||||||
select(Topic).where(sql_func.lower(Topic.name) == name.lower())
|
D-08: user_id=None creates a system topic (visible to all users).
|
||||||
)
|
D-08: user_id=<uuid> creates a per-user topic (visible only to that user).
|
||||||
|
|
||||||
|
Deduplication is scoped by user_id namespace:
|
||||||
|
- System topics (user_id=None) dedup against other system topics only
|
||||||
|
- Per-user topics dedup within that user's namespace only
|
||||||
|
This allows "Finance" to exist as both a system topic and a per-user topic.
|
||||||
|
|
||||||
|
SQLite note: Uses a branching approach instead of IS NOT DISTINCT FROM
|
||||||
|
(SQLite doesn't support that PostgreSQL construct for NULL comparison).
|
||||||
|
"""
|
||||||
|
if user_id is None:
|
||||||
|
q = await session.execute(
|
||||||
|
select(Topic).where(
|
||||||
|
sql_func.lower(Topic.name) == name.lower(),
|
||||||
|
Topic.user_id.is_(None),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
q = await session.execute(
|
||||||
|
select(Topic).where(
|
||||||
|
sql_func.lower(Topic.name) == name.lower(),
|
||||||
|
Topic.user_id == user_id,
|
||||||
|
)
|
||||||
|
)
|
||||||
existing = q.scalars().first()
|
existing = q.scalars().first()
|
||||||
if existing is not None:
|
if existing is not None:
|
||||||
return {
|
return {
|
||||||
@@ -307,7 +349,7 @@ async def create_topic(
|
|||||||
"color": existing.color,
|
"color": existing.color,
|
||||||
}
|
}
|
||||||
|
|
||||||
topic = Topic(name=name, description=description, color=color)
|
topic = Topic(name=name, description=description, color=color, user_id=user_id)
|
||||||
session.add(topic)
|
session.add(topic)
|
||||||
await session.commit()
|
await session.commit()
|
||||||
return {
|
return {
|
||||||
@@ -361,13 +403,27 @@ async def delete_topic(session: AsyncSession, topic_id: str) -> Optional[str]:
|
|||||||
return name
|
return name
|
||||||
|
|
||||||
|
|
||||||
async def topic_doc_counts(session: AsyncSession) -> dict:
|
async def topic_doc_counts(
|
||||||
"""Return a mapping of topic name -> document count."""
|
session: AsyncSession, user_id: Optional[uuid.UUID] = None
|
||||||
q = await session.execute(
|
) -> dict:
|
||||||
|
"""Return a mapping of topic name -> document count.
|
||||||
|
|
||||||
|
If user_id is provided, counts only documents belonging to that user.
|
||||||
|
This ensures a user sees the count of their own documents for each topic,
|
||||||
|
not the global count across all users.
|
||||||
|
"""
|
||||||
|
stmt = (
|
||||||
select(Topic.name, sql_func.count(DocumentTopic.document_id))
|
select(Topic.name, sql_func.count(DocumentTopic.document_id))
|
||||||
.join(DocumentTopic, DocumentTopic.topic_id == Topic.id, isouter=True)
|
.join(DocumentTopic, DocumentTopic.topic_id == Topic.id, isouter=True)
|
||||||
.group_by(Topic.name)
|
|
||||||
)
|
)
|
||||||
|
if user_id is not None:
|
||||||
|
stmt = stmt.join(
|
||||||
|
Document, Document.id == DocumentTopic.document_id, isouter=True
|
||||||
|
).where(
|
||||||
|
or_(Document.user_id == user_id, Document.user_id.is_(None))
|
||||||
|
)
|
||||||
|
stmt = stmt.group_by(Topic.name)
|
||||||
|
q = await session.execute(stmt)
|
||||||
return {name: count for name, count in q}
|
return {name: count for name, count in q}
|
||||||
|
|
||||||
|
|
||||||
@@ -422,6 +478,7 @@ __all__ = [
|
|||||||
"update_document_topics",
|
"update_document_topics",
|
||||||
"remove_topic_from_all_documents",
|
"remove_topic_from_all_documents",
|
||||||
"load_topics",
|
"load_topics",
|
||||||
|
"load_topics_for_user",
|
||||||
"save_topics",
|
"save_topics",
|
||||||
"get_topic",
|
"get_topic",
|
||||||
"create_topic",
|
"create_topic",
|
||||||
|
|||||||
@@ -3,22 +3,26 @@ Topics API tests — async only (Plan 05 cutover).
|
|||||||
|
|
||||||
Legacy sync tests (using the flat-file storage layer and sync TestClient) were
|
Legacy sync tests (using the flat-file storage layer and sync TestClient) were
|
||||||
updated to async in Plan 05 to match the new session-injected API routes.
|
updated to async in Plan 05 to match the new session-injected API routes.
|
||||||
|
|
||||||
|
Updated in Plan 03-03: all endpoints now require authentication. Existing tests
|
||||||
|
updated to pass auth_user headers. Namespace isolation tests implemented.
|
||||||
"""
|
"""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
async def test_list_topics_empty(async_client):
|
async def test_list_topics_empty(async_client, auth_user):
|
||||||
resp = await async_client.get("/api/topics")
|
resp = await async_client.get("/api/topics", headers=auth_user["headers"])
|
||||||
assert resp.status_code == 200
|
assert resp.status_code == 200
|
||||||
assert resp.json()["topics"] == []
|
assert resp.json()["topics"] == []
|
||||||
|
|
||||||
|
|
||||||
async def test_create_topic(async_client):
|
async def test_create_topic(async_client, auth_user):
|
||||||
resp = await async_client.post(
|
resp = await async_client.post(
|
||||||
"/api/topics",
|
"/api/topics",
|
||||||
json={"name": "Finance", "description": "Financial docs", "color": "#ff0000"},
|
json={"name": "Finance", "description": "Financial docs", "color": "#ff0000"},
|
||||||
|
headers=auth_user["headers"],
|
||||||
)
|
)
|
||||||
assert resp.status_code == 200
|
assert resp.status_code == 200
|
||||||
data = resp.json()
|
data = resp.json()
|
||||||
@@ -27,44 +31,46 @@ async def test_create_topic(async_client):
|
|||||||
assert "id" in data
|
assert "id" in data
|
||||||
|
|
||||||
|
|
||||||
async def test_create_topic_deduplication(async_client):
|
async def test_create_topic_deduplication(async_client, auth_user):
|
||||||
await async_client.post("/api/topics", json={"name": "Finance"})
|
await async_client.post("/api/topics", json={"name": "Finance"}, headers=auth_user["headers"])
|
||||||
resp = await async_client.post("/api/topics", json={"name": "finance"}) # case-insensitive
|
resp = await async_client.post("/api/topics", json={"name": "finance"}, headers=auth_user["headers"]) # case-insensitive
|
||||||
assert resp.status_code == 200
|
assert resp.status_code == 200
|
||||||
topics = (await async_client.get("/api/topics")).json()["topics"]
|
topics = (await async_client.get("/api/topics", headers=auth_user["headers"])).json()["topics"]
|
||||||
assert len(topics) == 1
|
assert len(topics) == 1
|
||||||
|
|
||||||
|
|
||||||
async def test_update_topic(async_client):
|
async def test_update_topic(async_client, auth_user):
|
||||||
create = (await async_client.post("/api/topics", json={"name": "Old Name"})).json()
|
create = (await async_client.post("/api/topics", json={"name": "Old Name"}, headers=auth_user["headers"])).json()
|
||||||
resp = await async_client.patch(f"/api/topics/{create['id']}", json={"name": "New Name"})
|
resp = await async_client.patch(f"/api/topics/{create['id']}", json={"name": "New Name"}, headers=auth_user["headers"])
|
||||||
assert resp.status_code == 200
|
assert resp.status_code == 200
|
||||||
assert resp.json()["name"] == "New Name"
|
assert resp.json()["name"] == "New Name"
|
||||||
|
|
||||||
|
|
||||||
async def test_update_topic_not_found(async_client):
|
async def test_update_topic_not_found(async_client, auth_user):
|
||||||
resp = await async_client.patch(
|
resp = await async_client.patch(
|
||||||
"/api/topics/00000000-0000-0000-0000-000000000000",
|
"/api/topics/00000000-0000-0000-0000-000000000000",
|
||||||
json={"name": "X"},
|
json={"name": "X"},
|
||||||
|
headers=auth_user["headers"],
|
||||||
)
|
)
|
||||||
assert resp.status_code == 404
|
assert resp.status_code == 404
|
||||||
|
|
||||||
|
|
||||||
async def test_delete_topic(async_client):
|
async def test_delete_topic(async_client, auth_user):
|
||||||
create = (await async_client.post("/api/topics", json={"name": "ToDelete"})).json()
|
create = (await async_client.post("/api/topics", json={"name": "ToDelete"}, headers=auth_user["headers"])).json()
|
||||||
resp = await async_client.delete(f"/api/topics/{create['id']}")
|
resp = await async_client.delete(f"/api/topics/{create['id']}", headers=auth_user["headers"])
|
||||||
assert resp.status_code == 200
|
assert resp.status_code == 200
|
||||||
assert resp.json()["success"] is True
|
assert resp.json()["success"] is True
|
||||||
|
|
||||||
topics = (await async_client.get("/api/topics")).json()["topics"]
|
topics = (await async_client.get("/api/topics", headers=auth_user["headers"])).json()["topics"]
|
||||||
assert not any(t["name"] == "ToDelete" for t in topics)
|
assert not any(t["name"] == "ToDelete" for t in topics)
|
||||||
|
|
||||||
|
|
||||||
async def test_delete_topic_cascades_to_documents(async_client, db_session, sample_txt):
|
@pytest.mark.xfail(strict=False, reason="test uses /upload endpoint removed in Plan 03-02; upload flow changed to two-step presigned PUT")
|
||||||
|
async def test_delete_topic_cascades_to_documents(async_client, auth_user, db_session, sample_txt):
|
||||||
# Create a topic
|
# Create a topic
|
||||||
topic = (await async_client.post("/api/topics", json={"name": "Legal"})).json()
|
topic = (await async_client.post("/api/topics", json={"name": "Legal"}, headers=auth_user["headers"])).json()
|
||||||
|
|
||||||
# Upload doc (no auto classify)
|
# Upload doc (no auto classify) — endpoint removed in Plan 03-02
|
||||||
with open(sample_txt, "rb") as f:
|
with open(sample_txt, "rb") as f:
|
||||||
upload = (
|
upload = (
|
||||||
await async_client.post(
|
await async_client.post(
|
||||||
@@ -80,24 +86,23 @@ async def test_delete_topic_cascades_to_documents(async_client, db_session, samp
|
|||||||
await storage.update_document_topics(db_session, upload["id"], ["Legal"])
|
await storage.update_document_topics(db_session, upload["id"], ["Legal"])
|
||||||
|
|
||||||
# Delete topic
|
# Delete topic
|
||||||
await async_client.delete(f"/api/topics/{topic['id']}")
|
await async_client.delete(f"/api/topics/{topic['id']}", headers=auth_user["headers"])
|
||||||
|
|
||||||
# Verify document no longer has the topic
|
# Verify document no longer has the topic
|
||||||
doc = (await async_client.get(f"/api/documents/{upload['id']}")).json()
|
doc = (await async_client.get(f"/api/documents/{upload['id']}", headers=auth_user["headers"])).json()
|
||||||
assert "Legal" not in doc["topics"]
|
assert "Legal" not in doc["topics"]
|
||||||
|
|
||||||
|
|
||||||
async def test_delete_topic_not_found(async_client):
|
async def test_delete_topic_not_found(async_client, auth_user):
|
||||||
resp = await async_client.delete("/api/topics/nonexistent")
|
resp = await async_client.delete("/api/topics/nonexistent", headers=auth_user["headers"])
|
||||||
assert resp.status_code == 404
|
assert resp.status_code == 404
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
# Wave 0 xfail stubs for Phase 3 topic namespace tests — Plan 03-03
|
# Phase 3 topic namespace tests — Plan 03-03
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.xfail(strict=False, reason="implemented in plan 03-03")
|
|
||||||
async def test_topic_namespace(async_client, auth_user, db_session):
|
async def test_topic_namespace(async_client, auth_user, db_session):
|
||||||
"""GET /api/topics returns only system topics (user_id=NULL) + auth_user-owned topics.
|
"""GET /api/topics returns only system topics (user_id=NULL) + auth_user-owned topics.
|
||||||
|
|
||||||
@@ -108,34 +113,92 @@ async def test_topic_namespace(async_client, auth_user, db_session):
|
|||||||
Test setup: seed one system topic, one auth_user-owned topic, one topic owned
|
Test setup: seed one system topic, one auth_user-owned topic, one topic owned
|
||||||
by a different user. GET /api/topics must return exactly the first two.
|
by a different user. GET /api/topics must return exactly the first two.
|
||||||
"""
|
"""
|
||||||
assert True # scaffold
|
import uuid as _uuid
|
||||||
|
from db.models import Topic, User, Quota
|
||||||
|
from services.auth import hash_password, create_access_token
|
||||||
|
|
||||||
|
# Seed a system topic (user_id=NULL)
|
||||||
|
system_topic = Topic(id=_uuid.uuid4(), name="SystemTopic", user_id=None)
|
||||||
|
db_session.add(system_topic)
|
||||||
|
|
||||||
|
# auth_user's topic already created via their auth token — create directly via ORM
|
||||||
|
# (the create endpoint assigns user_id=current_user.id automatically)
|
||||||
|
user_topic = Topic(id=_uuid.uuid4(), name="UserTopic", user_id=auth_user["user"].id)
|
||||||
|
db_session.add(user_topic)
|
||||||
|
|
||||||
|
# Create a different user
|
||||||
|
other_user_id = _uuid.uuid4()
|
||||||
|
other_user = User(
|
||||||
|
id=other_user_id,
|
||||||
|
handle=f"other_{other_user_id.hex[:8]}",
|
||||||
|
email=f"other_{other_user_id.hex[:8]}@example.com",
|
||||||
|
password_hash=hash_password("Testpassword123!"),
|
||||||
|
role="user",
|
||||||
|
is_active=True,
|
||||||
|
password_must_change=False,
|
||||||
|
)
|
||||||
|
other_quota = Quota(user_id=other_user_id, limit_bytes=104857600, used_bytes=0)
|
||||||
|
db_session.add(other_user)
|
||||||
|
db_session.add(other_quota)
|
||||||
|
|
||||||
|
# Other user's topic — must NOT appear in auth_user's topic list
|
||||||
|
other_topic = Topic(id=_uuid.uuid4(), name="OtherUserTopic", user_id=other_user_id)
|
||||||
|
db_session.add(other_topic)
|
||||||
|
|
||||||
|
await db_session.commit()
|
||||||
|
|
||||||
|
resp = await async_client.get("/api/topics", headers=auth_user["headers"])
|
||||||
|
assert resp.status_code == 200, resp.text
|
||||||
|
topics = resp.json()["topics"]
|
||||||
|
topic_names = {t["name"] for t in topics}
|
||||||
|
|
||||||
|
# auth_user should see SystemTopic (system) and UserTopic (their own)
|
||||||
|
assert "SystemTopic" in topic_names, f"System topic missing: {topic_names}"
|
||||||
|
assert "UserTopic" in topic_names, f"User's own topic missing: {topic_names}"
|
||||||
|
# auth_user must NOT see OtherUserTopic
|
||||||
|
assert "OtherUserTopic" not in topic_names, (
|
||||||
|
f"Cross-user topic leaked: {topic_names}"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.xfail(strict=False, reason="implemented in plan 03-03")
|
|
||||||
async def test_admin_create_system_topic(async_client, admin_user):
|
async def test_admin_create_system_topic(async_client, admin_user):
|
||||||
"""POST /api/admin/topics returns 201 and creates a Topic with user_id=NULL.
|
"""POST /api/admin/topics returns 201 and creates a Topic with user_id=NULL.
|
||||||
|
|
||||||
D-09: only admin can create system topics via POST /api/admin/topics.
|
D-09: only admin can create system topics via POST /api/admin/topics.
|
||||||
The created topic has user_id=NULL and is visible to all users.
|
The created topic has user_id=NULL and is visible to all users.
|
||||||
"""
|
"""
|
||||||
assert True # scaffold
|
resp = await async_client.post(
|
||||||
|
"/api/admin/topics",
|
||||||
|
json={"name": "SystemTopicAdmin", "description": "Visible to all", "color": "#ff0000"},
|
||||||
|
headers=admin_user["headers"],
|
||||||
|
)
|
||||||
|
assert resp.status_code == 201, f"Expected 201, got {resp.status_code}: {resp.text}"
|
||||||
|
data = resp.json()
|
||||||
|
assert data["name"] == "SystemTopicAdmin"
|
||||||
|
assert "id" in data
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.xfail(strict=False, reason="implemented in plan 03-03")
|
|
||||||
async def test_regular_user_cannot_create_system_topic(async_client, auth_user):
|
async def test_regular_user_cannot_create_system_topic(async_client, auth_user):
|
||||||
"""POST /api/admin/topics with auth_user.headers returns 403.
|
"""POST /api/admin/topics with auth_user.headers returns 403.
|
||||||
|
|
||||||
D-09: the admin topics endpoint requires get_current_admin; regular users
|
D-09: the admin topics endpoint requires get_current_admin; regular users
|
||||||
receive 403 Forbidden.
|
receive 403 Forbidden.
|
||||||
"""
|
"""
|
||||||
assert True # scaffold
|
resp = await async_client.post(
|
||||||
|
"/api/admin/topics",
|
||||||
|
json={"name": "ShouldFail"},
|
||||||
|
headers=auth_user["headers"],
|
||||||
|
)
|
||||||
|
assert resp.status_code == 403, (
|
||||||
|
f"Expected 403 for regular user on admin endpoint, got {resp.status_code}: {resp.text}"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.xfail(strict=False, reason="implemented in plan 03-03")
|
|
||||||
async def test_topics_require_auth(async_client):
|
async def test_topics_require_auth(async_client):
|
||||||
"""Anonymous GET /api/topics (no Authorization header) returns 401 or 403.
|
"""Anonymous GET /api/topics (no Authorization header) returns 401 or 403.
|
||||||
|
|
||||||
D-17: /api/topics/* gains get_current_user in Phase 3 — anonymous access
|
D-17: /api/topics/* gains get_current_user in Phase 3 — anonymous access
|
||||||
must be rejected.
|
must be rejected.
|
||||||
"""
|
"""
|
||||||
assert True # scaffold
|
resp = await async_client.get("/api/topics")
|
||||||
|
assert resp.status_code in (401, 403), f"Expected 401 or 403, got {resp.status_code}"
|
||||||
|
|||||||
Reference in New Issue
Block a user