diff --git a/src/MigrationChecker.php b/src/MigrationChecker.php index 35c73ac..8ba05a9 100644 --- a/src/MigrationChecker.php +++ b/src/MigrationChecker.php @@ -28,10 +28,17 @@ public function postMigrationCheckStructure(Role $mainRoleWithGrants): void $warehouses = $mainRoleWithGrants->getAssignedGrants()->getWarehouseGrants(); assert(count($warehouses) > 0); - $useWarehouse = sprintf('USE WAREHOUSE %s', Helper::quoteIdentifier(current($warehouses)->getName())); + $useWarehouseSource = sprintf( + 'USE WAREHOUSE %s', + Helper::quoteIdentifier($this->config->getSourceSnowflakeWarehouse()), + ); + $useWarehouseDestination = sprintf( + 'USE WAREHOUSE %s', + Helper::quoteIdentifier(current($warehouses)->getName()), + ); - $this->sourceConnection->query($useWarehouse); - $this->destinationConnection->query($useWarehouse); + $this->sourceConnection->query($useWarehouseSource); + $this->destinationConnection->query($useWarehouseDestination); foreach ($this->databases as $database) { $this->logger->info(sprintf('Checking database %s', $database));