feat: expand module workflows, dynamic config, notifications and console API

tests/modules/content_marketing/test_relogin_rollback.py

@@ -15,25 +15,31 @@ from gyxx_flow.modules.content_marketing.data.tools import (
 @pytest.mark.parametrize("module", [relogin_pgy, relogin_xingtu])
 def test_failed_relogin_restores_cookie_and_profile(tmp_path, monkeypatch, module):
     cookie = tmp_path / "cookies.json"
+    storage = tmp_path / "storage_state.json"
     profile = tmp_path / "profile"
     cookie.write_text("old-cookie", encoding="utf-8")
+    storage.write_text("old-storage", encoding="utf-8")
     profile.mkdir()
     (profile / "state.txt").write_text("old-profile", encoding="utf-8")
 
     monkeypatch.setattr(module, "COOKIE_FILE", cookie)
+    monkeypatch.setattr(module, "STATE_FILE", storage)
     monkeypatch.setattr(module, "PROFILE_DIR", profile)
 
-    cookie_backup, profile_backup = module.reset_state()
+    cookie_backup, profile_backup, state_backup = module.reset_state()
     assert not cookie.exists()
+    assert not storage.exists()
     assert not profile.exists()
 
     cookie.write_text("partial-new-cookie", encoding="utf-8")
+    storage.write_text("partial-new-storage", encoding="utf-8")
     profile.mkdir()
     (profile / "partial.txt").write_text("partial", encoding="utf-8")
 
-    module.restore_previous_state(cookie_backup, profile_backup)
+    module.restore_previous_state(cookie_backup, profile_backup, state_backup)
 
     assert cookie.read_text(encoding="utf-8") == "old-cookie"
+    assert storage.read_text(encoding="utf-8") == "old-storage"
     assert (profile / "state.txt").read_text(encoding="utf-8") == "old-profile"
     assert not (profile / "partial.txt").exists()
 
@@ -41,21 +47,27 @@ def test_failed_relogin_restores_cookie_and_profile(tmp_path, monkeypatch, modul
 @pytest.mark.parametrize("module", [relogin_pgy, relogin_xingtu])
 def test_successful_relogin_keeps_new_cookie(tmp_path, monkeypatch, module):
     cookie = tmp_path / "cookies.json"
+    storage = tmp_path / "storage_state.json"
     profile = tmp_path / "profile"
     cookie.write_text("old-cookie", encoding="utf-8")
+    storage.write_text("old-storage", encoding="utf-8")
     profile.mkdir()
 
     monkeypatch.setattr(module, "COOKIE_FILE", cookie)
+    monkeypatch.setattr(module, "STATE_FILE", storage)
     monkeypatch.setattr(module, "PROFILE_DIR", profile)
 
-    cookie_backup, profile_backup = module.reset_state()
+    cookie_backup, profile_backup, state_backup = module.reset_state()
     cookie.write_text("new-cookie", encoding="utf-8")
+    storage.write_text("new-storage", encoding="utf-8")
     profile.mkdir()
 
     module.finish_successful_relogin(profile_backup)
 
     assert cookie.read_text(encoding="utf-8") == "new-cookie"
+    assert storage.read_text(encoding="utf-8") == "new-storage"
     assert cookie_backup.exists()
+    assert state_backup.exists()
     assert not profile_backup.exists()
 
 
@@ -127,6 +139,82 @@ def test_xingtu_index_page_is_recognized_as_logged_in():
     assert xingtu_scraper_v2.is_logged_in(Page()) is True
 
 
+def test_xingtu_index_page_with_xingtu_session_is_logged_in_without_market_ui():
+    class Locator:
+        @property
+        def first(self):
+            return self
+
+        def count(self):
+            return 0
+
+        def wait_for(self, timeout):
+            raise xingtu_scraper_v2.BrowserTimeoutError("not on market page")
+
+        def is_visible(self):
+            return False
+
+    class Context:
+        def cookies(self):
+            return [
+                {
+                    "name": "sessionid",
+                    "value": "fresh",
+                    "domain": ".xingtu.cn",
+                }
+            ]
+
+    class Page:
+        url = "https://www.xingtu.cn/ad/creator/index"
+        context = Context()
+
+        def get_by_text(self, *_args, **_kwargs):
+            return Locator()
+
+        def locator(self, *_args, **_kwargs):
+            return Locator()
+
+    assert xingtu_scraper_v2.is_logged_in(Page()) is True
+
+
+def test_xingtu_index_page_without_xingtu_session_is_not_logged_in():
+    class Locator:
+        @property
+        def first(self):
+            return self
+
+        def count(self):
+            return 0
+
+        def wait_for(self, timeout):
+            raise xingtu_scraper_v2.BrowserTimeoutError("missing")
+
+        def is_visible(self):
+            return False
+
+    class Context:
+        def cookies(self):
+            return [
+                {
+                    "name": "sessionid",
+                    "value": "douyin-only",
+                    "domain": ".douyin.com",
+                }
+            ]
+
+    class Page:
+        url = "https://www.xingtu.cn/ad/creator/index"
+        context = Context()
+
+        def get_by_text(self, *_args, **_kwargs):
+            return Locator()
+
+        def locator(self, *_args, **_kwargs):
+            return Locator()
+
+    assert xingtu_scraper_v2.is_logged_in(Page()) is False
+
+
 def test_xingtu_public_homepage_is_not_logged_in_even_with_stale_cookie():
     class Context:
         def cookies(self):
@@ -149,7 +237,7 @@ def test_xingtu_market_page_without_business_ui_rejects_stale_cookie():
             return self
 
         def wait_for(self, **_kwargs):
-            raise xingtu_scraper_v2.PlaywrightTimeoutError("missing")
+            raise xingtu_scraper_v2.BrowserTimeoutError("missing")
 
         def count(self):
             return 0
@@ -282,25 +370,64 @@ def test_xingtu_scan_login_follows_replacement_page_after_oauth_tab_closes(monke
     assert saved == [market_page]
 
 
+def test_xingtu_scan_login_detects_business_tab_while_qr_tab_stays_open(monkeypatch):
+    class Context:
+        def __init__(self):
+            self.pages = []
+
+    class Page:
+        def __init__(self, context, url):
+            self.context = context
+            self.url = url
+
+        def is_closed(self):
+            return False
+
+        def wait_for_timeout(self, _milliseconds):
+            raise AssertionError("authenticated tab should be detected immediately")
+
+    context = Context()
+    qr_page = Page(context, "https://open.douyin.com/platform/oauth/pc/auth")
+    index_page = Page(context, "https://www.xingtu.cn/ad/creator/index")
+    context.pages = [qr_page, index_page]
+    saved = []
+
+    monkeypatch.setattr(xingtu_scraper_v2, "open_scan_login", lambda _page: None)
+    monkeypatch.setattr(
+        xingtu_scraper_v2, "is_logged_in", lambda page: page is index_page
+    )
+    monkeypatch.setattr(xingtu_scraper_v2, "save_state", lambda page: saved.append(page))
+
+    result = xingtu_scraper_v2.wait_for_scan_login(qr_page, 2)
+
+    assert result is index_page
+    assert saved == [index_page]
+
+
 def test_xingtu_interrupted_transaction_is_recovered(tmp_path, monkeypatch):
     cookie = tmp_path / "xingtu_cookies.json"
+    storage = tmp_path / "xingtu_storage_state.json"
     profile = tmp_path / "profile"
     cookie.write_text("old-cookie", encoding="utf-8")
+    storage.write_text("old-storage", encoding="utf-8")
     profile.mkdir()
     (profile / "old.txt").write_text("old-profile", encoding="utf-8")
 
     monkeypatch.setattr(relogin_xingtu, "COOKIE_FILE", cookie)
+    monkeypatch.setattr(relogin_xingtu, "STATE_FILE", storage)
     monkeypatch.setattr(relogin_xingtu, "PROFILE_DIR", profile)
 
-    cookie_backup, profile_backup = relogin_xingtu.reset_state()
+    cookie_backup, profile_backup, state_backup = relogin_xingtu.reset_state()
     profile.mkdir()
     (profile / "partial.txt").write_text("partial", encoding="utf-8")
 
     assert relogin_xingtu.recover_interrupted_state() is True
     assert cookie.read_text(encoding="utf-8") == "old-cookie"
+    assert storage.read_text(encoding="utf-8") == "old-storage"
     assert (profile / "old.txt").read_text(encoding="utf-8") == "old-profile"
     assert not (profile / "partial.txt").exists()
     assert cookie_backup is not None
+    assert state_backup is not None
     assert profile_backup is not None