def test_prompt_to_choose_backend_vertex(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
  """Choosing Vertex backend returns (None, project, region)."""
  monkeypatch.setattr(click, "prompt", lambda *a, **k: "2")
  monkeypatch.setattr(cli_create, "_prompt_for_google_cloud", lambda _v: "proj")
  monkeypatch.setattr(
      cli_create, "_prompt_for_google_cloud_region", lambda _v: "region"
  )

  api_key, proj, region = cli_create._prompt_to_choose_backend(None, None, None)
  assert api_key is None
  assert proj == "proj"
  assert region == "region"
Choosing Vertex backend returns (None, project, region).
tests/unittests/cli/utils/test_cli_create.py