656f596d7e
Core modules: - Laws: CRUD, search, AI-powered QA - Analysis: legal research and case management - Contracts: lifecycle management with templates - Signatures: electronic signature workflow Infrastructure: - FastAPI + SQLite + async SQLAlchemy - Docker deployment support - 54 unit tests passing Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
"""Unit tests for Contract service."""
|
|
import pytest
|
|
from datetime import date
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.contract import ContractStatus
|
|
from app.services.contract_service import ContractService, ContractTemplateService
|
|
|
|
|
|
class TestContractService:
|
|
"""Test cases for Contract service."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_contract(self, db_session: AsyncSession):
|
|
"""Test creating a contract."""
|
|
service = ContractService(db_session)
|
|
|
|
contract = await service.create_contract(
|
|
title="测试合同",
|
|
party_a="甲方",
|
|
party_b="乙方",
|
|
content="合同内容",
|
|
created_by=1,
|
|
)
|
|
|
|
assert contract.id is not None
|
|
assert contract.title == "测试合同"
|
|
assert contract.status == ContractStatus.DRAFT
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_contract_by_id(self, db_session: AsyncSession):
|
|
"""Test getting a contract by ID."""
|
|
service = ContractService(db_session)
|
|
|
|
created = await service.create_contract(
|
|
title="测试合同",
|
|
party_a="甲方",
|
|
party_b="乙方",
|
|
content="合同内容",
|
|
created_by=1,
|
|
)
|
|
|
|
contract = await service.get_contract_by_id(created.id)
|
|
|
|
assert contract is not None
|
|
assert contract.title == "测试合同"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_submit_for_approval(self, db_session: AsyncSession):
|
|
"""Test submitting a contract for approval."""
|
|
service = ContractService(db_session)
|
|
|
|
contract = await service.create_contract(
|
|
title="测试合同",
|
|
party_a="甲方",
|
|
party_b="乙方",
|
|
content="合同内容",
|
|
created_by=1,
|
|
)
|
|
|
|
updated = await service.submit_for_approval(contract.id)
|
|
|
|
assert updated.status == ContractStatus.PENDING_APPROVAL
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_approve_contract(self, db_session: AsyncSession):
|
|
"""Test approving a contract."""
|
|
service = ContractService(db_session)
|
|
|
|
contract = await service.create_contract(
|
|
title="测试合同",
|
|
party_a="甲方",
|
|
party_b="乙方",
|
|
content="合同内容",
|
|
created_by=1,
|
|
)
|
|
|
|
approval = await service.approve_contract(contract.id, 1, "同意")
|
|
|
|
assert approval.status == ContractStatus.APPROVED
|
|
|
|
|
|
class TestContractTemplateService:
|
|
"""Test cases for ContractTemplate service."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_template(self, db_session: AsyncSession):
|
|
"""Test creating a contract template."""
|
|
service = ContractTemplateService(db_session)
|
|
|
|
template = await service.create_template(
|
|
name="测试模板",
|
|
content="模板内容",
|
|
created_by=1,
|
|
contract_type="销售合同",
|
|
)
|
|
|
|
assert template.id is not None
|
|
assert template.name == "测试模板"
|