Fix fetch auth token tests

googleapis · Oct 27, 2023 · 5b5cdf9 · 5b5cdf9
1 parent c2cb07d
commit 5b5cdf9
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tests/FetchAuthTokenTest.php b/tests/FetchAuthTokenTest.php
@@ -25,6 +25,7 @@
 use Google\Auth\CredentialsLoader;
 use Google\Auth\FetchAuthTokenInterface;
 use Google\Auth\OAuth2;
+use Google\Auth\UpdateMetadataInterface;
 use Prophecy\Argument;
 use Prophecy\PhpUnit\ProphecyTrait;
 
@@ -53,9 +54,19 @@ class_implements($fetcherClass)
         )) {
             $mockFetcher->getQuotaProject()->shouldBeCalledTimes(1);
         }
-        $mockFetcher->fetchAuthToken(Argument::any())
+
+        if (is_a($fetcherClass, UpdateMetadataInterface::class, true)) {
+            $mockFetcher->updateMetadata(Argument::cetera())
+                ->shouldBeCalledTimes(1)
+                ->willReturn(['authorization' => 'Bearer xyz']);
+            $mockFetcher->getLastReceivedToken()
+                ->shouldBeCalledTimes(1)
+                ->will($httpHandler);
+        } else {
+            $mockFetcher->fetchAuthToken(Argument::any())
             ->shouldBeCalledTimes(1)
             ->will($httpHandler);
+        }
         $mockFetcher->getCacheKey()->willReturn('');
 
         $tokenCallbackCalled = false;