shanxia2.0/backend/tests/test_pl_plan_resolve_escort.py

"""运输计划解析押运员：车辆未关联押运时允许为空。"""
from __future__ import annotations

from unittest.mock import MagicMock

from app.services.pl_plan_service import _resolve_escort


def test_resolve_escort_without_vehicle_link_returns_none():
    db = MagicMock()
    db.query.return_value.filter.return_value.order_by.return_value.first.return_value = None

    result = _resolve_escort(db, "tenant-1", "vehicle-1", None)

    assert result == (None, None, None)


def test_resolve_escort_with_explicit_id_returns_personnel():
    db = MagicMock()
    personnel = MagicMock()
    personnel.id = "escort-1"
    personnel.personnel_name = "王押"
    personnel.phone_no = "13900000001"
    db.query.return_value.filter.return_value.first.return_value = personnel

    result = _resolve_escort(db, "tenant-1", "vehicle-1", "escort-1")

    assert result == ("escort-1", "王押", "13900000001")