From dcde5a5844c5657384574081016564d519a43f42 Mon Sep 17 00:00:00 2001
From: huashen <2494946808@qq.com>
Date: Tue, 1 Sep 2026 10:00:17 +0800
Subject: [PATCH] test: read channel generation from v3 host

---
 tests/test_manager_integration.py | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/tests/test_manager_integration.py b/tests/test_manager_integration.py
index 38d04a2..fc0bf5d 100644
--- a/tests/test_manager_integration.py
+++ b/tests/test_manager_integration.py
@@ -15,6 +15,15 @@
 ROOT = Path(__file__).parents[1]
 
 
+def _current_channel_generation(
+    manager: PluginManager,
+) -> channel_generation_host.ChannelGeneration | None:
+    snapshot = manager.current_snapshot
+    if snapshot is None:
+        return None
+    return manager.channel_generation_host.get(snapshot.snapshot_id)
+
+
 class FakeProviderClient:
     def __init__(self) -> None:
         self.closed = 0
@@ -110,7 +119,7 @@ def wrapped(context):
 
     await manager.load_all()
     stable = manager.current_snapshot
-    runtime = manager.active_channel_generation
+    runtime = _current_channel_generation(manager)
     assert stable is not None and stable.state == "committed"
     assert runtime is not None and runtime.channel("feishu").admission_open
     assert factory.create_calls == 1
@@ -139,11 +148,11 @@ def wrapped(context):
     assert publication["publication_state"] == "committed"
     assert manager.current_snapshot is not stable
     assert factory.create_calls == 2
-    assert manager.active_channel_generation is not None
-    assert manager.active_channel_generation.channel("feishu").admission_open
+    runtime = _current_channel_generation(manager)
+    assert runtime is not None and runtime.channel("feishu").admission_open
 
     await manager.terminate_all()
-    assert manager.active_channel_generation is None
+    assert _current_channel_generation(manager) is None
     assert factory.close_calls == 2
     assert factory.client.closed == 2
 
@@ -211,4 +220,4 @@ def wrapped(context):
     assert adapter._ws_thread is None
     assert not adapter._ws_thread_started
     assert not adapter._started
-    assert manager.active_channel_generation is None
+    assert _current_channel_generation(manager) is None