Mocked return values for the tag, push, and cleanup functions in the workflow

Author: tieneupin

tests/cli/test_build_images.py

@@ -87,10 +87,25 @@ def test_run(
     mock_subprocess.return_value = 0
 
     # Mock 'build_image' return values
-    mock_build.side_effect = [
+    image_paths = [
         f"{dst if dst else def_dst}/{image[0]}:{tags[0] if tags else def_tags[0]}"
         for image in images
     ]
+    mock_build.side_effect = image_paths
+
+    # Mock all the return values when tagging the images
+    all_tags = [
+        f"{image.split(':')[0]}:{tag}"
+        for image in image_paths
+        for tag in (tags if tags else def_tags)
+    ]
+    mock_tag.side_effect = all_tags
+
+    # Mock the push function
+    mock_push.return_value = True
+
+    # Mock the cleanup function
+    mock_clean.return_value = True
 
     # Run the function with the command
     run()