playwright解决重复登录问题,通过pytest夹具自动读取storage_state用户状态信息

playwright解决重复登录问题,通过pytest夹具自动读取storage_state用户状态信息

conftest.py文件下封装两个夹具

夹具一:将storage_state登录状态导出为json文件

  • 开启夹具自动调用 autouse=True
  • browser夹具,来源于pytest-playwright包封装的夹具,替代该代码browser = playwright.chromium.launch()。
  • base_url夹具,来源于pytest-base-url包封装好的夹具,用于在pytest.ini设置全局的base_url。
  • pytestconfig夹具,来源于pytest包,用于处理pytest.ini配置文件
python 复制代码
@pytest.fixture(scope='session', autouse=True)
def login(browser, base_url, pytestconfig):
    """
        封装统一的登录夹具,配置免登录
    :param browser: 插件pytest-playwright封装好的夹具
    :param base_url: 插件pytest-base-url插件,封装好的夹具,获取pytest.ini文件中的base_url=
    :param pytestconfig: pytest提供的夹具,用于获取pytest.ini配置文件
    :return:
    """
    context = browser.new_context(base_url=base_url)
    page = context.new_page()
    login = LoginPage(page)
    # 用户名和密码
    login.login_flow('test_0001', 'admin123')
    # pytestconfig.rootpath : 输出项目根目录,路径根据pytest.ini配置文件,定位项目目录
    context.storage_state(path=pytestconfig.rootpath.joinpath('auth').joinpath('ry_cookie.json'))
    page.close()
    context.close()

夹具二:重写夹具browser_context_args,添加storage_state登录状态

  • browser_context_args:来源于pytest-playwright包。
python 复制代码
@pytest.fixture(scope='session', autouse=True)
def browser_context_args(browser_context_args, pytestconfig):
    """
        重写browser_context_args方法,添加cookie用户状态信息
    :param browser_context_args: pytest-playwright插件中封装的方法
    :param pytestconfig: pytest封装的方法
    :return:
    """
    return {**browser_context_args, 'storage_state': pytestconfig.rootpath.joinpath('auth').joinpath('ry_cookie.json')}

登录LoginPage对象

python 复制代码
from playwright.sync_api import Page


class LoginPage:
    def __init__(self, page: Page):
        self.page = page
        # 使用xpath语法获取用户名文本框
        self.locator_username = page.locator("//form[@id='signupForm']/input[@class='form-control uname']")
        # 密码框
        self.locator_password = page.locator("//form[@id='signupForm']/input[@class='form-control pword']")
        # 登录按钮
        self.login_button = page.locator("//button[@id='btnSubmit']")

        # 必填提示
        self.username_tips = page.locator("//label[@id='username-error']")
        self.password_tips = page.locator("//label[@id='password-error']")

        self.login_error_tips = page.get_by_text("用户不存在/密码错误")
        self.login_error_tips2 = page.locator("text=用户不存在/密码错误")
        self.login_error_tips3 = page.locator("//div[@class='layui-layer-content']")

        self.navigate()

    def navigate(self):
        self.page.goto('/login')

    def fill_username(self, username):
        # 必须先点一下才能获取到
        self.locator_username.click()
        self.locator_username.fill(username)

    def fill_password(self, password):
        self.locator_password.click()
        self.locator_password.fill(password)

    def click_login(self):
        self.login_button.click()

    def login_flow(self, username, password):
        self.navigate()
        # 定位用户名
        self.fill_username(username)
        # 定位密码
        self.fill_password(password)
        # 点击登录按钮
        self.click_login()
        # 等待页面跳转,作为登录成功的标准,等待下个页面加载完成
        self.page.wait_for_url(url='/index')

    def clean_username_password(self):
        self.fill_password('')
        self.fill_username('')

演示示例:test_ry_login.py

  • 切记:设置login_page夹具,只能设置为function级别。保持默认即可。
python 复制代码
@pytest.fixture()
def login_page(page):
    yield LoginPage(page)

def test_login_success(login_page):
    """
    断言:登录成功
        断言title和url
    :param login_page:
    :return:
    """
    login_page.login_flow('admin', 'admin123')
    expect(login_page.page).to_have_title(re.compile('首页'))
    expect(login_page.page).to_have_url('/index')
相关推荐
Ada's4 小时前
【计算机基础系列】003:Python数据结构
开发语言·数据结构·python
2601_962885725 小时前
如何用 Python 计算 TRIX 三重指数平滑均线指标?
开发语言·python
还卿一钵无情泪5 小时前
Unsloth 微调 构建自己的大模型 没有GPU也能微调
linux·开发语言·人工智能·python·大模型·nlp·unsloth
用户033307413957 小时前
TypeScript using 与 AsyncDisposableStack:在真实服务端代码中协调多资源清理
前端
daisychey7 小时前
用 Playwright 做多平台内容分发,我踩过的 8 个坑
前端
蜗牛互联网7 小时前
Python消费Responses SSE事件:增量文本、超时与取消
java·开发语言·人工智能·后端·python
statistican_ABin8 小时前
中国国际旅游发展分析报告—基于世界银行国际旅游数据的多维度分析
python·机器学习·旅游
青少儿编程课堂8 小时前
背包问题(0/1 背包与完全背包)解题精讲——动态规划入门
c++·python·算法·bfs·信息学竞赛
code2cat8 小时前
【随笔】MCP工具注解的信任边界:四个提示如何参与调用决策
python·ai agent·mcp
慢云智慧空间8 小时前
从设备智能到空间理解,慢云科技如何重新定义智慧建筑
python·科技