113 lines
4.0 KiB
Python
113 lines
4.0 KiB
Python
#!/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()
|