尝试修改单元测试

sql/test_archiver.py

@@ -16,9 +16,10 @@
  ArchiveConfig,
  WorkflowAudit,
  WorkflowAuditSetting,
+ WorkflowAuditDetail,
 )
 from sql.tests import User
-
+from datetime import timedelta, datetime
 
 class TestArchiver(TestCase):
  """
@@ -79,6 +80,13 @@ def setUp(self):
  create_user="",
  create_user_display="",
  )
+ self.audit_detail = WorkflowAuditDetail.objects.create(
+ audit_id=self.audit_flow.id,
+ audit_user="",
+ audit_time= self.now - timedelta(days=1),
+ audit_status=WorkflowStatus.PASSED,
+ remark="",
+ )
  self.sys_config = SysConfig()
  self.client = Client()
 
@@ -248,17 +256,17 @@ def test_archive_apply_auto_pass(self, mock_generate_setting):
  assert archive_config.status == WorkflowStatus.PASSED
 
  @patch("sql.utils.workflow_audit.AuditV2.operate")
- @patch("sql.archiver.async_task")
- def test_archive_audit(self, _async_task, mock_operate):
+ @patch("sql.notify.notify_for_audit.apply_async")
+ def test_archive_audit(self, mock_apply_async, mock_operate):
  """
  测试审核归档实例数据
  :return:
  """
  mock_operate.return_value = None
  data = {
  "archive_id": self.archive_apply.id,
- "audit_status": WorkflowStatus.PASSED,
- "audit_remark": "xxxx",
+ "audit_status": self.audit_detail.audit_status,
+ "audit_remark": self.audit_detail.remark,
  }
  # operate 被 patch 了, 这里强制设置一下, 走一下流程
  self.audit_flow.current_status = WorkflowStatus.PASSED
@@ -272,16 +280,13 @@ def test_archive_audit(self, _async_task, mock_operate):
  assert self.archive_apply.state == True
  assert self.archive_apply.status == WorkflowStatus.PASSED
 
- @patch("sql.archiver.async_task")
- def test_add_archive_task(self, _async_task):
- """
- 测试添加异步归档任务
- :return:
- """
- add_archive_task()
+ @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.async_task")
- def test_add_archive(self, _async_task):
+ @patch("sql.archiver.archive.apply_async")
+ def test_add_archive(self, mock_apply_async):
  """
  测试执行归档任务
  :return: