提交

backend/tests/test_pl_plan_contract_gap.py

@@ -1,7 +1,13 @@
 from sqlalchemy.dialects import mysql
 
+from app.schemas.pl.plan import PlanDeliveryRead, PlanLoadRead
 from app.services.pl_internal_customer import INTERNAL_CUSTOMER_NAMES
-from app.services.pl_plan_contract_gap import compute_contract_gap_sides, plan_has_contract_gap_exists
+from app.services.pl_plan_contract_gap import (
+    compute_contract_gap_sides,
+    compute_plan_gap_sides_from_stops,
+    plan_has_contract_gap_exists,
+    stop_contract_gap,
+)
 
 INTERNAL = next(iter(INTERNAL_CUSTOMER_NAMES))
 
@@ -56,6 +62,74 @@ def test_empty_names_treated_as_non_internal_gap_both():
     assert sides == ["supplier", "demander"]
 
 
+def test_stop_contract_gap_internal_exempt():
+    assert stop_contract_gap(INTERNAL, 0) is False
+    assert stop_contract_gap("外部客户", 0) is True
+    assert stop_contract_gap("外部客户", 1) is False
+
+
+def test_compute_plan_gap_sides_from_stops_multi_delivery():
+    load = PlanLoadRead(
+        id="load1",
+        planId="p1",
+        tenantNo="t1",
+        customerName=INTERNAL,
+        products=[],
+        contracts=[],
+        createTime="2026-08-27T00:00:00",
+        updateTime="2026-08-27T00:00:00",
+    )
+    delivery_ok = PlanDeliveryRead(
+        id="d1",
+        planId="p1",
+        tenantNo="t1",
+        seq=1,
+        customerName="外部客户A",
+        products=[],
+        contracts=[{"contractId": "c1", "contractNo": "HT001"}],
+        createTime="2026-08-27T00:00:00",
+        updateTime="2026-08-27T00:00:00",
+    )
+    delivery_gap = PlanDeliveryRead(
+        id="d2",
+        planId="p1",
+        tenantNo="t1",
+        seq=2,
+        customerName="外部客户B",
+        products=[],
+        contracts=[],
+        createTime="2026-08-27T00:00:00",
+        updateTime="2026-08-27T00:00:00",
+    )
+    sides = compute_plan_gap_sides_from_stops(load, [delivery_ok, delivery_gap])
+    assert sides == ["demander"]
+
+
+def test_compute_plan_gap_sides_from_stops_internal_load_no_gap():
+    load = PlanLoadRead(
+        id="load1",
+        planId="p1",
+        tenantNo="t1",
+        customerName=INTERNAL,
+        products=[],
+        contracts=[],
+        createTime="2026-08-27T00:00:00",
+        updateTime="2026-08-27T00:00:00",
+    )
+    delivery = PlanDeliveryRead(
+        id="d1",
+        planId="p1",
+        tenantNo="t1",
+        seq=1,
+        customerName=INTERNAL,
+        products=[],
+        contracts=[],
+        createTime="2026-08-27T00:00:00",
+        updateTime="2026-08-27T00:00:00",
+    )
+    assert compute_plan_gap_sides_from_stops(load, [delivery]) == []
+
+
 def _compile(clause) -> str:
     return str(
         clause.compile(dialect=mysql.dialect(), compile_kwargs={"literal_binds": True})
@@ -67,5 +141,5 @@ def test_plan_has_contract_gap_exists_mentions_contract_and_internal_names():
     sql = _compile(clause)
 
     assert "pl_plan_contract" in sql
-    assert "supplier" in sql
-    assert "demander" in sql
+    assert "pl_plan_load" in sql
+    assert "pl_plan_delivery" in sql