Fix test

Author: bpblanken

v03_pipeline/lib/tasks/dataproc/base_run_job_on_dataproc.py

@@ -61,7 +61,7 @@ def complete(self) -> bool:
             google.cloud.dataproc_v1.types.jobs.JobStatus.State.ERROR,
             google.cloud.dataproc_v1.types.jobs.JobStatus.State.ATTEMPT_FAILURE,
         }:
-            msg = f'Job {self.task.task_family}-{self.run_id} entered {job.status.state!s} state'
+            msg = f'Job {self.task.task_family}-{self.run_id} entered {job.status.state.name} state'
             logger.error(msg)
             logger.error(job.status.details)
         return (

v03_pipeline/lib/tasks/dataproc/create_dataproc_cluster.py

@@ -171,7 +171,7 @@ def complete(self) -> bool:
             google.cloud.dataproc_v1.types.clusters.ClusterStatus.State.ERROR_DUE_TO_UPDATE,
         }:
             msg = (
-                f'Cluster {cluster.cluster_name} entered {cluster.status.state!s} state'
+                f'Cluster {cluster.cluster_name} entered {cluster.status.state.name} state'
             )
             logger.error(msg)
         # This will return False when the cluster is "CREATING"

v03_pipeline/lib/tasks/dataproc/run_pipeline_on_dataproc_test.py

@@ -55,7 +55,7 @@ def test_job_already_exists_failed(
         mock_logger.error.assert_has_calls(
             [
                 call(
-                    'Job RunPipelineTask-manual__2024-04-03 entered State.ERROR state',
+                    'Job RunPipelineTask-manual__2024-04-03 entered ERROR state',
                 ),
             ],
         )