1、将数据库内自动化用例相关信息读取出来，生成可执行（测试）文件
通过之前的设计，我们实现了在平台上，增删改查用例、关键字、断言等操作。但最终数据库中用例的数据，要组合成可执行文件，来进行测试活动。
我们需要设计一个方法，输入project name， testcase title，定位唯一的testcase，然后通过testcase、TestCaseKeyword、assertion，这些是组成一个http请求测试的充分必要元素。然后把这个写到当前目录的新py文件，文件名就是project_name和title的拼接。构建一个可执行的python文件，执行python文件内的函数，我们就得到本次测试的结果。
回顾通常的python测试脚本，它一般有：
import requests import json def projectA_testcase1(): # step1: 有几步keyword，就有几个step注释 url = "example.com" headers = {'Content-Type': 'application/json'} data = { "name": "abc", "email": "abc@example.com" } response = requests.post(url, data=json.dumps(data), headers=headers) assert response.status_code == 200 assert response.message == "success"
而这个测试脚本需要的元素，刚好存储在我们的数据库内。我们读取数据库，传递project和testcase title，拿到需要的信息，然后生成测试文件。
import os import json import django from django.core.exceptions import ValidationError # 设置 Django 环境 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'TestBench.settings') # 修改为您的 settings 模块路径 django.setup() from testplatform.models import Project, Testcase, TestCaseKeyword, Assertion def create_test_script(project_name, testcase_title): try: project = Project.objects.get(name=project_name) except Project.DoesNotExist: raise ValidationError(f"Project '{project_name}' not found") try: testcase = Testcase.objects.get(project=project, title=testcase_title) except Testcase.DoesNotExist: raise ValidationError(f"Testcase with title '{testcase_title}' not found in project '{project_name}'") testcase_keywords = TestCaseKeyword.objects.filter(test_case=testcase) steps = [] for tk in testcase_keywords: assertions = Assertion.objects.filter(testcase_keyword=tk).values('target_value', 'operator', 'compared_value') steps.append({ 'url': tk.url, 'method': tk.method, 'params': tk.params, 'headers': tk.headers, 'body_type': tk.body_type, 'body': tk.body, 'assertions': list(assertions) }) # 生成文件内容 script_content = f""" import requests import json def {project_name}_{testcase_title}(): """ for i, step in enumerate(steps, start=1): script_content += f""" # step{i}: 执行请求并验证断言 url = "{step['url']}" method = "{step['method']}" params = {step['params']} headers = {step['headers']} body = {step['body']} """ if step['body_type'] == "application/x-www-form-urlencoded": script_content += f""" response = requests.request(method, url, params=params, headers=headers, data=body) """ else: script_content += f""" response = requests.request(method, url, params=params, headers=headers, json=json.dumps(body)) """ for assertion in step['assertions']: target_value = assertion['target_value'] operator = assertion['operator'] compared_value = assertion['compared_value'] if operator == 'greater_than': script_content += f" assert {target_value} > {compared_value}\n" elif operator == 'less_than': script_content += f" assert {target_value} < {compared_value}\n" elif operator == 'equal': script_content += f" assert {target_value} == {compared_value}\n" elif operator == 'greater_than_or_equal': script_content += f" assert {target_value} >= {compared_value}\n" elif operator == 'less_than_or_equal': script_content += f" assert {target_value} <= {compared_value}\n" elif operator == 'equal_to': script_content += f" assert '{compared_value}' in {target_value}\n" elif operator == 'contains': script_content += f" assert '{compared_value}' in {target_value}\n" elif operator == 'in': script_content += f" assert {target_value} in '{compared_value}'\n" script_content += f""" if __name__ == "__main__": {project_name}_{testcase_title}() """ # 文件名 file_name = f"{project_name}_{testcase_title}.py" # 上一级目录的casefile文件夹路径 casefile_dir = os.path.join(os.path.dirname(os.getcwd()), 'casefile') # 确保casefile文件夹存在 os.makedirs(casefile_dir, exist_ok=True) # 完整的文件路径 file_path = os.path.join(casefile_dir, file_name) # 写入文件 with open(file_path, 'w', encoding='utf-8') as file: file.write(script_content) print(f"Test script written to {file_path}") if __name__ == '__main__': create_test_script(project_name="Project B", testcase_title="autocase0807_001")