修改单元测试方法

sql/engines/tests.py

@@ -1840,16 +1840,24 @@ def test_fill_query_columns(self):
 
     @patch("sql.engines.mongo.MongoEngine.get_connection")
     def test_current_op(self, mock_get_connection):
-        class Aggregate:
-            def __enter__(self):
-                yield {"client": "single_client"}
-                yield {"clientMetadata": {"mongos": {"client": "sharding_client"}}}
-
-            def __exit__(self, *arg, **kwargs):
-                pass
-
         mock_conn = Mock()
-        mock_conn.admin.aggregate.return_value = Aggregate()
+        # 模拟 cursor 返回的 operation 列表
+        mock_cursor = [
+            {
+                "client": "single_client",
+                "effectiveUsers": [{"user": "user_1"}],
+                "clientMetadata": {"mongos": {"client": "sharding_client"}},
+            },
+            {
+                "clientMetadata": {"mongos": {}},
+                "effectiveUsers": [{"user": "user_2"}],
+            },
+            {
+                "effectiveUsers": [],
+            },
+        ]
+
+        mock_conn.admin.aggregate.return_value = mock_cursor
         mock_get_connection.return_value = mock_conn
         command_types = ["Full", "All", "Inner", "Active"]
         for command_type in command_types: