[migrate]:[FL-27][补齐杆塔专业参数编辑与导出字段]

Co-authored-by: multica-agent <github@multica.ai>

api/tests/test_tower_profile_migration.py

@@ -1,18 +1,112 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+
+from sqlalchemy import create_engine
+from sqlalchemy.orm import Session, sessionmaker
+
+from app.core.database import Base
+from app.models.line import Line
+from app.models.line_tower import LineTower
+from app.models.tower_profile import TowerProfile
 from app.schemas.tower_profile import TowerProfileUpsertRequest
+from app.services import tower_profile_service
+
+
+def _build_session() -> Session:
+    engine = create_engine("sqlite+pysqlite:///:memory:")
+    Base.metadata.create_all(bind=engine, tables=[Line.__table__, LineTower.__table__, TowerProfile.__table__])
+    testing_session = sessionmaker(bind=engine, autocommit=False, autoflush=False, expire_on_commit=False)
+    return testing_session()
 
 
 def test_tower_profile_upsert_request_accepts_new_professional_fields() -> None:
     payload = TowerProfileUpsertRequest(
         structure_kind="直线杆塔",
         stroke_mode="反击",
+        protection_angle_left_deg=11.5,
+        protection_angle_right_deg=13.5,
+        shield_wire_height_m=41.0,
+        insulator_length_m=4200.0,
+        call_height_m=39.5,
+        angle_deg=18.0,
+        current_a=31.0,
+        current_b=2.6,
+        current_type="Heidler",
+        current_head_time_us=2.6,
+        current_tail_time_us=50.0,
         geometry_layers_json={
             "I": {
                 "phase_spacing_m": {"upper": 5.1, "middle": 4.2, "lower": 3.3},
                 "phase_height_m": {"upper": 25.0, "middle": 22.0, "lower": 19.0},
             }
         },
+        extra_profile_json={"cause_analysis": "接地电阻偏高"},
     )
 
     assert payload.structure_kind == "直线杆塔"
     assert payload.stroke_mode == "反击"
-    assert payload.geometry_layers_json["I"]["phase_spacing_m"]["upper"] == 5.1
+    assert payload.protection_angle_left_deg == 11.5
+    assert payload.current_type == "Heidler"
+    assert payload.geometry_layers_json["I"]["phase_spacing_m"]["upper"] == 5.1
+    assert payload.extra_profile_json["cause_analysis"] == "接地电阻偏高"
+
+
+def test_upsert_tower_profile_persists_professional_fields() -> None:
+    session = _build_session()
+    try:
+        line = Line(code="PL-TP-001", name="塔参线路", status="enabled")
+        session.add(line)
+        session.flush()
+
+        tower = LineTower(line_id=line.id, seq_no=1, tower_no="N-01")
+        session.add(tower)
+        session.commit()
+
+        payload = TowerProfileUpsertRequest(
+            structure_kind="耐张杆塔",
+            stroke_mode="反击",
+            phase_sequence_1="ABC",
+            arrester_a="是",
+            protection_angle_left_deg=12.3,
+            protection_angle_right_deg=14.6,
+            shield_wire_height_m=40.2,
+            insulator_length_m=4100.0,
+            call_height_m=38.4,
+            angle_deg=16.5,
+            current_a=29.1,
+            current_b=2.4,
+            current_type="Heidler",
+            current_head_time_us=2.7,
+            current_tail_time_us=48.0,
+            geometry_layers_json={
+                "I": {
+                    "phase_spacing_m": {"upper": 5.1, "middle": 4.0, "lower": 3.0},
+                    "phase_height_m": {"upper": 29.0, "middle": 27.0, "lower": 25.0},
+                }
+            },
+            extra_profile_json={"cause_analysis": "接地电阻偏高"},
+        )
+
+        detail = tower_profile_service.upsert_tower_profile(
+            session,
+            tower.id,
+            payload,
+            actor=SimpleNamespace(id="tester"),
+        )
+
+        saved = tower_profile_service.get_tower_profile_by_tower_id(session, tower.id)
+
+        assert detail is not None
+        assert saved is not None
+        assert saved.structure_kind == "耐张杆塔"
+        assert saved.stroke_mode == "反击"
+        assert saved.protection_angle_left_deg == 12.3
+        assert saved.shield_wire_height_m == 40.2
+        assert saved.current_type == "Heidler"
+        assert saved.current_head_time_us == 2.7
+        assert saved.extra_profile_json["cause_analysis"] == "接地电阻偏高"
+        assert detail.geometry_layers_json["I"]["phase_spacing_m"]["upper"] == 5.1
+        assert detail.extra_profile_json["cause_analysis"] == "接地电阻偏高"
+    finally:
+        session.close()