MAGETWO-51951: SQL Error when cloning product with custom option with CatalogStaging enabled

Author: mslabko

app/code/Magento/Catalog/Model/Product/Copier.php

@@ -7,6 +7,8 @@
  */
 namespace Magento\Catalog\Model\Product;
 
+use Magento\Catalog\Api\Data\ProductInterface;
+
 class Copier
 {
     /**
@@ -24,6 +26,11 @@ class Copier
      */
     protected $productFactory;
 
+    /**
+     * @var \Magento\Framework\EntityManager\MetadataPool
+     */
+    protected $metadataPool;
+
     /**
      * @param CopyConstructorInterface $copyConstructor
      * @param \Magento\Catalog\Model\ProductFactory $productFactory
@@ -71,14 +78,18 @@ public function copy(\Magento\Catalog\Model\Product $product)
             } catch (\Magento\Framework\Exception\AlreadyExistsException $e) {
             }
         } while (!$isDuplicateSaved);
-
         $this->getOptionRepository()->duplicate($product, $duplicate);
-        $product->getResource()->duplicate($product->getEntityId(), $duplicate->getEntityId());
+        $metadata = $this->getMetadataPool()->getMetadata(ProductInterface::class);
+        $product->getResource()->duplicate(
+            $product->getData($metadata->getLinkField()),
+            $duplicate->getData($metadata->getLinkField())
+        );
         return $duplicate;
     }
 
     /**
      * @return Option\Repository
+     * @deprecated
      */
     private function getOptionRepository()
     {
@@ -88,4 +99,17 @@ private function getOptionRepository()
         }
         return $this->optionRepository;
     }
+
+    /**
+     * @return \Magento\Framework\EntityManager\MetadataPool
+     * @deprecated
+     */
+    private function getMetadataPool()
+    {
+        if (null === $this->metadataPool) {
+            $this->metadataPool = \Magento\Framework\App\ObjectManager::getInstance()
+                ->get('Magento\Framework\EntityManager\MetadataPool');
+        }
+        return $this->metadataPool;
+    }
 }

app/code/Magento/Catalog/Model/Product/Option/Repository.php

@@ -113,10 +113,11 @@ public function duplicate(
         \Magento\Catalog\Api\Data\ProductInterface $product,
         \Magento\Catalog\Api\Data\ProductInterface $duplicate
     ) {
+        $metadata = $this->getMetadataPool()->getMetadata(ProductInterface::class);
         return $this->optionResource->duplicate(
             $this->getOptionFactory()->create([]),
-            $product->getId(),
-            $duplicate->getId()
+            $product->getData($metadata->getLinkField()),
+            $duplicate->getData($metadata->getLinkField())
         );
     }
 
@@ -189,6 +190,7 @@ protected function markRemovedValues($newValues, $originalValues)
 
     /**
      * @return \Magento\Catalog\Model\Product\OptionFactory
+     * @deprecated
      */
     private function getOptionFactory()
     {
@@ -201,6 +203,7 @@ private function getOptionFactory()
 
     /**
      * @return \Magento\Catalog\Model\ResourceModel\Product\Option\CollectionFactory
+     * @deprecated
      */
     private function getCollectionFactory()
     {
@@ -213,6 +216,7 @@ private function getCollectionFactory()
 
     /**
      * @return \Magento\Framework\EntityManager\MetadataPool
+     * @deprecated
      */
     private function getMetadataPool()
     {

app/code/Magento/Catalog/Test/Unit/Model/Product/CopierTest.php

@@ -34,6 +34,11 @@ class CopierTest extends \PHPUnit_Framework_TestCase
      */
     protected $productMock;
 
+    /**
+     * @var \PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $metadata;
+
     protected function setUp()
     {
         $this->copyConstructorMock = $this->getMock('\Magento\Catalog\Model\Product\CopyConstructorInterface');
@@ -54,22 +59,33 @@ protected function setUp()
         $this->optionRepositoryMock;
         $this->productMock = $this->getMock('\Magento\Catalog\Model\Product', [], [], '', false);
         $this->productMock->expects($this->any())->method('getEntityId')->willReturn(1);
-        $this->productMock->expects($this->any())->method('getData')->will($this->returnValue('product data'));
 
+        $this->metadata = $this->getMockBuilder('Magento\Framework\EntityManager\EntityMetadata')
+            ->disableOriginalConstructor()
+            ->getMock();
+        $metadataPool = $this->getMockBuilder('Magento\Framework\EntityManager\MetadataPool')
+            ->disableOriginalConstructor()
+            ->getMock();
+        $metadataPool->expects($this->any())->method('getMetadata')->willReturn($this->metadata);
         $this->_model = new Copier(
             $this->copyConstructorMock,
             $this->productFactoryMock
         );
 
         $this->setProperties($this->_model, [
-            'optionRepository' => $this->optionRepositoryMock
+            'optionRepository' => $this->optionRepositoryMock,
+            'metadataPool' => $metadataPool,
         ]);
     }
 
     public function testCopy()
     {
         $this->productMock->expects($this->atLeastOnce())->method('getWebsiteIds');
         $this->productMock->expects($this->atLeastOnce())->method('getCategoryIds');
+        $this->productMock->expects($this->any())->method('getData')->willReturnMap([
+            ['', null, 'product data'],
+            ['linkField', null, '1'],
+        ]);
 
         $resourceMock = $this->getMock('\Magento\Catalog\Model\ResourceModel\Product', [], [], '', false);
         $this->productMock->expects($this->once())->method('getResource')->will($this->returnValue($resourceMock));
@@ -79,6 +95,7 @@ public function testCopy()
             [
                 '__wakeup',
                 'setData',
+                'getData',
                 'setIsDuplicate',
                 'setOriginalId',
                 'setStatus',
@@ -121,8 +138,12 @@ public function testCopy()
         $duplicateMock->expects($this->once())->method('getUrlKey')->willReturn('urk-key-1');
         $duplicateMock->expects($this->once())->method('setUrlKey')->with('urk-key-2');
         $duplicateMock->expects($this->once())->method('save');
-        $duplicateMock->expects($this->any())->method('getEntityId')->willReturn(2);
-        $this->optionRepositoryMock->expects($this->once())
+
+        $this->metadata->expects($this->any())->method('getLinkField')->willReturn('linkField');
+
+        $duplicateMock->expects($this->any())->method('getData')->willReturnMap([
+            ['linkField', null, '2'],
+        ]);        $this->optionRepositoryMock->expects($this->once())
             ->method('duplicate')
             ->with($this->productMock, $duplicateMock);
         $resourceMock->expects($this->once())->method('duplicate')->with(1, 2);