TEST: Added test cases for Upload Documents HTTP API (#5991)

### What problem does this PR solve?

cover upload docments endpoints

### Type of change

- [x] add test cases

sdk/python/test/test_http_api/test_dataset_mangement/common.py

@@ -1,57 +0,0 @@
-#
-#  Copyright 2025 The InfiniFlow Authors. All Rights Reserved.
-#
-#  Licensed under the Apache License, Version 2.0 (the "License");
-#  you may not use this file except in compliance with the License.
-#  You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-#  Unless required by applicable law or agreed to in writing, software
-#  distributed under the License is distributed on an "AS IS" BASIS,
-#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#  See the License for the specific language governing permissions and
-#  limitations under the License.
-#
-
-import os
-
-import requests
-
-HOST_ADDRESS = os.getenv("HOST_ADDRESS", "http://127.0.0.1:9380")
-API_URL = f"{HOST_ADDRESS}/api/v1/datasets"
-HEADERS = {"Content-Type": "application/json"}
-
-
-INVALID_API_TOKEN = "invalid_key_123"
-DATASET_NAME_LIMIT = 128
-
-
-def create_dataset(auth, payload):
-    res = requests.post(url=API_URL, headers=HEADERS, auth=auth, json=payload)
-    return res.json()
-
-
-def list_dataset(auth, params=None):
-    res = requests.get(url=API_URL, headers=HEADERS, auth=auth, params=params)
-    return res.json()
-
-
-def update_dataset(auth, dataset_id, payload):
-    res = requests.put(
-        url=f"{API_URL}/{dataset_id}", headers=HEADERS, auth=auth, json=payload 
-    )
-    return res.json()
-
-
-def delete_dataset(auth, payload=None):
-    res = requests.delete(url=API_URL, headers=HEADERS, auth=auth, json=payload)
-    return res.json()
-
-
-def create_datasets(auth, num):
-    ids = []
-    for i in range(num):
-        res = create_dataset(auth, {"name": f"dataset_{i}"})
-        ids.append(res["data"]["id"])
-    return ids

sdk/python/test/test_http_api/test_dataset_mangement/conftest.py

@@ -1,24 +0,0 @@
-#
-#  Copyright 2025 The InfiniFlow Authors. All Rights Reserved.
-#
-#  Licensed under the Apache License, Version 2.0 (the "License");
-#  you may not use this file except in compliance with the License.
-#  You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-#  Unless required by applicable law or agreed to in writing, software
-#  distributed under the License is distributed on an "AS IS" BASIS,
-#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#  See the License for the specific language governing permissions and
-#  limitations under the License.
-#
-
-import pytest
-from common import delete_dataset
-
-
-@pytest.fixture(scope="function", autouse=True)
-def clear_datasets(get_http_api_auth):
-    yield
-    delete_dataset(get_http_api_auth)

sdk/python/test/test_http_api/test_dataset_mangement/test_create_dataset.py

@@ -13,12 +13,12 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 #
-import base64
-from pathlib import Path
 
 import pytest
 from common import DATASET_NAME_LIMIT, INVALID_API_TOKEN, create_dataset
 from libs.auth import RAGFlowHttpApiAuth
+from libs.utils import encode_avatar
+from libs.utils.file_utils import create_image_file
 
 
 class TestAuthorization:
@@ -75,18 +75,11 @@ class TestDatasetCreation:
 
 
 class TestAdvancedConfigurations:
-    def test_avatar(self, get_http_api_auth, request):
-        def encode_avatar(image_path):
-            with Path.open(image_path, "rb") as file:
-                binary_data = file.read()
-            base64_encoded = base64.b64encode(binary_data).decode("utf-8")
-            return base64_encoded
-
+    def test_avatar(self, get_http_api_auth, tmp_path):
+        fn = create_image_file(tmp_path / "ragflow_test.png")
         payload = {
             "name": "avatar_test",
-            "avatar": encode_avatar(
-                Path(request.config.rootdir) / "test/data/logo.svg"
-            ),
+            "avatar": encode_avatar(fn),
         }
         res = create_dataset(get_http_api_auth, payload)
         assert res["code"] == 0

sdk/python/test/test_http_api/test_dataset_mangement/test_update_dataset.py

@@ -13,9 +13,7 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 #
-import base64
 from concurrent.futures import ThreadPoolExecutor
-from pathlib import Path
 
 import pytest
 from common import (
@@ -26,6 +24,8 @@ from common import (
     update_dataset,
 )
 from libs.auth import RAGFlowHttpApiAuth
+from libs.utils import encode_avatar
+from libs.utils.file_utils import create_image_file
 
 # TODO: Missing scenario for updating embedding_model with chunk_count != 0
 
@@ -171,19 +171,10 @@ class TestDatasetUpdate:
         else:
             assert res["message"] == expected_message
 
-    def test_avatar(self, get_http_api_auth, request):
-        def encode_avatar(image_path):
-            with Path.open(image_path, "rb") as file:
-                binary_data = file.read()
-            base64_encoded = base64.b64encode(binary_data).decode("utf-8")
-            return base64_encoded
-
+    def test_avatar(self, get_http_api_auth, tmp_path):
         ids = create_datasets(get_http_api_auth, 1)
-        payload = {
-            "avatar": encode_avatar(
-                Path(request.config.rootdir) / "test/data/logo.svg"
-            ),
-        }
+        fn = create_image_file(tmp_path / "ragflow_test.png")
+        payload = {"avatar": encode_avatar(fn)}
         res = update_dataset(get_http_api_auth, ids[0], payload)
         assert res["code"] == 0