ACP2E-1007 Compare list redirect to old store

Author: Chhandak.Barua

app/code/Magento/Catalog/Test/Unit/CustomerData/CompareProductsTest.php

@@ -17,8 +17,10 @@
 use Magento\Catalog\Model\ResourceModel\Product\Compare\Item\Collection;
 use Magento\Framework\App\Config\ScopeConfigInterface;
 use Magento\Framework\TestFramework\Unit\Helper\ObjectManager as ObjectManagerHelper;
+use Magento\Store\Model\Website;
 use PHPUnit\Framework\MockObject\MockObject;
 use PHPUnit\Framework\TestCase;
+use Magento\Store\Model\StoreManagerInterface;
 
 class CompareProductsTest extends TestCase
 {
@@ -52,6 +54,16 @@ class CompareProductsTest extends TestCase
      */
     private $scopeConfigMock;
 
+    /**
+     * @var StoreManagerInterface
+     */
+    private $storeManagerMock;
+
+    /**
+     * @var Website|MockObject
+     */
+    private $websiteMock;
+
     /**
      * @var array
      */
@@ -77,6 +89,18 @@ protected function setUp(): void
             ->disableOriginalConstructor()
             ->getMockForAbstractClass();
 
+        $this->storeManagerMock = $this->getMockBuilder(
+            StoreManagerInterface::class
+        )->disableOriginalConstructor()
+            ->getMockForAbstractClass();
+
+        $this->websiteMock = $this->getMockBuilder(
+            Website::class
+        )->onlyMethods(
+            ['getId',]
+        )->disableOriginalConstructor()
+            ->getMock();
+
         $this->objectManagerHelper = new ObjectManagerHelper($this);
 
         $this->model = $this->objectManagerHelper->getObject(
@@ -85,7 +109,9 @@ protected function setUp(): void
                 'helper' => $this->helperMock,
                 'productUrl' => $this->productUrlMock,
                 'outputHelper' => $this->outputHelperMock,
-                'scopeConfig'  => $this->scopeConfigMock
+                'scopeConfig'  => $this->scopeConfigMock,
+                'storeManager' => $this->storeManagerMock
+
             ]
         );
     }
@@ -196,7 +222,8 @@ public function testGetSectionData()
         $this->helperMock->expects($this->once())
             ->method('getListUrl')
             ->willReturn('http://list.url');
-
+        $this->storeManagerMock->expects($this->any())->method('getWebsite')->willReturn($this->websiteMock);
+        $this->websiteMock->expects($this->any())->method('getId')->willReturn(1);
         $this->assertEquals(
             [
                 'count' => $count,
@@ -224,7 +251,8 @@ public function testGetSectionData()
                         'remove_url' => 'http://remove.url/3',
                         'productScope' => null
                     ]
-                ]
+                ],
+                'websiteId' => 1
             ],
             $this->model->getSectionData()
         );
@@ -245,12 +273,16 @@ public function testGetSectionDataNoItems()
             ->method('getListUrl')
             ->willReturn('http://list.url');
 
+        $this->storeManagerMock->expects($this->any())->method('getWebsite')->willReturn($this->websiteMock);
+        $this->websiteMock->expects($this->any())->method('getId')->willReturn(1);
+
         $this->assertEquals(
             [
                 'count' => $count,
                 'countCaption' =>  __('%1 items', $count),
                 'listUrl' => 'http://list.url',
-                'items' => []
+                'items' => [],
+                'websiteId' => 1
             ],
             $this->model->getSectionData()
         );
@@ -264,6 +296,9 @@ public function testGetSectionDataSingleItem()
             ->method('getItemCount')
             ->willReturn($count);
 
+        $this->storeManagerMock->expects($this->any())->method('getWebsite')->willReturn($this->websiteMock);
+        $this->websiteMock->expects($this->any())->method('getId')->willReturn(1);
+
         $items = $this->prepareProductsWithCorrespondingMocks(
             [
                 [
@@ -296,7 +331,8 @@ public function testGetSectionDataSingleItem()
                         'remove_url' => 'http://remove.url/12345',
                         'productScope' => null
                     ]
-                ]
+                ],
+                'websiteId' => 1
             ],
             $this->model->getSectionData()
         );