Merge remote-tracking branch 'remotes/github/MAGETWO-94565' into EPAM-PR-74

VeronikaYarats · VeronikaYarats · commit fe280cfea5ad · 2019-09-17T18:49:44.000+03:00
diff --git a/app/code/Magento/Catalog/Model/ResourceModel/Category.php b/app/code/Magento/Catalog/Model/ResourceModel/Category.php
@@ -9,13 +9,15 @@
  *
  * @author      Magento Core Team <core@magentocommerce.com>
  */
+declare(strict_types=1);
+
 namespace Magento\Catalog\Model\ResourceModel;
 
 use Magento\Catalog\Model\Indexer\Category\Product\Processor;
 use Magento\Framework\App\ObjectManager;
 use Magento\Framework\DataObject;
 use Magento\Framework\EntityManager\EntityManager;
-use Magento\Catalog\Model\Category as CategoryEntity;
+use Magento\Catalog\Setup\CategorySetup;
 
 /**
  * Resource model for category entity
@@ -92,6 +94,7 @@ class Category extends AbstractResource
      * @var Processor
      */
     private $indexerProcessor;
+
     /**
      * Category constructor.
      * @param \Magento\Eav\Model\Entity\Context $context
@@ -275,7 +278,7 @@ protected function _beforeSave(\Magento\Framework\DataObject $object)
             if ($object->getPosition() === null) {
                 $object->setPosition($this->_getMaxPosition($object->getPath()) + 1);
             }
-            $path = explode('/', $object->getPath());
+            $path = explode('/', (string)$object->getPath());
             $level = count($path)  - ($object->getId() ? 1 : 0);
             $toUpdateChild = array_diff($path, [$object->getId()]);
 
@@ -314,7 +317,7 @@ protected function _afterSave(\Magento\Framework\DataObject $object)
         /**
          * Add identifier for new category
          */
-        if (substr($object->getPath(), -1) == '/') {
+        if (substr((string)$object->getPath(), -1) == '/') {
             $object->setPath($object->getPath() . $object->getId());
             $this->_savePath($object);
         }
@@ -352,7 +355,7 @@ protected function _getMaxPosition($path)
     {
         $connection = $this->getConnection();
         $positionField = $connection->quoteIdentifier('position');
-        $level = count(explode('/', $path));
+        $level = count(explode('/', (string)$path));
         $bind = ['c_level' => $level, 'c_path' => $path . '/%'];
         $select = $connection->select()->from(
             $this->getTable('catalog_category_entity'),
@@ -717,7 +720,7 @@ public function getCategories($parent, $recursionLevel = 0, $sorted = false, $as
      */
     public function getParentCategories($category)
     {
-        $pathIds = array_reverse(explode(',', $category->getPathInStore()));
+        $pathIds = array_reverse(explode(',', (string)$category->getPathInStore()));
         /** @var \Magento\Catalog\Model\ResourceModel\Category\Collection $categories */
         $categories = $this->_categoryCollectionFactory->create();
         return $categories->setStore(
@@ -1134,4 +1137,44 @@ private function getAggregateCount()
         }
         return $this->aggregateCount;
     }
+
+    /**
+     * Get category with children.
+     *
+     * @param int $categoryId
+     * @return array
+     */
+    public function getCategoryWithChildren(int $categoryId): array
+    {
+        $connection = $this->getConnection();
+
+        $selectAttributeCode = $connection->select()
+            ->from(
+                ['eav_attribute' => $this->getTable('eav_attribute')],
+                ['attribute_id']
+            )->where('entity_type_id = ?', CategorySetup::CATEGORY_ENTITY_TYPE_ID)
+            ->where('attribute_code = ?', 'is_anchor')
+            ->limit(1);
+        $isAnchorAttributeCode = $connection->fetchOne($selectAttributeCode);
+        if (empty($isAnchorAttributeCode) || (int)$isAnchorAttributeCode <= 0) {
+            return [];
+        }
+
+        $select = $connection->select()
+            ->from(
+                ['cce' => $this->getTable('catalog_category_entity')],
+                ['entity_id', 'parent_id', 'path']
+            )->join(
+                ['cce_int' => $this->getTable('catalog_category_entity_int')],
+                'cce.entity_id = cce_int.entity_id',
+                ['is_anchor' => 'cce_int.value']
+            )->where(
+                'cce_int.attribute_id = ?',
+                $isAnchorAttributeCode
+            )->where(
+                "cce.path LIKE '%/{$categoryId}' OR cce.path LIKE '%/{$categoryId}/%'"
+            )->order('path');
+
+        return $connection->fetchAll($select);
+    }
 }
diff --git a/app/code/Magento/Catalog/Model/ResourceModel/Product/Collection.php b/app/code/Magento/Catalog/Model/ResourceModel/Product/Collection.php
@@ -3,6 +3,7 @@
  * Copyright © Magento, Inc. All rights reserved.
  * See COPYING.txt for license details.
  */
+declare(strict_types=1);
 
 namespace Magento\Catalog\Model\ResourceModel\Product;
 
@@ -22,6 +23,7 @@
 use Magento\Framework\Indexer\DimensionFactory;
 use Magento\Store\Model\Indexer\WebsiteDimensionProvider;
 use Magento\Store\Model\Store;
+use Magento\Catalog\Model\ResourceModel\Category;
 
 /**
  * Product collection
@@ -302,6 +304,11 @@ class Collection extends \Magento\Catalog\Model\ResourceModel\Collection\Abstrac
      */
     private $urlFinder;
 
+    /**
+     * @var Category
+     */
+    private $categoryResourceModel;
+
     /**
      * Collection constructor
      *
@@ -330,6 +337,7 @@ class Collection extends \Magento\Catalog\Model\ResourceModel\Collection\Abstrac
      * @param TableMaintainer|null $tableMaintainer
      * @param PriceTableResolver|null $priceTableResolver
      * @param DimensionFactory|null $dimensionFactory
+     * @param Category|null $categoryResourceModel
      *
      * @SuppressWarnings(PHPMD.ExcessiveParameterList)
      */
@@ -358,7 +366,8 @@ public function __construct(
         MetadataPool $metadataPool = null,
         TableMaintainer $tableMaintainer = null,
         PriceTableResolver $priceTableResolver = null,
-        DimensionFactory $dimensionFactory = null
+        DimensionFactory $dimensionFactory = null,
+        Category $categoryResourceModel = null
     ) {
         $this->moduleManager = $moduleManager;
         $this->_catalogProductFlatState = $catalogProductFlatState;
@@ -392,6 +401,8 @@ public function __construct(
         $this->priceTableResolver = $priceTableResolver ?: ObjectManager::getInstance()->get(PriceTableResolver::class);
         $this->dimensionFactory = $dimensionFactory
             ?: ObjectManager::getInstance()->get(DimensionFactory::class);
+        $this->categoryResourceModel = $categoryResourceModel ?: ObjectManager::getInstance()
+            ->get(Category::class);
     }
 
     /**
@@ -1673,7 +1684,11 @@ public function addFilterByRequiredOptions()
     public function setVisibility($visibility)
     {
         $this->_productLimitationFilters['visibility'] = $visibility;
-        $this->_applyProductLimitations();
+        if ($this->getStoreId() == Store::DEFAULT_STORE_ID) {
+            $this->addAttributeToFilter('visibility', $visibility);
+        } else {
+            $this->_applyProductLimitations();
+        }
 
         return $this;
     }
@@ -2053,12 +2068,13 @@ protected function _applyProductLimitations()
     protected function _applyZeroStoreProductLimitations()
     {
         $filters = $this->_productLimitationFilters;
+        $categories = $this->getChildrenCategories((int)$filters['category_id']);
 
         $conditions = [
             'cat_pro.product_id=e.entity_id',
             $this->getConnection()->quoteInto(
-                'cat_pro.category_id=?',
-                $filters['category_id']
+                'cat_pro.category_id IN (?)',
+                $categories
             ),
         ];
         $joinCond = join(' AND ', $conditions);
@@ -2079,6 +2095,39 @@ protected function _applyZeroStoreProductLimitations()
         return $this;
     }
 
+    /**
+     * Get children categories.
+     *
+     * @param int $categoryId
+     * @return array
+     */
+    private function getChildrenCategories(int $categoryId): array
+    {
+        $categoryIds[] = $categoryId;
+        $anchorCategory = [];
+
+        $categories = $this->categoryResourceModel->getCategoryWithChildren($categoryId);
+        if (empty($categories)) {
+            return $categoryIds;
+        }
+
+        $firstCategory = array_shift($categories);
+        if ($firstCategory['is_anchor'] == 1) {
+            $anchorCategory[] = (int)$firstCategory['entity_id'];
+            foreach ($categories as $category) {
+                if (in_array($category['parent_id'], $categoryIds)
+                    && in_array($category['parent_id'], $anchorCategory)) {
+                    $categoryIds[] = (int)$category['entity_id'];
+                    if ($category['is_anchor'] == 1) {
+                        $anchorCategory[] = (int)$category['entity_id'];
+                    }
+                }
+            }
+        }
+
+        return $categoryIds;
+    }
+
     /**
      * Add category ids to loaded items
      *
diff --git a/dev/tests/integration/testsuite/Magento/Catalog/Model/ResourceModel/Product/CollectionTest.php b/dev/tests/integration/testsuite/Magento/Catalog/Model/ResourceModel/Product/CollectionTest.php
@@ -3,10 +3,20 @@
  * Copyright © Magento, Inc. All rights reserved.
  * See COPYING.txt for license details.
  */
+declare(strict_types=1);
+
 namespace Magento\Catalog\Model\ResourceModel\Product;
 
+use Magento\Catalog\Model\Product\Visibility;
+use Magento\Framework\App\Area;
+use Magento\Framework\App\State;
+use Magento\Store\Model\Store;
+use Magento\TestFramework\Helper\Bootstrap;
+
 /**
  * Collection test
+ *
+ * @SuppressWarnings(PHPMD.CouplingBetweenObjects)
  */
 class CollectionTest extends \PHPUnit\Framework\TestCase
 {
@@ -31,15 +41,15 @@ class CollectionTest extends \PHPUnit\Framework\TestCase
      */
     protected function setUp()
     {
-        $this->collection = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+        $this->collection = Bootstrap::getObjectManager()->create(
             \Magento\Catalog\Model\ResourceModel\Product\Collection::class
         );
 
-        $this->processor = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+        $this->processor = Bootstrap::getObjectManager()->create(
             \Magento\Catalog\Model\Indexer\Product\Price\Processor::class
         );
 
-        $this->productRepository = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+        $this->productRepository = Bootstrap::getObjectManager()->create(
             \Magento\Catalog\Api\ProductRepositoryInterface::class
         );
     }
@@ -54,7 +64,7 @@ public function testAddPriceDataOnSchedule()
         $this->processor->getIndexer()->setScheduled(true);
         $this->assertTrue($this->processor->getIndexer()->isScheduled());
 
-        $productRepository = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()
+        $productRepository = Bootstrap::getObjectManager()
             ->create(\Magento\Catalog\Api\ProductRepositoryInterface::class);
         /** @var \Magento\Catalog\Api\Data\ProductInterface $product */
         $product = $productRepository->get('simple');
@@ -73,7 +83,7 @@ public function testAddPriceDataOnSchedule()
         //reindexing
         $this->processor->getIndexer()->reindexList([1]);
 
-        $this->collection = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+        $this->collection = Bootstrap::getObjectManager()->create(
             \Magento\Catalog\Model\ResourceModel\Product\Collection::class
         );
         $this->collection->addPriceData(0, 1);
@@ -89,6 +99,69 @@ public function testAddPriceDataOnSchedule()
         $this->processor->getIndexer()->setScheduled(false);
     }
 
+    /**
+     * @magentoDataFixture Magento/Catalog/_files/products.php
+     * @magentoAppIsolation enabled
+     * @magentoDbIsolation disabled
+     */
+    public function testSetVisibility()
+    {
+        $appState = Bootstrap::getObjectManager()
+            ->create(State::class);
+        $appState->setAreaCode(Area::AREA_CRONTAB);
+        $this->collection->setStoreId(Store::DEFAULT_STORE_ID);
+        $this->collection->setVisibility([Visibility::VISIBILITY_BOTH]);
+        $this->collection->load();
+        /** @var \Magento\Catalog\Api\Data\ProductInterface[] $product */
+        $items = $this->collection->getItems();
+        $this->assertCount(2, $items);
+    }
+
+    /**
+     * @magentoDataFixture Magento/Catalog/_files/category_product.php
+     * @magentoAppIsolation enabled
+     * @magentoDbIsolation disabled
+     */
+    public function testSetCategoryWithStoreFilter()
+    {
+        $appState = Bootstrap::getObjectManager()
+            ->create(State::class);
+        $appState->setAreaCode(Area::AREA_CRONTAB);
+
+        $category = \Magento\Framework\App\ObjectManager::getInstance()->get(
+            \Magento\Catalog\Model\Category::class
+        )->load(333);
+        $this->collection->addCategoryFilter($category)->addStoreFilter(1);
+        $this->collection->load();
+
+        $collectionStoreFilterAfter = Bootstrap::getObjectManager()->create(
+            \Magento\Catalog\Model\ResourceModel\Product\CollectionFactory::class
+        )->create();
+        $collectionStoreFilterAfter->addStoreFilter(1)->addCategoryFilter($category);
+        $collectionStoreFilterAfter->load();
+        $this->assertEquals($this->collection->getItems(), $collectionStoreFilterAfter->getItems());
+        $this->assertCount(1, $collectionStoreFilterAfter->getItems());
+    }
+
+    /**
+     * @magentoDataFixture Magento/Catalog/_files/categories.php
+     * @magentoAppIsolation enabled
+     * @magentoDbIsolation disabled
+     */
+    public function testSetCategoryFilter()
+    {
+        $appState = Bootstrap::getObjectManager()
+            ->create(State::class);
+        $appState->setAreaCode(Area::AREA_CRONTAB);
+
+        $category = \Magento\Framework\App\ObjectManager::getInstance()->get(
+            \Magento\Catalog\Model\Category::class
+        )->load(3);
+        $this->collection->addCategoryFilter($category);
+        $this->collection->load();
+        $this->assertEquals($this->collection->getSize(), 3);
+    }
+
     /**
      * @magentoDataFixture Magento/Catalog/_files/products.php
      * @magentoAppIsolation enabled
@@ -98,7 +171,7 @@ public function testAddPriceDataOnSave()
     {
         $this->processor->getIndexer()->setScheduled(false);
         $this->assertFalse($this->processor->getIndexer()->isScheduled());
-        $productRepository = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()
+        $productRepository = Bootstrap::getObjectManager()
             ->create(\Magento\Catalog\Api\ProductRepositoryInterface::class);
         /** @var \Magento\Catalog\Api\Data\ProductInterface $product */
         $product = $productRepository->get('simple');
@@ -184,7 +257,7 @@ public function testJoinTable()
         $productTable = $this->collection->getTable('catalog_product_entity');
         $urlRewriteTable = $this->collection->getTable('url_rewrite');
 
-        // phpcs:ignore
+        // phpcs:ignore Magento2.SQL.RawQuery
         $expected = 'SELECT `e`.*, `alias`.`request_path` FROM `' . $productTable . '` AS `e`'
             . ' LEFT JOIN `' . $urlRewriteTable . '` AS `alias` ON (alias.entity_id =e.entity_id)'
             . ' AND (alias.entity_type = \'product\')';
