资讯

历史

科技

环境与自然

成长

游戏

财经

文学与艺术

美食

健康

家居

文化

情感

汽车

三农

军事

旅行

运动

教育

生活

星座命理

自动化测试如何取验证码

创作时间:

作者:

@小白创作中心

自动化测试如何取验证码

引用

来源

https://docs.pingcode.com/baike/2696251

在自动化测试中，验证码的处理是一个常见的挑战。本文将详细介绍几种常用的验证码获取方法，包括OCR技术、短信或邮件服务API、直接访问数据库、前端劫持技术和请求拦截。每种方法都有其适用场景和优缺点，读者可以根据具体需求选择合适的方法。

自动化测试取验证码的方法包括：OCR技术、短信或邮件服务API、直接访问数据库、前端劫持技术、请求拦截。在自动化测试中，验证码的处理是一个难点，因为验证码的设计初衷是为了防止自动化脚本的攻击。以下将详细介绍一种常见的方法：OCR技术。

一、OCR技术

OCR技术是自动化测试中获取验证码的一种常见方法。OCR技术的核心在于将图片中的文字转换成可编辑的文本。以下是OCR技术的详细介绍。

1、OCR技术的工作原理

OCR（光学字符识别）是一种将扫描的图片或照片中的文字内容转化为可编辑文本的技术。它的工作原理主要包括以下几个步骤：

图像预处理：在进行OCR识别之前，对图像进行预处理是至关重要的。预处理步骤包括去噪、二值化、倾斜校正等。去噪是为了移除图像中的杂点，使字符更加清晰。二值化是将图像转换成黑白图像，以便于字符的识别。倾斜校正是调整图像中的文字方向，使其水平排列。
特征提取：特征提取是OCR识别的核心步骤。它通过分析图像中的像素点，提取出字符的形状、轮廓、边缘等特征。这些特征将作为后续识别的基础。
字符识别：字符识别是将提取的特征与字符库中的特征进行匹配，从而识别出图像中的文字。字符库通常包含各种字体、大小、风格的字符样本，以提高识别的准确性。
后处理：识别结果可能会包含一些错误，因此需要进行后处理。后处理步骤包括拼写检查、语法检查等，以提高识别的准确率。

2、OCR技术的应用场景

OCR技术在自动化测试中的应用场景包括但不限于以下几个方面：

验证码识别：在自动化测试中，验证码常常是一个难点。通过OCR技术，可以自动识别验证码，从而实现自动化登录、注册等操作。
文本提取：OCR技术可以用于从图片、扫描件中提取文本内容。例如，在自动化测试中，可能需要从截图中提取测试结果、日志等信息。
表单自动填充：在自动化测试中，可能需要填充大量的表单。通过OCR技术，可以从图片中提取表单内容，并自动填充到相应的字段中。

3、OCR技术的实现

以下是一个使用Tesseract库进行OCR识别的示例代码：

from PIL import Image
import pytesseract

# 打开验证码图片
image = Image.open('captcha.png')

# 进行OCR识别
text = pytesseract.image_to_string(image)
print(text)

在这个示例中，我们首先使用PIL库打开验证码图片，然后通过pytesseract库进行OCR识别，并打印识别结果。

二、短信或邮件服务API

利用短信或邮件服务API也是自动化测试中获取验证码的一种有效方法。以下是关于短信或邮件服务API的详细介绍。

1、短信服务API

工作原理

短信服务API通过向指定的手机号码发送短信来传递验证码。在自动化测试中，可以通过调用短信服务API，获取发送到指定手机号码的验证码。

应用场景

短信服务API主要应用于以下场景：

注册验证：在用户注册时，通过短信发送验证码进行验证。
登录验证：在用户登录时，通过短信发送验证码进行二次验证。
密码重置：在用户忘记密码时，通过短信发送验证码进行身份验证。

实现方法

以下是一个使用Twilio短信服务API获取验证码的示例代码：

from twilio.rest import Client

# 你的Twilio账户SID和认证令牌
account_sid = 'your_account_sid'
auth_token = 'your_auth_token'
client = Client(account_sid, auth_token)

# 发送短信
message = client.messages.create(
    body='Your verification code is 123456',
    from_='+1234567890',
    to='+0987654321'
)
print(message.sid)

在这个示例中，我们使用Twilio短信服务API发送验证码短信。account_sid和auth_token是你的Twilio账户信息，from_是发送短信的电话号码，to是接收短信的电话号码。

2、邮件服务API

工作原理

邮件服务API通过向指定的电子邮件地址发送邮件来传递验证码。在自动化测试中，可以通过调用邮件服务API，获取发送到指定电子邮件地址的验证码。

应用场景

邮件服务API主要应用于以下场景：

注册验证：在用户注册时，通过邮件发送验证码进行验证。
登录验证：在用户登录时，通过邮件发送验证码进行二次验证。
密码重置：在用户忘记密码时，通过邮件发送验证码进行身份验证。

实现方法

以下是一个使用SendGrid邮件服务API获取验证码的示例代码：

import sendgrid
from sendgrid.helpers.mail import Mail, Email, To, Content

# 你的SendGrid API密钥
sg = sendgrid.SendGridAPIClient(api_key='your_sendgrid_api_key')
from_email = Email("test@example.com")
to_email = To("test@example.com")
subject = "Your verification code"
content = Content("text/plain", "Your verification code is 123456")
mail = Mail(from_email, to_email, subject, content)
response = sg.client.mail.send.post(request_body=mail.get())
print(response.status_code)
print(response.body)
print(response.headers)

在这个示例中，我们使用SendGrid邮件服务API发送验证码邮件。api_key是你的SendGrid API密钥，from_email是发送邮件的电子邮件地址，to_email是接收邮件的电子邮件地址。

三、直接访问数据库

直接访问数据库是自动化测试中获取验证码的一种有效方法。以下是关于直接访问数据库的详细介绍。

1、工作原理

在一些系统中，验证码会存储在数据库中。通过直接访问数据库，可以获取存储的验证码。在自动化测试中，可以通过查询数据库表，获取最新的验证码。

2、应用场景

直接访问数据库主要应用于以下场景：

内部测试：在内部测试中，通常可以直接访问数据库，获取验证码。
开发环境：在开发环境中，通常可以直接访问数据库，获取验证码。

3、实现方法

以下是一个使用Python和MySQL数据库获取验证码的示例代码：

import mysql.connector

# 连接数据库
conn = mysql.connector.connect(
    host="localhost",
    user="your_username",
    password="your_password",
    database="your_database"
)
cursor = conn.cursor()

# 查询最新的验证码
cursor.execute("SELECT code FROM verification_codes ORDER BY created_at DESC LIMIT 1")
result = cursor.fetchone()
print(result[0])

# 关闭连接
cursor.close()
conn.close()

在这个示例中，我们使用Python和MySQL数据库查询最新的验证码。your_username、your_password和your_database是你的数据库连接信息。

四、前端劫持技术

前端劫持技术是自动化测试中获取验证码的一种有效方法。以下是关于前端劫持技术的详细介绍。

1、工作原理

前端劫持技术通过劫持网页的前端请求，获取验证码。在自动化测试中，可以通过拦截网页的请求，获取验证码的响应数据。

2、应用场景

前端劫持技术主要应用于以下场景：

Web应用测试：在Web应用测试中，可以通过劫持前端请求，获取验证码。
动态验证码：对于动态生成的验证码，可以通过劫持前端请求，获取验证码。

3、实现方法

以下是一个使用Selenium和BrowserMob Proxy进行前端劫持获取验证码的示例代码：

from selenium import webdriver
from browsermobproxy import Server

# 启动BrowserMob Proxy服务器
server = Server("path/to/browsermob-proxy")
server.start()
proxy = server.create_proxy()

# 设置Selenium使用的代理
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument("--proxy-server={0}".format(proxy.proxy))

# 启动Selenium WebDriver
driver = webdriver.Chrome(chrome_options=chrome_options)

# 开始拦截请求
proxy.new_har("captcha")

# 打开网页
driver.get("http://example.com")

# 获取验证码请求的响应数据
har = proxy.har
for entry in har['log']['entries']:
    if "captcha" in entry['request']['url']:
        captcha_response = entry['response']['content']['text']
        print(captcha_response)

# 关闭Selenium WebDriver
driver.quit()

# 停止BrowserMob Proxy服务器
server.stop()

在这个示例中，我们使用Selenium和BrowserMob Proxy劫持前端请求，获取验证码。path/to/browsermob-proxy是BrowserMob Proxy的路径，http://example.com是要测试的网页地址。

五、请求拦截

请求拦截是自动化测试中获取验证码的一种有效方法。以下是关于请求拦截的详细介绍。

1、工作原理

请求拦截通过拦截网页的请求，获取验证码。在自动化测试中，可以通过拦截网页的请求，获取验证码的响应数据。

2、应用场景

请求拦截主要应用于以下场景：

Web应用测试：在Web应用测试中，可以通过拦截请求，获取验证码。
API测试：在API测试中，可以通过拦截请求，获取验证码。

3、实现方法

以下是一个使用Python和requests库进行请求拦截获取验证码的示例代码：

import requests

# 拦截验证码请求
response = requests.get("http://example.com/api/captcha")

# 获取验证码响应数据
captcha_response = response.json()
print(captcha_response['code'])

在这个示例中，我们使用Python和requests库拦截验证码请求，获取验证码。http://example.com/api/captcha是验证码的API接口地址。