更新

app/tests/test_issued_prescription_ai.py

@@ -794,6 +794,33 @@ def test_revoked_agreement_cache_discards_flat_comparison_metadata() -> None:
     assert value["models"]["qwen"]["reason"] == sentinel
 
 
+@pytest.mark.parametrize("flag", [False, None, 1, "true"])
+def test_unlisted_account_hides_ai_analysis_column_and_keeps_reports(
+    application: QApplication, immediate: None, monkeypatch: pytest.MonkeyPatch, flag: Any,
+) -> None:
+    repository = Repository()
+    repository.agreement_visible = flag
+    opened = []
+    monkeypatch.setattr(page_module, "present_issued_prescription_ai", lambda *_args, **kwargs: opened.append(kwargs))
+    page = page_module.PrescriptionsPage(repository, ["*"], {"root": 1})
+    try:
+        page.resize(1366, 800)
+        page.show()
+        application.processEvents()
+        page.table.selectRow(0)
+        assert page.table.isColumnHidden(11) and page.table.isColumnHidden(12)
+        # Later status refreshes must not re-expose the analysis column.
+        page._set_ai_columns(True)
+        assert page.table.isColumnHidden(11) and page.table.isColumnHidden(12)
+        assert page.ai_report_button.isVisible() and page.ai_report_button.isEnabled()
+        assert any(button.accessibleName() == "AI 报告"
+                   for button in page.table.cellWidget(0, 2).findChildren(QPushButton))
+        page.ai_report_button.click()
+        assert opened == [{"prescription_id": 801}]
+    finally:
+        page.close()
+
+
 def test_list_capability_failure_revocation_and_late_status_cannot_restore_scores(
     application: QApplication, immediate: None, monkeypatch: pytest.MonkeyPatch,
 ) -> None:
@@ -810,7 +837,7 @@ def test_list_capability_failure_revocation_and_late_status_cannot_restore_score
     page._load_ai_capabilities()
     page._ai_statuses_ready(old_statuses, [801, 802], old_token, page._generation)
     page._ai_capabilities_ready({"agreement_visible": True}, old_capability_token, page._generation)
-    assert page.table.isColumnHidden(12) and not page.table.isColumnHidden(11)
+    assert page.table.isColumnHidden(12) and page.table.isColumnHidden(11)
     assert page.ai_statistics_button.isHidden()
     assert page.ai_report_button.isVisible()
     assert page._ai_statuses