test failing fix

Author: kalindi-adhiya

src/Command/Pull/PullCommandBase.php

@@ -66,6 +66,28 @@ public function __construct(
         parent::__construct($this->localMachineHelper, $this->datastoreCloud, $this->datastoreAcli, $this->cloudCredentials, $this->telemetryHelper, $this->projectDir, $this->cloudApiClientService, $this->sshHelper, $this->sshDir, $logger, $this->selfUpdateManager);
     }
 
+    /**
+     * Get the backup download URL.
+     * This is primarily used for testing purposes.
+     */
+    public function getBackupDownloadUrl(): ?UriInterface
+    {
+        return $this->backupDownloadUrl ?? null;
+    }
+
+    /**
+     * Set the backup download URL.
+     * This is primarily used for testing purposes.
+     */
+    public function setBackupDownloadUrl(string|UriInterface $url): void
+    {
+        if (is_string($url)) {
+            $this->backupDownloadUrl = new \GuzzleHttp\Psr7\Uri($url);
+        } else {
+            $this->backupDownloadUrl = $url;
+        }
+    }
+
     /**
      * @see https://github.com/drush-ops/drush/blob/c21a5a24a295cc0513bfdecead6f87f1a2cf91a2/src/Sql/SqlMysql.php#L168
      * @return string[]
@@ -404,6 +426,28 @@ private function validateGzipFile(string $localFilepath): void
         }
     }
 
+    public static function displayDownloadProgress(mixed $totalBytes, mixed $downloadedBytes, mixed &$progress, OutputInterface $output): void
+    {
+        if ($totalBytes > 0 && is_null($progress)) {
+            $progress = new \Symfony\Component\Console\Helper\ProgressBar($output, $totalBytes);
+            $progress->setFormat('        %current%/%max% [%bar%] %percent:3s%%');
+            $progress->setProgressCharacter('💧');
+            $progress->setOverwrite(true);
+            $progress->start();
+        }
+
+        if (!is_null($progress)) {
+            if ($totalBytes === $downloadedBytes && $progress->getProgressPercent() !== 1.0) {
+                $progress->finish();
+                if ($output instanceof \Symfony\Component\Console\Output\ConsoleSectionOutput) {
+                    $output->clear();
+                }
+                return;
+            }
+            $progress->setProgress($downloadedBytes);
+        }
+    }
+
     /**
      * Create an on-demand backup and wait for it to become available.
      */

tests/phpunit/src/Commands/Pull/PullCommandTestBase.php

@@ -421,6 +421,12 @@ protected function mockDownloadBackup(object $database, object $environment, obj
             $backup->completedAt,
         ]) . '.sql.gz';
         $localFilepath = Path::join(sys_get_temp_dir(), $filename);
+
+        // Create a valid gzip file for validation.
+        $content = 'Mock SQL dump content for testing';
+        $gzippedContent = gzencode($content);
+        file_put_contents($localFilepath, $gzippedContent);
+
         $this->clientProphecy->addOption('sink', $localFilepath)
             ->shouldBeCalled();
         $this->clientProphecy->addOption('curl.options', [
@@ -462,6 +468,7 @@ protected function mockDownloadCodebaseBackup(object $database, string $url, obj
         // Mock the HTTP client request for codebase downloads.
         $downloadUrl = $backup->links->download->href ?? 'https://example.com/download-backup';
         $response = $this->prophet->prophesize(ResponseInterface::class);
+        $response->getStatusCode()->willReturn(200);
 
         $capturedOpts = null;
         $this->httpClientProphecy
@@ -484,7 +491,12 @@ protected function mockDownloadCodebaseBackup(object $database, string $url, obj
                     return true;
                 })
             )
-            ->will(function () use (&$capturedOpts, $response): ResponseInterface {
+            ->will(function () use (&$capturedOpts, $response, $localFilepath): ResponseInterface {
+                // Create a valid gzip file for validation.
+                $content = 'Mock SQL dump content for testing';
+                $gzippedContent = gzencode($content);
+                file_put_contents($localFilepath, $gzippedContent);
+
                 // Simulate the download to force progress rendering.
                 $progress = $capturedOpts['progress'];
                 $progress(100, 0);