MAGETWO-66189: [GitHub][PR] Bugfix for _getProductCollection on a product page #7598

Author: magento-team

app/code/Magento/Catalog/Block/Product/ListProduct.php

@@ -84,7 +84,7 @@ public function __construct(
     }
 
     /**
-     * Retrieve loaded category collection
+     * Retrieve loaded product collection
      *
      * The goal of this method is to choose whether the existing collection should be returned
      * or a new one should be initialized.
@@ -357,7 +357,7 @@ private function initializeProductCollection()
             // if the product is associated with any category
             if ($categories->count()) {
                 // show products from this category
-                $this->setCategoryId(current($categories->getIterator()));
+                $this->setCategoryId(current($categories->getIterator())->getId());
             }
         }
 

app/code/Magento/Catalog/Test/Unit/Block/Product/ListProductTest.php

@@ -49,6 +49,26 @@ class ListProductTest extends \PHPUnit_Framework_TestCase
      * @var \Magento\Framework\Url\Helper\Data | \PHPUnit_Framework_MockObject_MockObject
      */
     protected $urlHelperMock;
+    
+    /**
+     * @var \Magento\Catalog\Model\ResourceModel\Category | \PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $catCollectionMock;
+
+    /**
+     * @var \Magento\Catalog\Model\ResourceModel\Product | \PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $prodCollectionMock;
+
+    /**
+     * @var \Magento\Framework\View\LayoutInterface | \PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $layoutMock;
+
+    /**
+     * @var \Magento\Catalog\Block\Product\ProductList\Toolbar | \PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $toolbarMock;
 
     protected function setUp()
     {
@@ -92,6 +112,34 @@ protected function setUp()
             '',
             false
         );
+        $this->catCollectionMock = $this->getMock(
+            \Magento\Catalog\Model\ResourceModel\Category\Collection::class,
+            [],
+            [],
+            '',
+            false
+        );
+        $this->prodCollectionMock = $this->getMock(
+            \Magento\Catalog\Model\ResourceModel\Product\Collection::class,
+            [],
+            [],
+            '',
+            false
+        );
+        $this->layoutMock = $this->getMock(
+            \Magento\Framework\View\LayoutInterface::class,
+            [],
+            [],
+            '',
+            false
+        );
+        $this->toolbarMock = $this->getMock(
+            \Magento\Catalog\Block\Product\ProductList\Toolbar::class,
+            [],
+            [],
+            '',
+            false
+        );
 
         $this->urlHelperMock = $this->getMockBuilder(\Magento\Framework\Url\Helper\Data::class)
             ->disableOriginalConstructor()->getMock();
@@ -105,6 +153,8 @@ protected function setUp()
                 'urlHelper' => $this->urlHelperMock,
             ]
         );
+        $this->block->setToolbarBlockName('mock');
+        $this->block->setLayout($this->layoutMock);
     }
 
     protected function tearDown()
@@ -121,26 +171,59 @@ public function testGetIdentities()
             ->method('getIdentities')
             ->will($this->returnValue([$productTag]));
 
-        $itemsCollection = new \ReflectionProperty(
-            \Magento\Catalog\Block\Product\ListProduct::class,
-            '_productCollection'
-        );
-        $itemsCollection->setAccessible(true);
-        $itemsCollection->setValue($this->block, [$this->productMock]);
+        $this->productMock->expects($this->once())
+            ->method('getCategoryCollection')
+            ->will($this->returnValue($this->catCollectionMock));
+
+        $this->catCollectionMock->expects($this->once())
+            ->method('load')
+            ->will($this->returnValue($this->catCollectionMock));
+
+        $this->catCollectionMock->expects($this->once())
+            ->method('setPage')
+            ->will($this->returnValue($this->catCollectionMock));
+
+        $this->catCollectionMock->expects($this->once())
+            ->method('count')
+            ->will($this->returnValue(1));
+
+        $this->registryMock->expects($this->any())
+            ->method('registry')
+            ->will($this->returnValue($this->productMock));
 
         $currentCategory = $this->getMock(\Magento\Catalog\Model\Category::class, [], [], '', false);
-        $currentCategory->expects($this->once())
+        $currentCategory->expects($this->any())
             ->method('getId')
             ->will($this->returnValue('1'));
 
-        $this->layerMock->expects($this->once())
+        $this->catCollectionMock->expects($this->once())
+            ->method('getIterator')
+            ->will($this->returnValue([$currentCategory]));
+
+        $this->prodCollectionMock->expects($this->any())
+            ->method('getIterator')
+            ->will($this->returnValue(new \ArrayIterator([$this->productMock])));
+ 
+        $this->layerMock->expects($this->any())
             ->method('getCurrentCategory')
             ->will($this->returnValue($currentCategory));
 
+        $this->layerMock->expects($this->once())
+            ->method('getProductCollection')
+            ->will($this->returnValue($this->prodCollectionMock));
+
+        $this->layoutMock->expects($this->once())
+            ->method('getBlock')
+            ->will($this->returnValue($this->toolbarMock));
+
         $this->assertEquals(
-            [$productTag, $categoryTag ],
+            [$productTag, $categoryTag],
             $this->block->getIdentities()
         );
+        $this->assertEquals(
+            '1',
+            $this->block->getCategoryId()
+        );
     }
 
     public function testGetAddToCartPostParams()