Merge remote-tracking branch 'origin/MC-33288' into 2.4-develop-pr101

Author: zakdma

app/code/Magento/Catalog/Model/ResourceModel/Product/Relation.php

@@ -5,13 +5,38 @@
  */
 namespace Magento\Catalog\Model\ResourceModel\Product;
 
+use Magento\Framework\App\ObjectManager;
+use Magento\Framework\Model\ResourceModel\Db\AbstractDb;
+use Magento\Framework\Model\ResourceModel\Db\Context;
+use Magento\Framework\EntityManager\MetadataPool;
+use Magento\Catalog\Api\Data\ProductInterface;
+
 /**
  * Catalog Product Relations Resource model
  *
  * @author      Magento Core Team <core@magentocommerce.com>
  */
-class Relation extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
+class Relation extends AbstractDb
 {
+    /**
+     * @var MetadataPool
+     */
+    private $metadataPool;
+
+    /**
+     * @param Context $context
+     * @param string $connectionName
+     * @param MetadataPool $metadataPool
+     */
+    public function __construct(
+        Context $context,
+        $connectionName = null,
+        MetadataPool $metadataPool = null
+    ) {
+        parent::__construct($context, $connectionName);
+        $this->metadataPool = $metadataPool ?: ObjectManager::getInstance()->get(MetadataPool::class);
+    }
+
     /**
      * Initialize resource model and define main table
      *
@@ -109,4 +134,27 @@ public function removeRelations($parentId, $childIds)
         }
         return $this;
     }
+
+    /**
+     * Finds parent relations by given children ids.
+     *
+     * @param array $childrenIds Child products entity ids.
+     * @return array Parent products entity ids.
+     */
+    public function getRelationsByChildren(array $childrenIds): array
+    {
+        $connection = $this->getConnection();
+        $linkField = $this->metadataPool->getMetadata(ProductInterface::class)
+            ->getLinkField();
+        $select = $connection->select()
+            ->from(
+                ['cpe' => $this->getTable('catalog_product_entity')],
+                'entity_id'
+            )->join(
+                ['relation' => $this->getTable('catalog_product_relation')],
+                'relation.parent_id = cpe.' . $linkField
+            )->where('relation.child_id IN(?)', $childrenIds);
+
+        return $connection->fetchCol($select);
+    }
 }

dev/tests/integration/testsuite/Magento/ConfigurableProduct/Model/ResourceModel/Product/RelationTest.php

@@ -0,0 +1,86 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\ConfigurableProduct\Model\ResourceModel\Product;
+
+use Magento\Catalog\Model\ResourceModel\Product\Relation;
+use Magento\Framework\ObjectManagerInterface;
+use Magento\Catalog\Api\ProductRepositoryInterface;
+use Magento\Framework\Api\SearchCriteriaBuilder;
+use Magento\TestFramework\Helper\Bootstrap;
+use PHPUnit\Framework\TestCase;
+
+/**
+ * Tests Catalog Product Relation resource model.
+ *
+ * @see Relation
+ */
+class RelationTest extends TestCase
+{
+    /**
+     * @var ObjectManagerInterface
+     */
+    private $objectManager;
+
+    /**
+     * @var Relation
+     */
+    private $model;
+
+    /**
+     * @var ProductRepositoryInterface
+     */
+    private $productRepository;
+
+    /**
+     * @var SearchCriteriaBuilder
+     */
+    private $searchCriteriaBuilder;
+
+    /**
+     * @inheritdoc
+     */
+    protected function setUp(): void
+    {
+        $this->objectManager = Bootstrap::getObjectManager();
+        $this->model = $this->objectManager->get(Relation::class);
+        $this->productRepository = $this->objectManager->get(ProductRepositoryInterface::class);
+        $this->searchCriteriaBuilder = $this->objectManager->get(SearchCriteriaBuilder::class);
+    }
+
+    /**
+     * Tests that getRelationsByChildren will return parent products entity ids of child products entity ids.
+     *
+     * @magentoDataFixture Magento/ConfigurableProduct/_files/configurable_products.php
+     */
+    public function testGetRelationsByChildren(): void
+    {
+        // Find configurable products options
+        $productOptionSkus = ['simple_10', 'simple_20', 'simple_30', 'simple_40'];
+        $searchCriteria = $this->searchCriteriaBuilder->addFilter('sku', $productOptionSkus, 'in')
+            ->create();
+        $productOptions = $this->productRepository->getList($searchCriteria)
+            ->getItems();
+
+        $productOptionsIds = [];
+
+        foreach ($productOptions as $productOption) {
+            $productOptionsIds[] = $productOption->getId();
+        }
+
+        // Find configurable products
+        $searchCriteria = $this->searchCriteriaBuilder->addFilter('sku', ['configurable', 'configurable_12345'], 'in')
+            ->create();
+        $configurableProducts = $this->productRepository->getList($searchCriteria)
+            ->getItems();
+
+        // Assert there are configurable products ids in result of getRelationsByChildren method.
+        $result = $this->model->getRelationsByChildren($productOptionsIds);
+
+        foreach ($configurableProducts as $configurableProduct) {
+            $this->assertContains($configurableProduct->getId(), $result);
+        }
+    }
+}