Updated generate testcase command to show popup to allow for replace all/only empty/cancel

Author: darginmathi

core/table_controller.py

@@ -133,7 +133,7 @@ def clear(self, model, selection):
                     self.undo_stack.push(command)
         self.undo_stack.endMacro()
 
-    def generate_test_cases(self, model):
+    def generate_test_cases(self, model, overwrite=False):
         test_case_generated = 0
         self.undo_stack.beginMacro("Generate Test Cases")
         for row in range(model.rowCount()):
@@ -143,7 +143,7 @@ def generate_test_cases(self, model):
             steps_performed_data = model.data(steps_performed_index, Qt.ItemDataRole.DisplayRole)
             expected_result_data = model.data(expected_result_index, Qt.ItemDataRole.DisplayRole)
 
-            if not steps_performed_data and not expected_result_data:
+            if overwrite or(not steps_performed_data and not expected_result_data):
                 command_index = model.index(row, TestScenarioModel.COMMAND_COL)
                 command = model.data(command_index, Qt.ItemDataRole.DisplayRole)
 

ui/menu.py

@@ -1,5 +1,7 @@
 from PyQt6.QtGui import QAction, QKeySequence
-from PyQt6.QtWidgets import QMenuBar
+from PyQt6.QtWidgets import QMenuBar, QMessageBox
+from core.commands import COMMANDS
+from models.test_scenario import TestScenarioModel
 from typing import TYPE_CHECKING
 
 from ui.tab import TabWidget
@@ -36,7 +38,7 @@ def create_menu_bar(self) -> None:
 
         generate_action = QAction("Generate Test Cases", self.main)
         generate_action.triggered.connect(self.generate_test_cases_action)
-        generate_action.setShortcut("")
+        generate_action.setShortcut("Ctrl+G")
         generate_action.setToolTip("Generate Test Cases")
         menubar.addAction(generate_action)
 
@@ -86,15 +88,54 @@ def _create_edit_menu(self, menubar: QMenuBar) -> None:
 
     def generate_test_cases_action(self) -> None:
         current_tab = self.main.get_current_tab()
-        if current_tab and hasattr(current_tab, 'controller'):
-            current_table = self.main.get_current_table()
-            if current_table:
-                num_generated = current_tab.controller.generate_test_cases(current_table.model)
+        if not current_tab and hasattr(current_tab, 'controller'):
+            return
+        current_table = self.main.get_current_table()
+        if not current_table:
+            return
+
+        model = current_table.model
+
+        for row in range(model.rowCount()):
+            command = model.data(model.index(row, TestScenarioModel.COMMAND_COL), 0)
+            if command in COMMANDS:
+                steps_data = model.data(model.index(row, TestScenarioModel.STEPS_COL), 0)
+                expected_data = model.data(model.index(row, TestScenarioModel.EXPECTED_COL), 0)
+                if steps_data or expected_data:
+                    overwrite_needed = True
+                    break
+
+        should_proceed = True
+        should_overwrite = False
+        if overwrite_needed:
+            msg_box = QMessageBox(self.main)
+            msg_box.setWindowTitle("Generate Test Cases")
+            msg_box.setText("Some test cases already have data.")
+            msg_box.setInformativeText("How would you like to proceed?")
+
+            reset_button = msg_box.addButton("Reset All", QMessageBox.ButtonRole.DestructiveRole)
+            fill_empty_button = msg_box.addButton("Fill Empty", QMessageBox.ButtonRole.AcceptRole)
+            cancel_button = msg_box.addButton("Cancel", QMessageBox.ButtonRole.RejectRole)
+
+            msg_box.setDefaultButton(fill_empty_button)
+            msg_box.exec()
+
+            clicked_button = msg_box.clickedButton()
+            if clicked_button == reset_button:
+                should_overwrite = True
+            elif clicked_button == fill_empty_button:
+                should_overwrite = False
+            else:
+                should_proceed = False
+                self.main.show_status_message("Operation cancelled.", "info", 3000)
+
+            if should_proceed:
+                num_generated = current_tab.controller.generate_test_cases(model, overwrite=should_overwrite)
                 if num_generated > 0:
-                    message = f"Generated {num_generated} test case(s)."
+                    message = f"Successfully generated {num_generated} test case(s)."
                     self.main.show_status_message(message, "success", 5000)
                 else:
-                    message = f"No empty fields found to generate."
+                    message = "No test cases were generated."
                     self.main.show_status_message(message, "info", 5000)
 
     def auto_adjust_cells(self) -> None:

ui/table.py

@@ -76,12 +76,12 @@ def show_context_menu(self, position):
 
         mark_skip_action = QAction("Mark Skip", self)
         mark_skip_action.triggered.connect(self.skip)
-        mark_skip_action.setShortcut("Ctrl+T")
+        mark_skip_action.setShortcut("Ctrl+R")
         menu.addAction(mark_skip_action)
 
         mark_unskip_action = QAction("Mark Unskip", self)
         mark_unskip_action.triggered.connect(self.unskip)
-        mark_unskip_action.setShortcut("Ctrl+G")
+        mark_unskip_action.setShortcut("Ctrl+T")
         menu.addAction(mark_unskip_action)
 
         menu.addSeparator()
@@ -117,10 +117,10 @@ def setup_shortcuts(self):
         self.remove_shortcut = QShortcut(QKeySequence("Ctrl+D"), self)
         self.remove_shortcut.activated.connect(self.remove_rows)
 
-        self.skip_shortcut = QShortcut(QKeySequence("Ctrl+T"), self)
+        self.skip_shortcut = QShortcut(QKeySequence("Ctrl+R"), self)
         self.skip_shortcut.activated.connect(self.skip)
 
-        self.unskip_shortcut = QShortcut(QKeySequence("Ctrl+G"), self)
+        self.unskip_shortcut = QShortcut(QKeySequence("Ctrl+T"), self)
         self.unskip_shortcut.activated.connect(self.unskip)
 
         self.copy_shortcut = QShortcut(QKeySequence("Ctrl+C"), self)