feat: add getProjectNumber to TestTrait (#145)

GoogleCloudPlatform · Jul 9, 2024 · 9c32fd8 · 9c32fd8
1 parent 8df1b94
commit 9c32fd8
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/TestUtils/TestTrait.php b/src/TestUtils/TestTrait.php
@@ -17,6 +17,9 @@
 
 namespace Google\Cloud\TestUtils;
 
+use Google\Auth\ApplicationDefaultCredentials;
+use Google\Auth\HttpHandler\HttpHandlerFactory;
+use GuzzleHttp\Psr7\Request;
 use ReflectionClass;
 
 trait TestTrait
@@ -150,4 +153,21 @@ private static function getLastReturnedSnippetValue()
     {
         return self::$lastSnippetReturnValue;
     }
+
+    private static function getProjectNumber(string $projectId): string
+    {
+        $credentials = ApplicationDefaultCredentials::getCredentials(
+            'https://www.googleapis.com/auth/cloud-platform'
+        );
+        $accessToken = $credentials->fetchAuthToken()['access_token'];
+
+        $url = 'https://cloudresourcemanager.googleapis.com/v1/projects/' . $projectId;
+        $request = new Request('GET', $url, ['authorization' => 'Bearer ' . $accessToken]);
+        $httpHandler = HttpHandlerFactory::build();
+        $response = $httpHandler($request);
+
+        $body = json_decode((string) $response->getBody(), true);
+
+        return $body['projectNumber'];
+    }
 }
diff --git a/test/TestUtils/TestTraitTest.php b/test/TestUtils/TestTraitTest.php
@@ -200,6 +200,15 @@ public function testRunInvalidFunctionSnippet()
         $this->runFunctionSnippet('function_snippet_invalid');
     }
 
+    public function testGetProjectNumber()
+    {
+        // This test requires ApplicatonDefaultCredentials
+        $this->requireEnv('GOOGLE_APPLICATION_CREDENTIALS');
+        $projectId = $this->requireEnv('GOOGLE_PROJECT_ID');
+
+        $this->assertTrue(!empty($this->getProjectNumber($projectId)));
+    }
+
     public function setUp(): void
     {
         // Clear backoffs before running each test