python-使用登录用户在Flask App上运行Selenium测试

我正在构建一个Flask支持的Web应用程序,其中所有有趣的页面都位于登录名后面.我想使用Selenium WebDriver对它进行自动化测试.我似乎无法弄清楚如何登录用户并将其与Selenium关联.我尝试的所有操作都会导致Selenium出现在“请登录”页面中.

不可接受的解决方案1：脚本化硒登录

我见过的许多资源(1、2、3等)建议使用脚本化测试手动登录,也就是说,将Selenium导航到登录页面,填写表格,然后单击按钮.由于许多原因,这是一个糟糕的解决方案,包括但不限于：

>在某处需要以明文形式对凭证进行硬编码(我的应用程序使用社交媒体登录,因此没有可用的测试或虚构凭证)
>引入依赖关系；现在必须在测试或固定装置中运行登录例程,并且不能单独运行其他测试
>慢；必须在每次测试之前作为浏览器操作运行

不可接受的解决方案2：运行Selenium测试之前需要手动登录

我可以做的另一件事是在运行Selenium测试并登录之前打开要定位的浏览器.这使凭据不受Selenium的控制.问题：

> PITA；很多时候,我会忘记这样做,导致不得不重新运行测试.因为这很麻烦,所以我可能会少运行测试套件.
>不可扩展；如果我曾经尝试过测试多个浏览器或在不同的环境下运行,则可以直接选择该窗口
>慢；仍然需要Selenium点击登录页面并在第三方网站上点击“授权应用”.

尝试解决方案1失败：会话中断

使用这样的东西：

with client.session_transaction() as session:
    session['user_id'] = test_user.id
    session['_fresh'] = True

其中client是app.test_client()的实例. (这是在pytest装置中；我正在使用Flask_login).这使我可以在常规单元测试期间登录,但无法弄清楚如何让Selenium使用同一会话.

尝试解决方案2失败：login_user

与上述类似,我可以尝试在固定装置中调用Flask_login.login_user(test_user),但是我无法成功将Selenium与该用户关联.由于会话已经失败,所以我唯一想到的就是在Selenium driver.set_cookie上为此登录用户设置cookie,但是我不知道如何从测试客户端或Flask登录中获取合适的cookie所以.

解决方案#3失败：LOGIN_DISABLED

根据Flask-Login docs,您可以在应用程序配置中设置LOGIN_DISABLED以绕过login_required装饰器.我这样做了,但仍然看到登录页面.

我对此进行了更多研究,问题不是LOGIN_DISABLED无法正常工作,而是很多视图取决于Flask_login的current_user.is_anonymous的行为.如果要走这条路线,我还需要一个解决方案以确保将current_user设置为测试用户,以便将应用程序中的操作与“登录”的用户正确地分组.

如何在不使用Selenium的情况下进行一次身份验证,然后让Selenium在该会话的其余会话中使用经过身份验证的用户？

解决方法:

登录后如何获取Cookie.可以模拟登录过程,从中找到表单的action属性以及用户名和密码的字段名称.看一下用于存储会话密钥的cookie,假设它是sid.

get_cookie.py：

import requests
r = requests.post(action, 
                  data={'login': 'your_username', 'password': 'You_p@ssw0rd'}, 
                  allow_redirects=False)
# now r.cookies contains all returned cookies,
# print cookie you need:
print(r.cookie['sid'])

尝试在webdriver会话初始化中执行此操作.或者,您可以运行它并将此cookie馈入Selenium,例如作为命令行参数(只需使用正确的参数)即可：

硒…. –session = $(./ get_cookie.py)

标签：selenium-webdriver,flask,flask-login,pytest,python
来源： https://codeday.me/bug/20191111/2022395.html