fix: fix mytest for timeout

agrawalradhika-cell · agrawalradhika-cell · commit da33b9c6cf65 · 2026-02-23T09:48:28.000-08:00
Signed-off-by: Radhika Agrawal &lt;agrawalradhika@google.com&gt;
diff --git a/google/auth/aio/transport/sessions.py b/google/auth/aio/transport/sessions.py
@@ -247,8 +247,11 @@ async def request(
                     self._auth_request, method, url, headers
                 )
             )
-            # Narrow the type: if it's an aiohttp.ClientTimeout, extract the float 'total'
-            timeout_float = timeout.total if hasattr(timeout, "total") else timeout
+            if timeout is None:
+                timeout_float = 0.0  # Or your preferred default numeric timeout
+            else:
+                # Narrow the type: if it's an aiohttp.ClientTimeout, extract the float 'total'
+                timeout_float = timeout.total if hasattr(timeout, "total") else timeout
             # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
             # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
             async for _ in retries:  # pragma: no branch

Original file line number	Diff line number	Diff line change
`@@ -247,8 +247,11 @@ async def request(`
`247`	`247`	`self._auth_request, method, url, headers`
`248`	`248`	`)`
`249`	`249`	`)`
`250`		`- # Narrow the type: if it's an aiohttp.ClientTimeout, extract the float 'total'`
`251`		`- timeout_float = timeout.total if hasattr(timeout, "total") else timeout`
	`250`	`+ if timeout is None:`
	`251`	`+ timeout_float = 0.0 # Or your preferred default numeric timeout`
	`252`	`+ else:`
	`253`	`+ # Narrow the type: if it's an aiohttp.ClientTimeout, extract the float 'total'`
	`254`	`+ timeout_float = timeout.total if hasattr(timeout, "total") else timeout`
`252`	`255`	# Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
`253`	`256`	`# See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372`
`254`	`257`	`async for _ in retries: # pragma: no branch`