初次提交

backend/simple_upload_test.py

@@ -0,0 +1,112 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+简单上传下载测试
+"""
+
+import requests
+import sys
+import io
+
+# 设置stdout编码为utf-8
+sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8")
+
+BASE_URL = "http://localhost:8000/api/v1"
+
+def simple_test():
+    print("=== 简单上传下载测试 ===")
+
+    # 使用存在的用户ID (用户ID=3存在)
+    user_id = 3
+    print(f"使用用户ID: {user_id}")
+
+    # 直接测试下载现有文件 (文件ID=24, test.txt)
+    file_id = 24
+    print(f"\n直接下载现有文件 ID: {file_id} (test.txt)...")
+
+    download_data = {
+        "file_id": file_id,
+        "user_id": user_id
+    }
+
+    try:
+        download_resp = requests.post(f"{BASE_URL}/files/download", json=download_data)
+        print(f"下载状态码: {download_resp.status_code}")
+        print(f"下载响应头: {dict(download_resp.headers)}")
+
+        if download_resp.status_code == 200:
+            downloaded_content = download_resp.text
+            print(f"[OK] 下载成功")
+            print(f"下载内容: {downloaded_content}")
+            print(f"内容长度: {len(downloaded_content)}")
+
+            # 读取原始文件内容进行对比
+            try:
+                with open(f"uploads/608408b0-02c7-449e-ba58-be43c4360333.txt", 'r', encoding='utf-8') as f:
+                    original_content = f.read()
+                print(f"原始文件长度: {len(original_content)}")
+
+                if original_content == downloaded_content:
+                    print("[OK] 内容一致！上传下载功能正常")
+                else:
+                    print("[ERROR] 内容不一致")
+                    print(f"原始: {repr(original_content[:100])}")
+                    print(f"下载: {repr(downloaded_content[:100])}")
+            except Exception as e:
+                print(f"无法读取原始文件进行对比: {e}")
+
+        else:
+            print(f"[ERROR] 下载失败: {download_resp.text}")
+    except Exception as e:
+        print(f"[ERROR] 下载异常: {e}")
+        import traceback
+        traceback.print_exc()
+
+    # 现在测试上传新文件
+    print(f"\n=== 测试上传新文件 ===")
+    test_content = f"这是全新的测试文件内容\nBrand new test file content\n唯一时间戳: {hash('unique_content_' + str(__import__('time').time()))}"
+
+    files = {"file": (f"unique_test_{hash(test_content) % 10000}.txt", test_content, "text/plain")}
+    data = {
+        "user_id": str(user_id),
+        "description": "新测试文件",
+        "tags": "test,new",
+        "is_public": "false"
+    }
+
+    try:
+        upload_resp = requests.post(f"{BASE_URL}/files/upload", files=files, data=data)
+        print(f"上传状态码: {upload_resp.status_code}")
+
+        if upload_resp.status_code == 201:
+            new_file_id = upload_resp.json()["data"]["file"]["id"]
+            print(f"[OK] 新文件上传成功，文件ID: {new_file_id}")
+
+            # 立即下载新文件
+            print(f"\n下载新上传的文件 ID: {new_file_id}...")
+            download_data = {
+                "file_id": new_file_id,
+                "user_id": user_id
+            }
+
+            download_resp = requests.post(f"{BASE_URL}/files/download", json=download_data)
+            if download_resp.status_code == 200:
+                new_downloaded_content = download_resp.text
+                print(f"[OK] 新文件下载成功")
+                print(f"下载内容: {new_downloaded_content}")
+
+                if test_content == new_downloaded_content:
+                    print("[OK] 新文件内容一致！")
+                else:
+                    print("[ERROR] 新文件内容不一致")
+            else:
+                print(f"[ERROR] 新文件下载失败: {download_resp.text}")
+        else:
+            print(f"[ERROR] 新文件上传失败: {upload_resp.text}")
+    except Exception as e:
+        print(f"[ERROR] 新文件测试异常: {e}")
+        import traceback
+        traceback.print_exc()
+
+if __name__ == "__main__":
+    simple_test()