template = """你是一个自动化测试工程师，精通的技术栈为 Python pytest requests库以下是这个接口的具体信息，
{context}
Question: {input}"""

#1. Test case: 正常情况下的请求import pytestimport requests
def test_normal_request():url = 'http://example.com/normal_request'response = requests.get(url)assert response.status_code == 200assert response.json()['ip_address'] == 'requester_ip'#2. Test case: 非法请求方法import pytestimport requestsdef test_invalid_request_method():url = 'http://example.com/invalid_request_method'response = requests.post(url)assert response.status_code == 405# 3. Test case: 无效的请求路径import pytestimport requestsdef test_invalid_request_path():url = 'http://example.com/invalid_request_path'response = requests.get(url)assert response.status_code == 404

template = """你是一个自动化测试工程师，你非常熟悉requests库{context}Question: 请根据传入的接口信息提取request中的 ip 、 url 、method、json。key值为前面提到的字段，如果没有则无需添加。如果有则提取对应的value。要求返回的格式为json格式"""

{"ip": "182.92.156.22","url": "https://httpbin.ceshiren.com/ip","method": "GET"}

from langchain_community.chat_models.openai import ChatOpenAIfrom langchain_community.document_loaders.text import TextLoaderfrom langchain_core.output_parsers import JsonOutputParser, StrOutputParserfrom langchain_core.prompts import PromptTemplatefrom langchain_core.runnables import RunnablePassthroughfrom utils.langchain_debug import langchain_debug
langchain_debug()llm = ChatOpenAI()

def get_by_filename(filename):info = TextLoader(f'./data/{filename}')return info.load()

def get_case_data(_):template = """你是一个自动化测试工程师，你非常熟悉requests库{context}Question: {input}请根据传入的接口信息提取request中的 ip 、 url 、method、json。key值为前面提到的字段，如果没有则无需添加。如果有则提取对应的value。要求返回的格式为json格式"""prompt = PromptTemplate.from_template(template=template, )data_chain = (RunnablePassthrough.assign(context=lambda x: get_by_filename("ip.har"), )| prompt| llm| JsonOutputParser())return data_chain

def get_case():"""通过大模型生成测试数据。:return:"""template = """你是一个自动化测试工程师，精通的技术栈为 Python pytest requests库以下是这个接口的具体信息，你的
{context}
请求的参数信息将输入一个字典，输入的内容为{req}
Question: {input}"""# 模板提示，输出 json 格式的回答prompt = PromptTemplate.from_template(template=template, )chain = (RunnablePassthrough.assign(context=lambda x: get_by_filename("获取ip测试用例.md"), req=get_case_data)| prompt| llm| StrOutputParser())
input_template = """根据每条测试用例的测试步骤，生成对应的测试数据信息，每条测试用例要求都有一条对应的单独的pytest函数"""print(chain.invoke({"input": input_template}))

if __name__ == '__main__':    get_case()