File size: 2,622 Bytes
59cd554 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import transformers
import streamlit as st
import requests
import json
import sqlite3
from transformers import pipeline
# Load AI model for test case generation
test_case_generator = pipeline("text-generation", model="microsoft/CodeGPT-small")
def fetch_swagger_data(swagger_url):
try:
response = requests.get(swagger_url)
return response.json()
except Exception as e:
st.error(f"Error fetching Swagger data: {e}")
return None
def generate_bdd_test_cases(swagger_data):
test_cases = {}
for path, methods in swagger_data.get("paths", {}).items():
for method, details in methods.items():
prompt = f"Generate BDD test case for {method.upper()} {path} with {details.get('parameters', [])}"
ai_response = test_case_generator(prompt, max_length=200, num_return_sequences=1)
test_cases[f"{method.upper()} {path}"] = ai_response[0]['generated_text']
return test_cases
def execute_test_script(test_case):
# Simulate API call execution
st.write(f"Executing test: {test_case}")
return {"status": "Passed", "response_time": "200ms"}
def store_test_results(test_case, result):
conn = sqlite3.connect("test_results.db")
cursor = conn.cursor()
cursor.execute("""
CREATE TABLE IF NOT EXISTS results (test_case TEXT, status TEXT, response_time TEXT)
""")
cursor.execute("INSERT INTO results (test_case, status, response_time) VALUES (?, ?, ?)",
(test_case, result["status"], result["response_time"]))
conn.commit()
conn.close()
def show_past_results():
conn = sqlite3.connect("test_results.db")
cursor = conn.cursor()
cursor.execute("SELECT * FROM results")
results = cursor.fetchall()
conn.close()
return results
# Streamlit UI
st.title("AI-Powered API Test Case Generator")
swagger_url = st.text_input("Enter Swagger URL")
if st.button("Generate Test Cases"):
swagger_data = fetch_swagger_data(swagger_url)
if swagger_data:
test_cases = generate_bdd_test_cases(swagger_data)
st.session_state["test_cases"] = test_cases
st.success("Test Cases Generated!")
st.json(test_cases)
if "test_cases" in st.session_state:
st.subheader("Execute Test Cases")
for api, test_case in st.session_state["test_cases"].items():
if st.button(f"Run {api}"):
result = execute_test_script(test_case)
store_test_results(api, result)
st.success(f"{api} - {result['status']}")
st.subheader("Past Execution Results")
past_results = show_past_results()
st.table(past_results)
|