Cheng Zhou
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.document import Document, DocumentScopeType
|
|
from app.models.document_version import DocumentVersion # noqa: F401
|
|
from app.models.distribution import Distribution # noqa: F401
|
|
from app.models.acknowledgement import Acknowledgement # noqa: F401
|
|
from app.models.etmf import EtmfNode
|
|
from app.services.etmf_service import calculate_node_status, list_etmf_tree
|
|
|
|
|
|
def test_required_node_without_documents_is_missing() -> None:
|
|
node = SimpleNamespace(is_active=True, required=True)
|
|
|
|
assert calculate_node_status(node, []) == "MISSING"
|
|
|
|
|
|
def test_node_with_document_without_effective_version_is_uploaded() -> None:
|
|
node = SimpleNamespace(is_active=True, required=True)
|
|
document = SimpleNamespace(current_effective_version_id=None)
|
|
|
|
assert calculate_node_status(node, [document]) == "UPLOADED"
|
|
|
|
|
|
def test_node_with_current_effective_version_is_effective() -> None:
|
|
node = SimpleNamespace(is_active=True, required=True)
|
|
document = SimpleNamespace(current_effective_version_id=uuid.uuid4())
|
|
|
|
assert calculate_node_status(node, [document]) == "EFFECTIVE"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_etmf_tree_returns_hierarchy_and_status(db_session: AsyncSession) -> None:
|
|
study_id = uuid.uuid4()
|
|
parent = EtmfNode(
|
|
id=uuid.uuid4(),
|
|
study_id=study_id,
|
|
code="01",
|
|
name="项目管理",
|
|
scope_type=DocumentScopeType.GLOBAL,
|
|
required=False,
|
|
sort_order=1,
|
|
)
|
|
child = EtmfNode(
|
|
id=uuid.uuid4(),
|
|
study_id=study_id,
|
|
parent_id=parent.id,
|
|
code="01.01",
|
|
name="方案与版本",
|
|
scope_type=DocumentScopeType.GLOBAL,
|
|
required=True,
|
|
sort_order=1,
|
|
)
|
|
db_session.add_all([parent, child])
|
|
await db_session.commit()
|
|
|
|
tree = await list_etmf_tree(db_session, study_id=study_id, current_user=None)
|
|
|
|
assert len(tree) == 1
|
|
assert tree[0].code == "01"
|
|
assert tree[0].children[0].code == "01.01"
|
|
assert tree[0].children[0].status == "MISSING"
|