Fix more tests

Author: rohitc33

axlearn/cloud/gcp/job_test.py

@@ -111,7 +111,7 @@ def test_delete(self):
         patch_delete = mock.patch(f"{job.__name__}.delete_k8s_jobset")
         with patch_delete as mock_delete:
             cfg, _ = self._job_config(command="test-command", bundler_cls=CloudBuildBundler)
-            gke_job = cfg.instantiate(bundler=mock.Mock())
+            gke_job = cfg.instantiate(bundler=mock.create_autospec(Bundler))
             gke_job._delete()  # pylint: disable=protected-access
             mock_delete.assert_called()
 
@@ -295,6 +295,6 @@ def test_delete(self):
         patch_delete = mock.patch(f"{job.__name__}.delete_k8s_leaderworkerset")
         with patch_delete as mock_delete:
             cfg, _ = self._job_config(command="test-command", bundler_cls=CloudBuildBundler)
-            gke_job = cfg.instantiate(bundler=mock.Mock())
+            gke_job = cfg.instantiate(bundler=mock.create_autospec(Bundler))
             gke_job._delete()  # pylint: disable=protected-access
             mock_delete.assert_called()

axlearn/cloud/gcp/jobset_utils_test.py

@@ -114,7 +114,7 @@ def test_validate_jobset_name(self):
             self._job_config(bundler_cls=ArtifactRegistryBundler) as (cfg, _),
         ):
             cfg.set(name="invalid_underscore_name", command="", output_dir="")
-            cfg.instantiate(bundler=mock.Mock())
+            cfg.instantiate(bundler=mock.create_autospec(Bundler))
 
     @parameterized.product(
         [
@@ -603,7 +603,7 @@ def __call__(self):
             self.assertEqual(cfg.inner[child].name, child)
             self.assertEqual(cfg.inner[child].command, f"{child}_command")
 
-        composite = cfg.instantiate(bundler=mock.Mock())
+        composite = cfg.instantiate(bundler=mock.create_autospec(Bundler))
         self.assertNestedEqual(
             [{"name": "a", "command": "a_command"}, {"name": "b", "command": "b_command"}],
             composite(),