Update mock of upgrade container

PrestaShop · Jan 24, 2025 · ff6ccf1 · ff6ccf1
1 parent 049fc74
commit ff6ccf1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/unit/Controller/AbstractGlobalControllerTest.php b/tests/unit/Controller/AbstractGlobalControllerTest.php
@@ -2,6 +2,7 @@
 
 use PHPUnit\Framework\TestCase;
 use PrestaShop\Module\AutoUpgrade\Router\Routes;
+use PrestaShop\Module\AutoUpgrade\Router\UrlGenerator;
 use PrestaShop\Module\AutoUpgrade\UpgradeContainer;
 use Symfony\Component\HttpFoundation\Request;
 
@@ -28,8 +29,12 @@ public function testRedirectTo($destination, $currentUrl, $expectedUrl)
     {
         $upgradeContainer = $this->getMockBuilder(UpgradeContainer::class)
             ->disableOriginalConstructor()
+            ->setMethods(['getUrlGenerator'])
             ->getMock();
 
+        $upgradeContainer->method('getUrlGenerator')
+            ->willReturn(new UrlGenerator('/yo/doge', 'admin-wololo'));
+
         $server = [
             'HTTP_HOST' => 'localhost',
             'SERVER_PORT' => '80',