feat: refactor: set global mock for Redis client - formatting #21324

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

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):
     """