接口测试进阶：文件上传接口的3个死穴，95%的测试都踩过_千锋教育

<div id="upload-container">
  <input type="file" id="file-input" style="display:none">
  <button id="upload-btn" onclick="initUpload()">选择文件</button>
</div>

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()
driver.get("https://example.com/upload")

# 错误的做法：硬等5秒
import time
time.sleep(5)  # 网络慢时可能还没渲染完
upload_success = driver.find_element(By.ID, "upload-success")
assert upload_success.is_displayed()  # 可能报错

# 正确的做法：显式等待
wait = WebDriverWait(driver, 10)
success_element = wait.until(
    EC.presence_of_element_located((By.ID, "upload-success"))
)
assert success_element.text == "上传成功"  # 100%稳定

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

# 等待上传按钮可点击
upload_btn = WebDriverWait(driver, 10).until(
    EC.element_to_be_clickable((By.ID, "upload-btn"))
)
upload_btn.click()

# 等待文件输入框出现（可能是隐藏的）
file_input = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.CSS_SELECTOR, "input[type='file']"))
)
# 发送文件路径
file_input.send_keys("/path/to/test.pdf")

# 点击上传按钮
upload_start_btn = driver.find_element(By.ID, "start-upload")
upload_start_btn.click()

# 等待上传完成状态（假设上传完成后隐藏进度条并显示成功信息）
success_message = WebDriverWait(driver, 30).until(
    EC.visibility_of_element_located((By.XPATH, "//div[contains(text(), '上传成功')]"))
)
print(f"上传结果: {success_message.text}")

# 等待文件URL元素出现（假设上传后显示文件链接）
file_url_element = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.CSS_SELECTOR, ".file-url"))
)
file_url = file_url_element.get_attribute("href")
assert file_url.endswith(".pdf"), f"文件URL格式错误: {file_url}"

# 进一步验证文件大小
file_size = driver.find_element(By.CLASS_NAME, "file-size").text
assert "2.5 MB" in file_size, f"文件大小不匹配: {file_size}"

class text_to_change(object):
    def __init__(self, locator, old_text):
        self.locator = locator
        self.old_text = old_text
    
    def __call__(self, driver):
        element = driver.find_element(*self.locator)
        return element.text != self.old_text

# 使用示例：等待上传状态文本变化
status_locator = (By.ID, "upload-status")
old_status = driver.find_element(*status_locator).text  # 初始状态："等待上传"
WebDriverWait(driver, 20).until(
    text_to_change(status_locator, old_status)
)
new_status = driver.find_element(*status_locator).text
assert new_status == "上传完成", f"状态异常: {new_status}"

class attribute_value_contains(object):
    def __init__(self, locator, attribute, value):
        self.locator = locator
        self.attribute = attribute
        self.value = value
    
    def __call__(self, driver):
        element = driver.find_element(*self.locator)
        attr_value = element.get_attribute(self.attribute)
        return self.value in attr_value

# 等待上传容器class包含"completed"
container_locator = (By.ID, "upload-container")
WebDriverWait(driver, 15).until(
    attribute_value_contains(container_locator, "class", "completed")
)

import os

# 动态获取文件路径
base_dir = os.path.dirname(os.path.abspath(__file__))
file_path = os.path.join(base_dir, "test_files", "sample.pdf")
assert os.path.exists(file_path), f"测试文件不存在: {file_path}"