跳至主要内容

认证

简介

Playwright在称为browser contexts的隔离环境中执行测试。这种隔离模式提高了可重现性并防止级联测试失败。测试可以加载现有的认证状态,这消除了在每个测试中进行认证的需要,并加快了测试执行速度。

核心概念

无论您选择哪种身份验证策略,都可能需要将经过身份验证的浏览器状态存储在文件系统中。

我们建议创建playwright/.auth目录并将其添加到您的.gitignore中。您的身份验证流程将生成经过验证的浏览器状态,并将其保存到此playwright/.auth目录中的文件中。之后,测试将重用此状态并直接以已验证状态启动。

mkdir -p playwright/.auth
echo $'\nplaywright/.auth' >> .gitignore

在每个测试前登录

Playwright API可以自动交互登录表单。

以下示例演示了如何登录GitHub。执行这些步骤后,浏览器上下文将完成身份认证。

page = context.new_page()
page.goto('https://github.com/login')

# Interact with login form
page.get_by_label("Username or email address").fill("username")
page.get_by_label("Password").fill("password")
page.get_by_role("button", name="Sign in").click()
# Continue with the test

为每个测试重新登录会减慢测试执行速度。为了缓解这个问题,可以复用现有的认证状态。

复用已登录状态

Playwright 提供了一种在测试中复用已登录状态的方法。这样您只需登录一次,就可以在所有测试中跳过登录步骤。

Web应用使用基于cookie或基于令牌的认证方式,认证状态以cookies形式存储在local storageIndexedDB中。Playwright提供了browser_context.storage_state()方法,可用于从已认证的上下文中检索存储状态,然后使用预先填充的状态创建新上下文。

Cookies、本地存储和IndexedDB状态可以在不同浏览器之间共享。它们取决于您应用程序的身份验证模型,可能需要结合使用cookies、本地存储或IndexedDB。

以下代码片段从一个已认证的上下文中获取状态,并使用该状态创建一个新的上下文。

# Save storage state into the file.
storage = context.storage_state(path="state.json")

# Create a new context with the saved storage state.
context = browser.new_context(storage_state="state.json")

高级场景

会话存储

复用认证状态涵盖基于cookieslocal storageIndexedDB的认证。极少数情况下,session storage会用于存储与登录状态相关的信息。会话存储是特定于某个域的,且不会在页面加载之间持久化。Playwright不提供持久化会话存储的API,但可以使用以下代码片段来保存/加载会话存储。

import os
# Get session storage and store as env variable
session_storage = page.evaluate("() => JSON.stringify(sessionStorage)")
os.environ["SESSION_STORAGE"] = session_storage

# Set session storage in a new context
session_storage = os.environ["SESSION_STORAGE"]
context.add_init_script("""(storage => {
if (window.location.hostname === 'example.com') {
const entries = JSON.parse(storage)
for (const [key, value] of Object.entries(entries)) {
window.sessionStorage.setItem(key, value)
}
}
})('""" + session_storage + "')")