Pytest接口自动化测试进阶
随着Web应用的发展,越来越多的功能需要用户登录才能使用。而在接口测试中,往往需要模拟用户的登录状态来进行测试。一种常见的做法是通过Cookie来维持用户的登录状态。然而,由于Cookie的有效期限制以及网站的安全策略,如何在接口测试中有效地管理和维护登录状态成为了一项挑战。
Pytest是一个功能强大、灵活且易于使用的Python测试框架,广泛应用于各种类型的测试,包括单元测试、集成测试和接口测试等。Pytest具有丰富的插件生态和简洁的语法,使得编写和维护测试代码变得更加高效和愉快。
首先,我们需要安装好Pytest库,并准备好接口自动化测试所需的测试代码和测试数据。
在进行京东网站的接口自动化测试之前,首先需要手动进行一次登录操作,并通过浏览器开发者工具或其他方式获取到登录成功后的Cookie信息。
在编写接口自动化测试的代码时,将获取到的Cookie信息加入到请求头中,以模拟用户登录状态。
接下来,我们将详细介绍如何使用Python和Pytest实现Cookie绕过,并以爬取京东网站为例进行说明。
首先,我们需要获取登录态的Cookie。我们可以使用Selenium模拟登录京东网站,并从浏览器中获取登录后的Cookie信息。
from selenium import webdriver
import random
# 设置代理信息
proxyHost = "www.16yun.cn"
proxyPort = "5445"
proxyUser = "16QMSOML"
proxyPass = "280651"
# 随机生成UA
user_agents = [
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36",
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.71 Safari/537.36",
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
]
# 随机选择一个UA
user_agent = random.choice(user_agents)
# 设置Chrome选项
options = webdriver.ChromeOptions()
# 设置UA
options.add_argument(f'user-agent={user_agent}')
# 设置代理
options.add_argument(f'--proxy-server=http://{proxyUser}:{proxyPass}@{proxyHost}:{proxyPort}')
# 启动浏览器
driver = webdriver.Chrome(options=options)
# 打开京东登录页面
driver.get('https://passport.jd.com/new/login.aspx')
# 输入账号密码并登录
driver.find_element_by_id('loginname').send_keys('your_username')
driver.find_element_by_id('nloginpwd').send_keys('your_password')
driver.find_element_by_id('loginsubmit').click()
# 获取登录后的Cookie
cookies = driver.get_cookies()
# 关闭浏览器
driver.quit()
# 将Cookie转换成字典形式
cookie_dict = {cookie['name']: cookie['value'] for cookie in cookies}
接下来,我们将使用Pytest编写测试用例,并在测试过程中使用已获取的登录态Cookie。
import pytest
import requests
@pytest.fixture(scope='session')
def login_cookie():
return cookie_dict
def test_crawl_jd_data(login_cookie):
url = 'https://www.jd.com/'
headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.71 Safari/537.36'}
response = requests.get(url, headers=headers, cookies=login_cookie)
assert response.status_code == 200
assert '京东' in response.text
在上述代码中,我们定义了一个login_cookie的fixture,用于获取登录态Cookie。然后,我们编写了一个测试用例test_crawl_jd_data,在该测试用例中,我们发送了一个带有登录态Cookie的请求,并断言返回状态码为200,并且响应中包含了“京东”关键字。
最后,我们可以使用Pytest运行测试。
$ pytest test_jd_crawler.py
Pytest接口自动化测试进阶
xiaotaomi
会员积分:6520
随着Web应用的发展,越来越多的功能需要用户登录才能使用。而在接口测试中,往往需要模拟用户的登录状态来进行测试。一种常见的做法是通过Cookie来维持用户的登录状态。然而,由于Cookie的有效期限制以及网站的安全策略,如何在接口测试中有效地管理和维护登录状态成为了一项挑战。
Pytest是一个功能强大、灵活且易于使用的Python测试框架,广泛应用于各种类型的测试,包括单元测试、集成测试和接口测试等。Pytest具有丰富的插件生态和简洁的语法,使得编写和维护测试代码变得更加高效和愉快。
首先,我们需要安装好Pytest库,并准备好接口自动化测试所需的测试代码和测试数据。
在进行京东网站的接口自动化测试之前,首先需要手动进行一次登录操作,并通过浏览器开发者工具或其他方式获取到登录成功后的Cookie信息。
在编写接口自动化测试的代码时,将获取到的Cookie信息加入到请求头中,以模拟用户登录状态。
接下来,我们将详细介绍如何使用Python和Pytest实现Cookie绕过,并以爬取京东网站为例进行说明。
首先,我们需要获取登录态的Cookie。我们可以使用Selenium模拟登录京东网站,并从浏览器中获取登录后的Cookie信息。
from selenium import webdriver
import random
# 设置代理信息
proxyHost = "www.16yun.cn"
proxyPort = "5445"
proxyUser = "16QMSOML"
proxyPass = "280651"
# 随机生成UA
user_agents = [
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36",
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.71 Safari/537.36",
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
]
# 随机选择一个UA
user_agent = random.choice(user_agents)
# 设置Chrome选项
options = webdriver.ChromeOptions()
# 设置UA
options.add_argument(f'user-agent={user_agent}')
# 设置代理
options.add_argument(f'--proxy-server=http://{proxyUser}:{proxyPass}@{proxyHost}:{proxyPort}')
# 启动浏览器
driver = webdriver.Chrome(options=options)
# 打开京东登录页面
driver.get('https://passport.jd.com/new/login.aspx')
# 输入账号密码并登录
driver.find_element_by_id('loginname').send_keys('your_username')
driver.find_element_by_id('nloginpwd').send_keys('your_password')
driver.find_element_by_id('loginsubmit').click()
# 获取登录后的Cookie
cookies = driver.get_cookies()
# 关闭浏览器
driver.quit()
# 将Cookie转换成字典形式
cookie_dict = {cookie['name']: cookie['value'] for cookie in cookies}
接下来,我们将使用Pytest编写测试用例,并在测试过程中使用已获取的登录态Cookie。
import pytest
import requests
@pytest.fixture(scope='session')
def login_cookie():
return cookie_dict
def test_crawl_jd_data(login_cookie):
url = 'https://www.jd.com/'
headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.71 Safari/537.36'}
response = requests.get(url, headers=headers, cookies=login_cookie)
assert response.status_code == 200
assert '京东' in response.text
在上述代码中,我们定义了一个login_cookie的fixture,用于获取登录态Cookie。然后,我们编写了一个测试用例test_crawl_jd_data,在该测试用例中,我们发送了一个带有登录态Cookie的请求,并断言返回状态码为200,并且响应中包含了“京东”关键字。
最后,我们可以使用Pytest运行测试。
$ pytest test_jd_crawler.py
24-04-09 16:55
643
0
回复
暂无评论