init

backend.py

@@ -16,15 +16,9 @@ def clean_test_case_output(text):
     """
     Cleans the output to handle HTML characters and unwanted tags.
     """
-    # Unescape HTML entities (convert &lt; back to < and &gt; back to >)
-    text = html.unescape(text)
-    
-    # Use BeautifulSoup to handle HTML tags more comprehensively
-    soup = BeautifulSoup(text, 'html.parser')
-    
-    # Convert <br> tags to newlines and remove all other tags
-    cleaned_text = soup.get_text(separator="\n").strip()
-    
+    text = html.unescape(text)  # Unescape HTML entities
+    soup = BeautifulSoup(text, 'html.parser')  # Use BeautifulSoup to handle HTML tags
+    cleaned_text = soup.get_text(separator="\n").strip()  # Remove tags and handle newlines
     return cleaned_text
 
 def generate_testcases(user_story):
@@ -36,12 +30,65 @@ def generate_testcases(user_story):
     :param user_story: A string representing the user story for which to generate test cases.
     :return: A list of test cases in the form of dictionaries.
     """
+
+    # Few-shot learning examples to guide the model
+    few_shot_examples = """
+    Example 1:
+    User Story:
+    Ensure the "Key Details" feature is enabled in the backend.
+    1. Open the Tech 360 app.
+    2. Click on the Search tab.
+    3. Enter the account number and click on Enter.
+    4. Choose the account from the Search result.
+    5. Verify that the "Key Details" button is displayed under the recommendation section.
+    6. Clicking on "Key Details" should navigate to the key details screen.
+    
+    Test Case:
+    Test Case: Verify "Key Details" Feature in Tech 360 App
+    Steps:
+    1. Launch the Tech 360 app.
+    2. Go to the Search tab.
+    3. Enter a valid account number and click Enter.
+    4. Select an account from the search results.
+    5. Ensure the "Key Details" button is displayed under the recommendation section.
+    6. Click on "Key Details" and verify that it navigates to the Key Details screen.
+    Expected Result: User is navigated to the Key Details screen successfully.
+    
+    Example 2:
+    User Story:
+    Open the Tech 360 app. Go On-Job on a Wi-Fi Ready Preinstall Job. Wait for PHT to finish. Go to the Device Details Page for XB7 and ONU.
+    
+    Test Case:
+    Test Case: Verify Device Details Page for XB7 and ONU in Wi-Fi Ready Preinstall Job
+    Steps:
+    1. Launch the Tech 360 app.
+    2. Go On-Job on a Wi-Fi Ready Preinstall Job.
+    3. Wait for PHT to finish.
+    4. Navigate to the Device Details Page for XB7 and ONU.
+    Expected Result: Device details for XB7 and ONU are displayed correctly.
+    
+    Example 3:
+    User Story:
+    Open the Tech 360 app. Go On-Job on a Wi-Fi Ready Preinstall Job. Initiate Add/Remove/Swap Flows for ONU and XB7.
+    
+    Test Case:
+    Test Case: Verify Add/Remove/Swap Flows for ONU and XB7 in Wi-Fi Ready Preinstall Job
+    Steps:
+    1. Launch the Tech 360 app.
+    2. Go On-Job on a Wi-Fi Ready Preinstall Job.
+    3. Initiate Add/Remove/Swap flows for ONU and XB7.
+    Expected Result: Add/Remove/Swap flows for ONU and XB7 are initiated and processed successfully.
+    """
+
+    # Combine the few-shot examples with the user story for the model to process
+    prompt = few_shot_examples + f"\nUser Story: {user_story}\n"
+
     try:
-        # Example few-shot learning prompt to guide the model
+        # Call the Nvidia Mistral API with the refined prompt
         completion = client.chat.completions.create(
             model="nv-mistralai/mistral-nemo-12b-instruct",  # Using Mistral model
             messages=[
-                {"role": "user", "content": f"Generate QA test cases for the following user story: {user_story}"}
+                {"role": "user", "content": prompt}
             ],
             temperature=0.06,  # Further lowering temperature for precise and deterministic output
             top_p=0.5,  # Prioritize high-probability tokens even more
@@ -81,6 +128,7 @@ def generate_testcases(user_story):
     except requests.exceptions.RequestException as e:
         print(f"API request failed: {str(e)}")
         return []
+
 # Add options for multiple test case formats
 def export_test_cases(test_cases, format='json'):
     if not test_cases:
@@ -93,7 +141,6 @@ def export_test_cases(test_cases, format='json'):
         # Improve JSON export to be line-by-line formatted
         return json.dumps(test_cases, indent=4, separators=(',', ': '))  # More readable format
     elif format == 'csv':
-        # Check if test_cases is a list of dicts
         if isinstance(test_cases, list) and isinstance(test_cases[0], dict):
             output = io.StringIO()
             csv_writer = csv.DictWriter(output, fieldnames=test_cases[0].keys(), quoting=csv.QUOTE_ALL)
@@ -119,4 +166,3 @@ def save_test_cases_as_file(test_cases, format='json'):
     else:
         return f"Unsupported format: {format}"
     return f'{format} file saved'
-