Selenium Email 測試

Selenium 負責操控瀏覽器；至於寄信那一步，你需要一個測試能讀、又能收信的收件匣。這篇 Selenium email 測試指南用 Python 搭配一個拋棄式收件匣 API——另見 Playwright 與 Cypress 版本，全都屬於開發者 email 測試的一環。

收件匣 helper

import os, time, requests
from selenium import webdriver
from selenium.webdriver.common.by import By

API = "https://moemail.app/api"
H = {"X-API-Key": os.environ["MAIL_KEY"]}

def create_inbox():
    return requests.post(API + "/emails/generate", headers=H).json()

def wait_for_email(inbox_id, timeout=30):
    end = time.time() + timeout
    while time.time() < end:
        msgs = requests.get(API + "/emails/" + inbox_id + "/messages", headers=H).json()["messages"]
        if msgs:
            return msgs[0]
        time.sleep(1.5)
    raise TimeoutError("no email arrived")

測試本體

import re

def test_signup_verification():
    inbox = create_inbox()
    driver = webdriver.Chrome()
    driver.get("https://example.test/signup")
    driver.find_element(By.ID, "email").send_keys(inbox["address"])
    driver.find_element(By.CSS_SELECTOR, "button[type=submit]").click()

    msg = wait_for_email(inbox["id"])
    code = re.search("[0-9]{6}", msg["text"]).group(0)
    driver.find_element(By.ID, "code").send_keys(code)
    assert "Welcome" in driver.page_source
    driver.quit()

頁面元素用顯式等待，email 則用那個輪詢 helper。正規表示式 [0-9]{6} 抓的是六位數代碼；若要抓連結，改成搜尋 URL 再呼叫 driver.get() 即可。

小撇步

一個測試一個收件匣，能讓每次執行彼此隔離；論速度，webhook 勝過輪詢。建立一個郵箱，再看看 OpenAPI 文件。