尝试修改单元测试

hhyo · Sep 20, 2024 · 27e44a0 · 27e44a0
1 parent c832bfe
commit 27e44a0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sql/test_archiver.py b/sql/test_archiver.py
@@ -273,12 +273,12 @@ def test_archive_audit(self, _async_task, mock_operate):
         assert self.archive_apply.status == WorkflowStatus.PASSED
 
 
-    @patch('sql.archiver.apply_async.apply_async')
-    def test_add_archive_task(self, apply_async_mock):
+    @patch('sql.archiver.add_archive_task.apply_async')
+    def test_add_archive_task(self, mock_apply_async):
         # 调用 add.apply_async
         add_archive_task.apply_async()
 
-    @patch("sql.archiver.apply_async")
+    @patch("sql.archiver.archive.apply_async")
     def test_add_archive(self, mock_apply_async):
         """
         测试执行归档任务