Fixed unit test.

Author: lu-ohai

tests/unitary/default_setup/model_deployment/test_model_deployment.py

@@ -28,8 +28,10 @@ class ModelDeploymentTestCase(unittest.TestCase):
         )
 
     @patch("requests.post")
-    def test_predict(self, mock_post):
+    @patch("ads.model.deployment.model_deployment.ModelDeployment.sync")
+    def test_predict(self, mock_sync, mock_post):
         """Ensures predict model passes with valid input parameters."""
+        mock_sync.return_value = Mock(lifecycle_state="ACTIVE")
         mock_post.return_value = Mock(
             status_code=200, json=lambda: {"result": "result"}
         )
@@ -50,8 +52,10 @@ def test_predict(self, mock_post):
             self.test_model_deployment.predict(data=np.array([1, 2, 3]))
 
     @patch("requests.post")
-    def test_predict_with_bytes(self, mock_post):
+    @patch("ads.model.deployment.model_deployment.ModelDeployment.sync")
+    def test_predict_with_bytes(self, mock_sync, mock_post):
         """Ensures predict model passes with bytes input."""
+        mock_sync.return_value = Mock(lifecycle_state="ACTIVE")
         byte_data = b"[[1,2,3,4]]"
         with patch.object(authutil, "default_signer") as mock_auth:
             auth = MagicMock()
@@ -66,8 +70,10 @@ def test_predict_with_bytes(self, mock_post):
             )
 
     @patch("requests.post")
-    def test_predict_with_auto_serialize_data(self, mock_post):
+    @patch("ads.model.deployment.model_deployment.ModelDeployment.sync")
+    def test_predict_with_auto_serialize_data(self, mock_sync, mock_post):
         """Ensures predict model passes with valid input parameters."""
+        mock_sync.return_value = Mock(lifecycle_state="ACTIVE")
         mock_post.return_value = Mock(
             status_code=200, json=lambda: {"result": "result"}
         )