From e91f78b5bdbe197f76e9691582e2a62b8d77a0a7 Mon Sep 17 00:00:00 2001
From: Judah Rand <17158624+judahrand@users.noreply.github.com>
Date: Sat, 2 Mar 2024 10:16:28 +0000
Subject: [PATCH] feat: add `GCE_METADATA_DETECT_RETRIES` to
 `environment_vars.py`

---
 google/auth/compute_engine/_metadata.py | 2 +-
 google/auth/environment_vars.py         | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/google/auth/compute_engine/_metadata.py b/google/auth/compute_engine/_metadata.py
index 045033c78..9a54c9158 100644
--- a/google/auth/compute_engine/_metadata.py
+++ b/google/auth/compute_engine/_metadata.py
@@ -54,7 +54,7 @@
 # Timeout in seconds to wait for the GCE metadata server when detecting the
 # GCE environment.
 try:
-    _METADATA_DEFAULT_TIMEOUT = int(os.getenv("GCE_METADATA_TIMEOUT", 3))
+    _METADATA_DEFAULT_TIMEOUT = int(os.getenv(environment_vars.GCE_METADATA_TIMEOUT, 3))
 except ValueError:  # pragma: NO COVER
     _METADATA_DEFAULT_TIMEOUT = 3
 
diff --git a/google/auth/environment_vars.py b/google/auth/environment_vars.py
index 2037504b3..1cc8c5ee8 100644
--- a/google/auth/environment_vars.py
+++ b/google/auth/environment_vars.py
@@ -60,6 +60,11 @@
 """Environment variable providing an alternate ip:port to be used for ip-only
 GCE metadata requests."""
 
+GCE_METADATA_TIMEOUT = "GCE_METADATA_TIMEOUT"
+"""Environment variable defining the timeout in seconds to wait for the
+GCE metadata server when detecting the GCE environment.
+"""
+
 GCE_METADATA_DETECT_RETRIES = "GCE_METADATA_DETECT_RETRIES"
 """Environment variable representing the number of retries that should be
 attempted on metadata lookup.