登录/ 注册
 找回密码
 立即注册
关闭

CSDN热搜

程序园 精品问答 技术交流 资源下载
返回列表 发新帖
首页 › 业界区 › 安全 ›自动生成测试用例Agent工作流搭建

自动生成测试用例Agent工作流搭建

高小雨 前天 14:41
前言
　　随着2024年以来AI的火爆，各行各业都在探索，当然做为测试也不能落后于人，AI是真的能解决我产很多繁杂工作，解放双手；
接下来开始今天正题；
 
本次使用的是trae国际版，主要是为了使用免费的cluade大模型；
 
Agent项目结构
  1. .trae
  2.    --rules
  3.      --project_rules.md  #存储规则
  4. markdown
  5.    --生成的用例.md
  6. task
  7.    --task.yaml   #任务流
  8. testcase   #生成的csv用例存储路径
  9. utils    #一些必要脚本
  10. workflows   #工作流
  11.    --flows.yaml
  12. xmind    #存储原始的需求分析文档
复制代码
1.png

 一、project_rules.md
  1. 你是与用户紧密协作的智能体，始终使用中文沟通，始终遵循以下顶级规则，严格遵守会受到奖赏，否则会受到惩罚。
  3. - 顶级工作模式：上传需求文件规则>预估生成用例数量>用例生成规则>用例对比补充>用例文件导出规则
  4. - 当用户输入以 "/" 开头时，必须触发对应的**工作管理任务指令**
  6. ## 任务字段说明
  8. - name: 命令名称、功能
  9. - triggers: 指令触发条件，用于判断用户是否输入了指令
  10. - description: 命令描述
  11. - file: 指令文件路径
  12. - steps: 指令运行步骤，请一步一步的进行
  13. - script: 指令运行脚本
  15. ## 工作管理任务指令
  17. [
  18.     {
  19.         "name":"将提供以下工作管理指令供用户选择",
  20.         "trigger":"/",
  21.         "description":"要求只向用户展示工作指令的trigger、name和description",
  22.     },
  23.     {
  24.         "name":"测试用例生成工作",
  25.         "description":"测试用例生成工作",
  26.         "trigger":"/test",
  27.         "file":"./task/task.yml",
  28.     },
  29.     {
  30.         "name":"xmind转md工作",
  31.         "description":"将xmind文件转换为md文件",
  32.         "trigger":"/tomd",
  33.         "script":"python ./utils/xmind_to_md.py",
  34.     },
  35.     {
  36.         "name":"初始化环境2",
  37.         "description":"初始化环境",
  38.         "trigger":"/init",
  39.         "script":"python ./utils/init_environment.py",
  40.     },
  41.     {
  42.         "name":"csv转xlsx",
  43.         "description":"所有csv用例转换为xlsx格式",
  44.         "trigger":"/toxls",
  45.         "script":"python ./utils/csv_to_xlsx.py",
  46.     }
  47. ]
复制代码
二、task.yaml
  1. # 工作流基本信息
  2. id: test_workflow
  3. name: "测试用例生成任务"
  4. description: "测试用例生成任务流程"
  5. version: "1.0"
  6. # 工作流规则
  7. rules:
  8.   - 分析要尽可能详细
  9.   - 分析需求md文件时必须阅读全部内容
  10.   - 生成的所有文档如果没有指定明确路径，则统一放在 {{testcase}} 目录下
  11.   - 任务执行时，必须查阅 input 中指定的文档，这是必要的前置知识
  12.   - 任务执行时，无论任何情况下，output内容都必须遵照doc指定文件格式
  13.   - 编写文档前，先输出内容给用户查阅，等用户确认后再写入文档，避免写错反复修改
  15. # 工作步骤定义
  16. tasks:
  17.   - name: 测试用例生成
  18.     description: "测试用例生成流程"
  19.     input:
  20.       - doc: ./agent_test/markdown/*.md
  21.     rule:
  22.       - "严格遵守当前工作、任务、步骤的规则"
  23.       - "任务结束后提示用户任务完成，并结束任务"
  24.       - doc: ./agent_test/workflows/flows.yml
  25.     output:
  26.       - doc: ./agent_test/testcase/TC_{{秒级时间戳}}.csv
复制代码
三、utils目录下的脚本
csv_to_xlsx.py
  1. import pandas as pd
  2. import os
  3. import sys
  5. def convert_csv_to_xlsx():
  6.     """将testcase目录下的所有CSV文件转换为XLSX格式，并删除原CSV文件"""
  7.     # 设置控制台输出编码为utf-8
  8.     sys.stdout.reconfigure(encoding='utf-8')
  9.    
  10.     testcase_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'testcase')
  11.    
  12.     # 确保testcase目录存在
  13.     if not os.path.exists(testcase_dir):
  14.         print(f"目录不存在: {testcase_dir}")
  15.         return
  16.    
  17.     # 遍历testcase目录下的所有csv文件
  18.     for filename in os.listdir(testcase_dir):
  19.         if filename.endswith('.csv'):
  20.             csv_path = os.path.join(testcase_dir, filename)
  21.             xlsx_path = os.path.join(testcase_dir, filename.replace('.csv', '.xlsx'))
  22.             
  23.             try:
  24.                 # 读取CSV文件
  25.                 df = pd.read_csv(csv_path, encoding='utf-8')
  26.                
  27.                 # 保存为XLSX文件
  28.                 df.to_excel(xlsx_path, index=False)
  29.                 print(f"已将 {filename} 转换为XLSX格式")
  30.                
  31.                 # 删除原CSV文件
  32.                 os.remove(csv_path)
  33.                 print(f"已删除原CSV文件: {filename}")
  34.                
  35.             except Exception as e:
  36.                 print(f"处理 {filename} 时出错: {str(e)}")
  38. if __name__ == '__main__':
  39.     convert_csv_to_xlsx()
复制代码
init_environment.py
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  4. import os
  5. import sys
  6. import shutil
  7. import platform
  8. from pathlib import Path
  10. def get_windsurf_config_path():
  11.     """
  12.     获取不同系统下 Windsurf 的配置目录路径
  13.     """
  14.     # 获取用户主目录
  15.     home = os.path.expanduser("~")
  16.    
  17.     # Windows: C:\Users\<username>\.codeium\windsurf\memories
  18.     # Linux: ~/.codeium/windsurf/memories
  19.     # macOS: ~/.codeium/windsurf/memories
  20.     system = platform.system().lower()
  21.    
  22.     if system == "windows":
  23.         config_path = os.path.join(home, ".codeium", "windsurf", "memories")
  24.     elif system in ["linux", "darwin"]:  # darwin 是 macOS 的系统名
  25.         config_path = os.path.join(home, ".codeium", "windsurf", "memories")
  26.     else:
  27.         raise Exception(f"不支持的操作系统: {system}")
  28.         
  29.     return config_path
  31. def ensure_directories():
  32.     """
  33.     确保必要的目录存在，如果markdown目录存在则清空
  34.     """
  35.     try:
  36.         # 获取当前脚本所在目录的上级目录（项目根目录）
  37.         root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  38.         # 需要确保存在的目录
  39.         required_dirs = ['markdown', 'testcase', 'xmind']
  40.         
  41.         for dir_name in required_dirs:
  42.             # 使用绝对路径
  43.             dir_path = Path(os.path.join(root_dir, dir_name))
  44.             
  45.             # 如果是markdown目录且存在，清空它
  46.             if dir_name == 'markdown' and dir_path.exists():
  47.                 shutil.rmtree(dir_path)
  48.                 dir_path.mkdir()
  49.                 print(f"已清空并重新创建目录: {dir_path}")
  50.             # 如果目录不存在，创建它
  51.             elif not dir_path.exists():
  52.                 dir_path.mkdir()
  53.                 print(f"已创建目录: {dir_path}")
  54.             
  55.         return True
  56.         
  57.     except Exception as e:
  58.         print(f"初始化目录时出错: {str(e)}")
  59.         return False
  61. def init_global_rules():
  62.     """
  63.     初始化全局AI规则
  64.     """
  65.     try:
  66.         # 获取当前脚本所在目录的上级目录
  67.         current_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  68.         # 使用绝对路径读取global_rules.md文件
  69.         rules_file = os.path.join(current_dir, 'global_rules.md')
  70.         
  71.         with open(rules_file, 'r', encoding='utf-8') as f:
  72.             rules_content = f.read()
  73.         
  74.         try:
  75.             # 获取 Windsurf 配置目录路径
  76.             config_path = get_windsurf_config_path()
  77.             
  78.             # 确保目录存在
  79.             os.makedirs(config_path, exist_ok=True)
  80.             
  81.             # 构建全局规则文件的完整路径
  82.             rules_file = os.path.join(config_path, "global_rules.md")
  83.             
  84.             # 写入全局规则
  85.             with open(rules_file, 'w', encoding='utf-8') as f:
  86.                 f.write(rules_content)
  87.             print(f"已初始化全局AI规则到: {rules_file}")
  88.             return True
  89.             
  90.         except Exception as e:
  91.             print(f"写入全局规则文件时出错: {str(e)}")
  92.             return False
  93.             
  94.     except Exception as e:
  95.         print(f"初始化全局AI规则时出错: {str(e)}")
  96.         return False
  98. def initialize_environment():
  99.     """
  100.     初始化环境：创建必要目录并清理markdown目录，初始化全局AI规则
  101.     """
  102.     try:
  103.         # 确保必要的目录存在
  104.         if not ensure_directories():
  105.             return False
  106.             
  107.         # 初始化全局AI规则
  108.         if not init_global_rules():
  109.             return False
  110.             
  111.         print("环境初始化完成")
  112.         return True
  113.         
  114.     except Exception as e:
  115.         print(f"初始化环境时出错: {str(e)}")
  116.         return False
  118. if __name__ == "__main__":
  119.     # 设置控制台输出编码为UTF-8
  120.     if sys.platform == 'win32':
  121.         sys.stdout.reconfigure(encoding='utf-8')
  122.     success = initialize_environment()
  123.     sys.exit(0 if success else 1)
复制代码
 
xmind_to_md.py
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  4. import os
  5. import sys
  6. import xmindparser
  7. import datetime
  9. # 设置默认编码为utf-8
  10. if sys.version_info[0] == 3:
  11.     sys.stdout.reconfigure(encoding='utf-8')
  13. class XMindToMarkdown:
  14.     def __init__(self):
  15.         self.current_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  16.         self.xmind_dir = os.path.join(self.current_dir, 'xmind')
  17.         self.markdown_dir = os.path.join(self.current_dir, 'markdown')
  18.         
  19.     def convert_to_markdown(self, xmind_file):
  20.         """
  21.         将XMind文件转换为Markdown格式
  22.         :param xmind_file: XMind文件名
  23.         :return: 转换后的Markdown内容
  24.         """
  25.         xmind_path = os.path.join(self.xmind_dir, xmind_file)
  26.         if not os.path.exists(xmind_path):
  27.             raise FileNotFoundError(f"XMind文件不存在: {xmind_path}")
  29.         # 解析XMind文件
  30.         xmind_content = xmindparser.xmind_to_dict(xmind_path)
  31.         if not xmind_content or not xmind_content[0].get('topic'):
  32.             raise ValueError("XMind文件格式错误或内容为空")
  34.         markdown_content = []
  35.         root_topic = xmind_content[0]['topic']
  36.         
  37.         # 添加标题
  38.         markdown_content.append(f"# {root_topic.get('title', '需求文档')}\n")
  39.         
  40.         # 添加生成时间
  41.         current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
  42.         markdown_content.append(f"生成时间：{current_time}\n")
  44.         # 处理子主题
  45.         if 'topics' in root_topic:
  46.             self._process_topics(root_topic['topics'], markdown_content, level=1)
  48.         return '\n'.join(markdown_content)
  50.     def _process_topics(self, topics, markdown_content, level):
  51.         """
  52.         递归处理XMind主题
  53.         :param topics: 主题列表
  54.         :param markdown_content: Markdown内容列表
  55.         :param level: 当前层级
  56.         """
  57.         for topic in topics:
  58.             # 添加标题
  59.             title = topic.get('title', '')
  60.             markdown_content.append(f"{'#' * (level + 1)} {title}\n")
  62.             # 处理备注
  63.             if 'note' in topic:
  64.                 note = topic['note'].strip()
  65.                 if note:
  66.                     markdown_content.append(f"{note}\n")
  68.             # 处理子主题
  69.             if 'topics' in topic:
  70.                 self._process_topics(topic['topics'], markdown_content, level + 1)
  72.     def convert_all_xminds(self):
  73.         """
  74.         转换xmind目录下的所有XMind文件为Markdown格式，并保存到markdown目录
  75.         """
  76.         # 确保xmind目录存在
  77.         if not os.path.exists(self.xmind_dir):
  78.             os.makedirs(self.xmind_dir)
  79.             print(f"创建目录: {self.xmind_dir}")
  80.             return
  82.         # 确保markdown目录存在
  83.         if not os.path.exists(self.markdown_dir):
  84.             os.makedirs(self.markdown_dir)
  85.             print(f"创建目录: {self.markdown_dir}")
  87.         xmind_files = [f for f in os.listdir(self.xmind_dir) if f.endswith('.xmind')]
  88.         if not xmind_files:
  89.             print("没有找到XMind文件")
  90.             return
  92.         for xmind_file in xmind_files:
  93.             try:
  94.                 markdown_content = self.convert_to_markdown(xmind_file)
  95.                 md_filename = os.path.splitext(xmind_file)[0] + '.md'
  96.                 md_path = os.path.join(self.markdown_dir, md_filename)
  97.                
  98.                 with open(md_path, 'w', encoding='utf-8') as f:
  99.                     f.write(markdown_content)
  100.                 print(f"成功转换: {xmind_file} -> {md_filename}")
  101.             except Exception as e:
  102.                 print(f"转换 {xmind_file} 失败: {str(e)}")
  104. if __name__ == '__main__':
  105.     converter = XMindToMarkdown()
  106.     converter.convert_all_xminds()
复制代码
 
使用方法：
1.需求分析xmind文件，存放在xmind目录下
命令：
2. / 列出所有命令
3. /tomd  将xmind目录下的需求分析文件，转换为md格式；
2.png

 4./test进行用例生成
3.png

 5./toxls  将csv转换成excel格式
4.png

 
5.png

 
结果如下：
6.png

 

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

使用道具 举报
相关推荐
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册
发帖
高小雨
前天 14:41

0

粉丝关注

15

主题发布

板块介绍填写区域，请于后台编辑
微信扫一扫