DMD-33 - update testRefreshTablesInformation

keboola · Jan 24, 2025 · fe57b22 · fe57b22
1 parent 02605f6
commit fe57b22
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/tests/Backend/ExternalBuckets/SnowflakeBucketsRefreshTablesInformationTest.php b/tests/Backend/ExternalBuckets/SnowflakeBucketsRefreshTablesInformationTest.php
@@ -53,11 +53,18 @@ public function testRefreshTablesInformation(): void
 
         $db = $this->ensureSnowflakeConnection();
 
+        $db->executeQuery(
+            sprintf(
+                'USE ROLE %s;',
+                SnowflakeQuote::quoteSingleIdentifier($this->getSnowflakeUser()),
+            ),
+        );
+
         $db->executeQuery(
             sprintf(
                 'GRANT ROLE %s TO USER %s;',
                 SnowflakeQuote::quoteSingleIdentifier($workspace['connection']['database']),
-                SnowflakeQuote::quoteSingleIdentifier((string) getenv('SNOWFLAKE_USER')),
+                SnowflakeQuote::quoteSingleIdentifier($this->getSnowflakeUser()),
             ),
         );
 
@@ -116,15 +123,15 @@ public function testRefreshTablesInformation(): void
         $db->executeQuery(
             sprintf(
                 'USE ROLE %s;',
-                SnowflakeQuote::quoteSingleIdentifier((string) getenv('SNOWFLAKE_USER')),
+                SnowflakeQuote::quoteSingleIdentifier($this->getSnowflakeUser()),
             ),
         );
 
         $db->executeQuery(
             sprintf(
                 'REVOKE ROLE %s FROM USER %s;',
                 SnowflakeQuote::quoteSingleIdentifier($workspace['connection']['database']),
-                SnowflakeQuote::quoteSingleIdentifier((string) getenv('SNOWFLAKE_USER')),
+                SnowflakeQuote::quoteSingleIdentifier($this->getSnowflakeUser()),
             ),
         );
     }