10个常用的python自动化测试脚本

大家好，欢迎来到IT知识分享网。

Python是一门功能强大的编程语言，不仅可以用于开发应用程序，还可以用于自动化测试。对于测试人员来说，掌握一些简单但实用的Python自动化测试脚本，可以提高工作效率并减少重复劳动。本文将介绍10个常用的Python自动化测试脚本，并提供实际接口自动化工作中的示例代码。

批量执行接口测试用例

import unittest class MyTestCase(unittest.TestCase): def test_case1(self): # 执行接口测试用例1 pass def test_case2(self): # 执行接口测试用例2 pass if __name__ == '__main__': unittest.main() 

数据驱动测试

import csv def read_test_data(file_path): test_data = [] with open(file_path, 'r') as file: reader = csv.DictReader(file) for row in reader: test_data.append(row) return test_data def test_with_data(test_data): for data in test_data: # 使用测试数据执行接口测试 pass if __name__ == '__main__': test_data = read_test_data('test_data.csv') test_with_data(test_data) 

接口监控

import requests def check_api_status(api_url): response = requests.get(api_url) if response.status_code == 200: print(f"接口{api_url}正常") else: print(f"接口{api_url}异常") if __name__ == '__main__': api_urls = ['https://api.example.com/user', 'https://api.example.com/product'] for url in api_urls: check_api_status(url) 

接口性能测试

import time import requests def measure_response_time(api_url): start_time = time.time() response = requests.get(api_url) end_time = time.time() response_time = end_time - start_time print(f"接口{api_url}的响应时间为：{response_time}秒") if __name__ == '__main__': api_urls = ['https://api.example.com/user', 'https://api.example.com/product'] for url in api_urls: measure_response_time(url) 

数据库测试

import pymysql def connect_to_database(host, user, password, database): conn = pymysql.connect(host=host, user=user, password=password, database=database) return conn def execute_sql_query(conn, sql_query): cursor = conn.cursor() cursor.execute(sql_query) result = cursor.fetchall() cursor.close() return result if __name__ == '__main__': db_conn = connect_to_database('localhost', 'root', 'password', 'mydatabase') result = execute_sql_query(db_conn, 'SELECT * FROM users') print(result) db_conn.close() 

页面截图

from selenium import webdriver def take_screenshot(url, file_path): driver = webdriver.Chrome() driver.get(url) driver.save_screenshot(file_path) driver.quit() if __name__ == '__main__': take_screenshot('https://www.example.com', 'screenshot.png') 

邮件发送报告

import smtplib from email.mime.multipart import MIMEMultipart from email.mime.text import MIMEText def send_email(sender, password, receiver, subject, message): smtp_server = 'smtp.example.com' smtp_port = 587 msg = MIMEMultipart() msg['From'] = sender msg['To'] = receiver msg['Subject'] = subject msg.attach(MIMEText(message, 'plain')) with smtplib.SMTP(smtp_server, smtp_port) as server: server.starttls() server.login(sender, password) server.send_message(msg) if __name__ == '__main__': sender = '' password = 'password' receiver = '' subject = '自动化测试报告' message = '这是自动化测试的报告，请查收。' send_email(sender, password, receiver, subject, message) 

数据生成器

import random def generate_random_number(min_value, max_value): return random.randint(min_value, max_value) def generate_test_data(num_samples): test_data = [] for _ in range(num_samples): test_data.append(generate_random_number(1, 100)) return test_data if __name__ == '__main__': data = generate_test_data(10) print(data) 

日志记录

import logging def configure_logging(log_file): logging.basicConfig(filename=log_file, level=logging.DEBUG) def log_info(message): logging.info(message) if __name__ == '__main__': log_file = 'test.log' configure_logging(log_file) log_info('这是一条日志记录') 

接口数据验证

import json def validate_response_data(response_data): try: data = json.loads(response_data) if 'name' in data and 'email' in data: print("接口返回数据验证通过") else: print("接口返回数据验证不通过") except ValueError: print("接口返回数据无效") if __name__ == '__main__': response_data = '{"name": "John", "email": ""}' validate_response_data(response_data) 

以上是常用的10个Python自动化测试脚本，涵盖了批量执行测试用例、数据驱动测试、接口监控、接口性能测试、数据库测试、页面截图、邮件发送报告、数据生成器、日志记录和接口数据验证等方面。通过学习和掌握这些脚本的使用，文职人员可以更好地应对接口自动化测试工作，提高工作效率和质量。希望本文对你有所帮助！

感兴趣的小伙伴，赠送全套Python学习资料，包含面试题、简历资料等具体看下方。

一、Python所有方向的学习路线

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照下面的知识点去找对应的学习资源，保证自己学得较为全面。

二、Python必备开发工具

工具都帮大家整理好了，安装就可直接上手！

三、最新Python学习笔记

当我学到一定基础，有自己的理解能力的时候，会去阅读一些前辈整理的书籍或者手写的笔记资料，这些笔记详细记载了他们对一些技术点的理解，这些理解是比较独到，可以学到不一样的思路。

四、Python视频合集

观看全面零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

五、实战案例

纸上得来终觉浅，要学会跟着视频一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

六、面试宝典

简历模板

若有侵权，请联系删除