fix: 正确的 mock 函数

nonebot · Jan 29, 2024 · dfbe745 · dfbe745
1 parent e27502a
commit dfbe745
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/plugins/publish/__init__.py b/src/plugins/publish/__init__.py
@@ -120,10 +120,10 @@ async def handle_pr_close(
         else:
             logger.info("发布的拉取请求未合并，已跳过")
 
-        # GitHub 的缓存一般 1 分钟左右会刷新
-        await asyncio.sleep(60)
         # 如果商店更新则触发 registry 更新
         if event.payload.pull_request.merged:
+            # GitHub 的缓存一般 1 分钟左右会刷新
+            await asyncio.sleep(60)
             await trigger_registry_update(bot, repo_info, publish_type, issue)
         else:
             logger.info("拉取请求未合并，跳过触发商店列表更新")

diff --git a/tests/publish/process/test_pull_request.py b/tests/publish/process/test_pull_request.py
@@ -18,6 +18,7 @@ async def test_process_pull_request(app: App, mocker: MockerFixture) -> None:
 
     mock_subprocess_run = mocker.patch("subprocess.run")
     mock_sleep = mocker.patch("asyncio.sleep")
+    mock_sleep.return_value = None
 
     mock_issue = mocker.MagicMock()
     mock_issue.state = "open"
@@ -203,6 +204,7 @@ async def test_process_pull_request_skip_plugin_test(
 
     mock_subprocess_run = mocker.patch("subprocess.run")
     mock_sleep = mocker.patch("asyncio.sleep")
+    mock_sleep.return_value = None
 
     mock_issue = mocker.MagicMock()
     mock_issue.state = "open"