code: def test_await_model_version_creation(mock_store, await_time):
      name = "Model 1"
      version = "1"
  
      mv = ModelVersion(
          name=name,
          version=version,
          creation_timestamp=123,
          status="PENDING_REGISTRATION",
      )
      mock_store.create_model_version.return_value = mv
  
      newModelRegistryClient().create_model_version(
          name, "uri:/source", "run123", await_creation_for=await_time
      )
      if await_time and await_time > 0:
          mock_store._await_model_version_creation.assert_called_once_with(
              mv, await_time
          )
      else:
          mock_store._await_model_version_creation.assert_not_called()
  
imports: from unittest import mock
  from unittest.mock import ANY
  import pytest
  from mlflow.entities.model_registry import ModelVersion, ModelVersionTag, RegisteredModel, RegisteredModelTag
  from mlflow.exceptions import MlflowException
  from mlflow.store.entities.paged_list import PagedList
  from mlflow.store.model_registry import SEARCH_MODEL_VERSION_MAX_RESULTS_DEFAULT, SEARCH_REGISTERED_MODEL_MAX_RESULTS_DEFAULT
  from mlflow.store.model_registry.sqlalchemy_store import SqlAlchemyStore
  from mlflow.tracking._model_registry.client import ModelRegistryClient
label_desc: Use image node_id 14 for calling a global function with example usage: test_await_model_version_creation(mock_store, await_time) without return types