API文档维护Agent
e3fb518ab2
- Remove workspace_id from Pydantic models in project_management routes - Remove workspace_id from SQLAlchemy and SQLite project management repos - Remove workspace_id from worker tasks (storage keys, entity creation) - Remove workspace_id from video dedup and title usage modules - Remove workspace_id from generation and ingest worker tasks - Clean workspace_id from all test files and scripts - Remove workspace-specific test files (list_workspaces, workspace repos) Task: #14 workspace_id 残留清理
109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
from packages.adapters.in_memory import (
|
|
InMemoryAssetLibraryRepository,
|
|
InMemoryAssetRepository,
|
|
InMemoryIngestJobRepository,
|
|
InMemoryProjectRepository,
|
|
)
|
|
from packages.application import (
|
|
CreateAssetCommand,
|
|
CreateAssetLibraryCommand,
|
|
CreateAssetLibraryUseCase,
|
|
CreateAssetUseCase,
|
|
CreateProjectCommand,
|
|
CreateProjectUseCase,
|
|
GetProjectUseCase,
|
|
ListAssetLibrariesUseCase,
|
|
ListAssetsUseCase,
|
|
ListProjectsUseCase,
|
|
SubmitIngestJobCommand,
|
|
SubmitIngestJobUseCase,
|
|
)
|
|
from packages.domain import AssetLibraryKind, IngestJobStatus
|
|
|
|
|
|
def test_create_and_list_projects():
|
|
repository = InMemoryProjectRepository()
|
|
create_use_case = CreateProjectUseCase(repository)
|
|
list_use_case = ListProjectsUseCase(repository)
|
|
|
|
assert project.name == "Demo Project"
|
|
|
|
items = list_use_case.execute("ws-1")
|
|
assert len(items) == 1
|
|
assert items[0].id == project.id
|
|
|
|
|
|
def test_get_project_by_id_restores_workspace_context():
|
|
repository = InMemoryProjectRepository()
|
|
create_use_case = CreateProjectUseCase(repository)
|
|
get_use_case = GetProjectUseCase(repository)
|
|
|
|
|
|
retrieved = get_use_case.execute(project.id)
|
|
assert retrieved is not None
|
|
assert retrieved.id == project.id
|
|
|
|
|
|
def test_create_and_list_asset_libraries():
|
|
repository = InMemoryAssetLibraryRepository()
|
|
create_use_case = CreateAssetLibraryUseCase(repository)
|
|
list_use_case = ListAssetLibrariesUseCase(repository)
|
|
|
|
library = create_use_case.execute(
|
|
CreateAssetLibraryCommand(
|
|
project_id="proj-1",
|
|
name=" 素材库 A ",
|
|
kind=AssetLibraryKind.VIDEO,
|
|
)
|
|
)
|
|
assert library.name == "素材库 A"
|
|
assert library.kind == AssetLibraryKind.VIDEO
|
|
|
|
items = list_use_case.execute("proj-1", kind=AssetLibraryKind.VIDEO)
|
|
assert len(items) == 1
|
|
assert items[0].id == library.id
|
|
|
|
|
|
def test_create_and_list_assets():
|
|
repository = InMemoryAssetRepository()
|
|
create_use_case = CreateAssetUseCase(repository)
|
|
list_use_case = ListAssetsUseCase(repository)
|
|
|
|
asset = create_use_case.execute(
|
|
CreateAssetCommand(
|
|
project_id="proj-1",
|
|
library_id="lib-1",
|
|
name=" demo.mp4 ",
|
|
storage_key="uploads/demo.mp4",
|
|
mime_type="video/mp4",
|
|
metadata={"duration": 12.5},
|
|
)
|
|
)
|
|
assert asset.name == "demo.mp4"
|
|
assert asset.storage_key == "uploads/demo.mp4"
|
|
assert asset.metadata["duration"] == 12.5
|
|
|
|
items = list_use_case.execute("lib-1")
|
|
assert len(items) == 1
|
|
assert items[0].id == asset.id
|
|
|
|
|
|
def test_submit_ingest_job():
|
|
repository = InMemoryIngestJobRepository()
|
|
use_case = SubmitIngestJobUseCase(repository)
|
|
|
|
job = use_case.execute(
|
|
SubmitIngestJobCommand(
|
|
project_id="proj-1",
|
|
library_id="lib-1",
|
|
storage_key="uploads/raw-video.mp4",
|
|
)
|
|
)
|
|
assert job.storage_key == "uploads/raw-video.mp4"
|
|
assert job.status == IngestJobStatus.PENDING
|
|
assert job.result_asset_id == ""
|
|
|
|
retrieved = repository.get(job.id)
|
|
assert retrieved is not None
|
|
assert retrieved.id == job.id
|