code: def mocked_get_task_status():
      with mock.patch.object(DmsHook, "get_task_status") as m:
          yield m
  
file_path: airflow/tests/providers/amazon/aws/sensors/test_dms.py
imports: from __future__ import annotations
  from unittest import mock
  import pytest
  from airflow.exceptions import AirflowException, AirflowSkipException
  from airflow.providers.amazon.aws.hooks.dms import DmsHook
  from airflow.providers.amazon.aws.sensors.dms import DmsTaskCompletedSensor
label_desc: Use image node_id 1 for calling a global function with example usage: mocked_get_task_status() without return types