allow use of custom workspace client in initTestWorkspace

Author: zajca

tests/Backend/Features/RealReadOnlyBranches/SnowflakeRealBranchesTest.php

@@ -108,6 +108,7 @@ public function testBucketsVisibilityBetweenBranchWorkspaces(): void
             true,
             true,
             $devClient,
+            $this->workspaceSapiClient->getBranchAwareClient($branchId),
         );
         $wsDevBackend = WorkspaceBackendFactory::createWorkspaceBackend($wsDev, true);
         /**

tests/Backend/Workspaces/ParallelWorkspacesTestCase.php

@@ -44,17 +44,21 @@ protected function initTestWorkspace(
         array $options = [],
         bool $forceRecreate = false,
         bool $async = true,
-        ?Client $client = null
+        ?Client $client = null,
+        ?Client $workspaceClient = null
     ): array {
         if ($backend) {
             $options['backend'] = $backend;
         }
         if ($client === null) {
             $client = $this->_client;
         }
+        if ($workspaceClient === null) {
+            $workspaceClient = $this->workspaceSapiClient;
+        }
 
         $oldWorkspaces = $this->listTestWorkspaces($client);
-        $workspaces = new Workspaces($this->workspaceSapiClient);
+        $workspaces = new Workspaces($workspaceClient);
 
         $oldWorkspace = $oldWorkspaces ? reset($oldWorkspaces) : null;
         if (!$oldWorkspace) {
@@ -64,7 +68,7 @@ protected function initTestWorkspace(
             || $oldWorkspace['connection']['backend'] === $options['backend'];
         if (!$forceRecreate && $couldReuseExistingWorkspace) {
             $result = $workspaces->resetWorkspacePassword($oldWorkspace['id']);
-            if ($this->getDefaultBackend($this->workspaceSapiClient) === self::BACKEND_BIGQUERY) {
+            if ($this->getDefaultBackend($workspaceClient) === self::BACKEND_BIGQUERY) {
                 $oldWorkspace['connection']['credentials'] = $result['credentials'];
             } else {
                 $oldWorkspace['connection']['password'] = $result['password'];