code: def mock_host_creds(uri):
      if uri == _DATABRICKS_TRACKING_URI:
          return _TRACKING_HOST_CREDS
      elif uri == _DATABRICKS_UC_REGISTRY_URI:
          return _REGISTRY_HOST_CREDS
      raise Exception(f"Got unexpected store URI {uri}")
  
imports: from unittest import mock
  import pytest
  import mlflow
  from mlflow.utils.rest_utils import MlflowHostCreds
label_desc: Use image node_id 1 for calling a global function with example usage: mock_host_creds(uri) and returns: _TRACKING_HOST_CREDS, _REGISTRY_HOST_CREDS