test: update create and delete button unit tests to include wait_ready checkbox

Signed-off-by: Bobbins228 <[email protected]>

Author: Bobbins228

tests/unit_test.py

@@ -2896,28 +2896,40 @@ def test_cluster_up_down_buttons(mocker):
     )
     cluster = Cluster(createClusterConfig())
 
-    with patch("ipywidgets.Button") as MockButton, patch("ipywidgets.Output"), patch(
-        "ipywidgets.HBox"
-    ), patch.object(cluster, "up") as mock_up, patch.object(
+    with patch("ipywidgets.Button") as MockButton, patch(
+        "ipywidgets.Checkbox"
+    ) as MockCheckbox, patch("ipywidgets.Output"), patch("ipywidgets.HBox"), patch(
+        "ipywidgets.VBox"
+    ), patch.object(
+        cluster, "up"
+    ) as mock_up, patch.object(
         cluster, "down"
-    ) as mock_down:
-        # Create mock button instances
+    ) as mock_down, patch.object(
+        cluster, "wait_ready"
+    ) as mock_wait_ready:
+        # Create mock button & CheckBox instances
         mock_up_button = MagicMock()
         mock_down_button = MagicMock()
+        mock_wait_ready_check_box = MagicMock()
 
         # Ensure the mock Button class returns the mock button instances in sequence
+        MockCheckbox.side_effect = [mock_wait_ready_check_box]
         MockButton.side_effect = [mock_up_button, mock_down_button]
 
         # Call the method under test
         cluster._cluster_up_down_buttons()
 
+        # Simulate checkbox being checked or unchecked
+        mock_wait_ready_check_box.value = True  # Simulate checkbox being checked
+
         # Simulate the button clicks by calling the mock on_click handlers
         mock_up_button.on_click.call_args[0][0](None)  # Simulate clicking "Cluster Up"
         mock_down_button.on_click.call_args[0][0](
             None
         )  # Simulate clicking "Cluster Down"
 
         # Check if the `up` and `down` methods were called
+        mock_wait_ready.assert_called_once()
         mock_up.assert_called_once()
         mock_down.assert_called_once()