wehub-resource-sync
fed8b2eed7
Backend release / release (push) Waiting to run
Bandit Security Scan / bandit_scan (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / build (linux/amd64, ubuntu-latest, amd64) (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / manifest (push) Blocked by required conditions
Build and push DocsGPT FE Docker image for development / build (linux/amd64, ubuntu-latest, amd64) (push) Waiting to run
Build and push DocsGPT FE Docker image for development / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Waiting to run
Build and push DocsGPT FE Docker image for development / manifest (push) Blocked by required conditions
Python linting / ruff (push) Waiting to run
Run python tests with pytest / Run tests and count coverage (3.12) (push) Waiting to run
React Widget Build / build (push) Waiting to run
108 lines
4.0 KiB
Python
108 lines
4.0 KiB
Python
"""Tests for the generalized GET /api/tools/artifact/<id> document branch."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
from contextlib import contextmanager
|
|
|
|
import pytest
|
|
from flask import request
|
|
|
|
from application.storage.db.repositories.artifacts import ArtifactsRepository
|
|
from application.storage.db.repositories.conversations import ConversationsRepository
|
|
from application.storage.db.repositories.notes import NotesRepository
|
|
from application.storage.db.repositories.user_tools import UserToolsRepository
|
|
|
|
|
|
@pytest.fixture
|
|
def _patch_db(pg_conn, monkeypatch):
|
|
"""Point both the tools and artifacts route DB helpers at the test conn."""
|
|
|
|
@contextmanager
|
|
def _use_conn():
|
|
yield pg_conn
|
|
|
|
monkeypatch.setattr("application.api.user.tools.routes.db_readonly", _use_conn)
|
|
monkeypatch.setattr("application.api.user.artifacts.routes.db_readonly", _use_conn)
|
|
return pg_conn
|
|
|
|
|
|
def _get(flask_app, artifact_id, token):
|
|
from application.api.user.tools.routes import GetArtifact
|
|
|
|
with flask_app.app_context():
|
|
with flask_app.test_request_context():
|
|
request.decoded_token = token
|
|
return GetArtifact().get(artifact_id)
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestGeneralizedToolsArtifact:
|
|
def test_document_branch_returns_metadata(
|
|
self, _patch_db, flask_app, decoded_token
|
|
):
|
|
conv = ConversationsRepository(_patch_db).create(
|
|
decoded_token["sub"], name="conv"
|
|
)
|
|
art = ArtifactsRepository(_patch_db).create_artifact(
|
|
decoded_token["sub"],
|
|
"document",
|
|
conversation_id=str(conv["id"]),
|
|
title="Report",
|
|
filename="report.pdf",
|
|
mime_type="application/pdf",
|
|
storage_path="inputs/u/artifacts/x/v1/report.pdf",
|
|
)
|
|
|
|
resp = _get(flask_app, art["id"], decoded_token)
|
|
assert resp.status_code == 200
|
|
body = resp.json["artifact"]
|
|
assert body["artifact_type"] == "document"
|
|
assert body["data"]["title"] == "Report"
|
|
assert body["data"]["filename"] == "report.pdf"
|
|
assert body["data"]["download_url"] == f"/api/artifacts/{art['id']}/download"
|
|
|
|
def test_file_kind_maps_to_file_type(self, _patch_db, flask_app, decoded_token):
|
|
conv = ConversationsRepository(_patch_db).create(
|
|
decoded_token["sub"], name="conv"
|
|
)
|
|
art = ArtifactsRepository(_patch_db).create_artifact(
|
|
decoded_token["sub"],
|
|
"file",
|
|
conversation_id=str(conv["id"]),
|
|
storage_path="inputs/u/artifacts/y/v1/data.bin",
|
|
)
|
|
resp = _get(flask_app, art["id"], decoded_token)
|
|
assert resp.status_code == 200
|
|
assert resp.json["artifact"]["artifact_type"] == "file"
|
|
|
|
def test_stranger_document_not_found(self, _patch_db, flask_app, decoded_token):
|
|
conv = ConversationsRepository(_patch_db).create("other_owner", name="conv")
|
|
art = ArtifactsRepository(_patch_db).create_artifact(
|
|
"other_owner",
|
|
"document",
|
|
conversation_id=str(conv["id"]),
|
|
storage_path="inputs/o/artifacts/z/v1/x.pdf",
|
|
)
|
|
# Parent-derived authz denies a stranger -> falls through to 404.
|
|
resp = _get(flask_app, art["id"], decoded_token)
|
|
assert resp.status_code == 404
|
|
|
|
def test_note_branch_still_works(self, _patch_db, flask_app, decoded_token):
|
|
tool = UserToolsRepository(_patch_db).create(
|
|
user_id=decoded_token["sub"], name="notes_tool"
|
|
)
|
|
note = NotesRepository(_patch_db).upsert(
|
|
user_id=decoded_token["sub"],
|
|
tool_id=str(tool["id"]),
|
|
title="t",
|
|
content="a\nb",
|
|
)
|
|
resp = _get(flask_app, str(note["id"]), decoded_token)
|
|
assert resp.status_code == 200
|
|
assert resp.json["artifact"]["artifact_type"] == "note"
|
|
|
|
def test_unknown_id_404(self, _patch_db, flask_app, decoded_token):
|
|
resp = _get(flask_app, str(uuid.uuid4()), decoded_token)
|
|
assert resp.status_code == 404
|