code: def test_extract_db_type_from_uri():
      uri = "{}://username:password@host:port/database"
      for legit_db in DATABASE_ENGINES:
          assert legit_db == extract_db_type_from_uri(
              uri.format(legit_db)
          )
          assert legit_db == get_uri_scheme(uri.format(legit_db))
  
          with_driver = legit_db + "+driver-string"
          assert legit_db == extract_db_type_from_uri(
              uri.format(with_driver)
          )
          assert legit_db == get_uri_scheme(uri.format(with_driver))
  
      for unsupported_db in ["a", "aa", "sql"]:
          with pytest.raises(
              MlflowException, match="Invalid database engine"
          ):
              extract_db_type_from_uri(unsupported_db)
  
imports: import pathlib
  import posixpath
  import pytest
  from mlflow.exceptions import MlflowException
  from mlflow.store.db.db_types import DATABASE_ENGINES
  from mlflow.utils.os import is_windows
  from mlflow.utils.uri import add_databricks_profile_info_to_artifact_uri, append_to_uri_path, append_to_uri_query_params, dbfs_hdfs_uri_to_fuse_path, extract_and_normalize_path, extract_db_type_from_uri, get_databricks_profile_uri_from_artifact_uri, get_db_info_from_uri, get_uri_scheme, is_databricks_acled_artifacts_uri, is_databricks_uri, is_fuse_or_uc_volumes_uri, is_http_uri, is_local_uri, is_valid_dbfs_uri, remove_databricks_profile_info_from_artifact_uri, resolve_uri_if_local, validate_path_is_safe
label_desc: Use image node_id 1 for calling a global function with example usage: test_extract_db_type_from_uri() without return types