From a070aeb694118813471809ef22ea38526ef59428 Mon Sep 17 00:00:00 2001
From: neatguycoding <15627489+NeatGuyCoding@users.noreply.github.com>
Date: Sun, 22 Jun 2025 16:47:45 +0800
Subject: [PATCH] feat: refactor: set global mock for Redis client - formatting
 #21324

---
 api/tests/unit_tests/conftest.py                      | 3 +--
 api/tests/unit_tests/services/test_dataset_service.py | 7 +++----
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/api/tests/unit_tests/conftest.py b/api/tests/unit_tests/conftest.py
index 273fb7208a..e94340fd60 100644
--- a/api/tests/unit_tests/conftest.py
+++ b/api/tests/unit_tests/conftest.py
@@ -1,8 +1,8 @@
 import os
+from unittest.mock import MagicMock, patch
 
 import pytest
 from flask import Flask
-from unittest.mock import MagicMock, patch
 
 # Getting the absolute path of the current file's directory
 ABS_PATH = os.path.dirname(os.path.abspath(__file__))
@@ -56,4 +56,3 @@ def reset_redis_mock():
     redis_mock.hgetall.return_value = {}
     redis_mock.hdel.return_value = None
     redis_mock.incr.return_value = 1
-    yield
diff --git a/api/tests/unit_tests/services/test_dataset_service.py b/api/tests/unit_tests/services/test_dataset_service.py
index 5c4a843b95..37fa9ca7b4 100644
--- a/api/tests/unit_tests/services/test_dataset_service.py
+++ b/api/tests/unit_tests/services/test_dataset_service.py
@@ -2,17 +2,16 @@ import datetime
 import unittest
 
 # Mock redis_client before importing dataset_service
-from unittest.mock import MagicMock, Mock, call, patch
+from unittest.mock import Mock, call, patch
 
 import pytest
 
-# 从 conftest.py 导入全局的 redis_mock
-from tests.unit_tests.conftest import redis_mock
-
 from models.dataset import Dataset, Document
 from services.dataset_service import DocumentService
 from services.errors.document import DocumentIndexingError
 
+from tests.unit_tests.conftest import redis_mock
+
 
 class TestDatasetServiceBatchUpdateDocumentStatus(unittest.TestCase):
     """