Merge remote-tracking branch 'origin/MAGETWO-71536' into 2.3-PR-develop-1

Author: VladimirZaets

app/code/Magento/CatalogImportExport/Model/Indexer/Product/Flat/Plugin/Import.php

@@ -5,19 +5,30 @@
  */
 namespace Magento\CatalogImportExport\Model\Indexer\Product\Flat\Plugin;
 
+use Magento\Catalog\Model\Indexer\Product\Flat\State as FlatState;
+
 class Import
 {
+    /**
+     * @var \Magento\Catalog\Model\Indexer\Product\Flat\State
+     */
+    private $flatState;
+
     /**
      * @var \Magento\Catalog\Model\Indexer\Product\Flat\Processor
      */
     protected $_productFlatIndexerProcessor;
 
     /**
      * @param \Magento\Catalog\Model\Indexer\Product\Flat\Processor $productFlatIndexerProcessor
+     * @param \Magento\Catalog\Model\Indexer\Product\Flat\State $flatState
      */
-    public function __construct(\Magento\Catalog\Model\Indexer\Product\Flat\Processor $productFlatIndexerProcessor)
-    {
+    public function __construct(
+        \Magento\Catalog\Model\Indexer\Product\Flat\Processor $productFlatIndexerProcessor,
+        FlatState $flatState
+    ) {
         $this->_productFlatIndexerProcessor = $productFlatIndexerProcessor;
+        $this->flatState = $flatState;
     }
 
     /**
@@ -31,7 +42,10 @@ public function __construct(\Magento\Catalog\Model\Indexer\Product\Flat\Processo
      */
     public function afterImportSource(\Magento\ImportExport\Model\Import $subject, $import)
     {
-        $this->_productFlatIndexerProcessor->markIndexerAsInvalid();
+        if ($this->flatState->isFlatEnabled() && !$this->_productFlatIndexerProcessor->isIndexerScheduled()) {
+            $this->_productFlatIndexerProcessor->markIndexerAsInvalid();
+        }
+
         return $import;
     }
 }

app/code/Magento/CatalogImportExport/Test/Unit/Model/Indexer/Product/Flat/Plugin/ImportTest.php

@@ -5,25 +5,79 @@
  */
 namespace Magento\CatalogImportExport\Test\Unit\Model\Indexer\Product\Flat\Plugin;
 
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
+
 class ImportTest extends \PHPUnit\Framework\TestCase
 {
-    public function testAfterImportSource()
+    /**
+     * @var \Magento\Catalog\Model\Indexer\Product\Flat\Processor|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $processorMock;
+
+    /**
+     * @var \Magento\CatalogImportExport\Model\Indexer\Product\Flat\Plugin\Import
+     */
+    private $model;
+
+    /**
+     * @var \Magento\Catalog\Model\Indexer\Product\Flat\State|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $flatStateMock;
+
+    /**
+     * @var \Magento\ImportExport\Model\Import|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $subjectMock;
+
+    protected function setUp()
     {
-        /**
-         * @var \Magento\Catalog\Model\Indexer\Product\Flat\Processor|
-         *      \PHPUnit_Framework_MockObject_MockObject $processorMock
-         */
-        $processorMock = $this->createPartialMock(
-            \Magento\Catalog\Model\Indexer\Product\Flat\Processor::class,
-            ['markIndexerAsInvalid']
+        $this->processorMock = $this->getMockBuilder(\Magento\Catalog\Model\Indexer\Product\Flat\Processor::class)
+            ->disableOriginalConstructor()
+            ->setMethods(['markIndexerAsInvalid', 'isIndexerScheduled'])
+            ->getMock();
+
+        $this->flatStateMock = $this->getMockBuilder(\Magento\Catalog\Model\Indexer\Product\Flat\State::class)
+            ->disableOriginalConstructor()
+            ->setMethods(['isFlatEnabled'])
+            ->getMock();
+
+        $this->subjectMock = $this->getMockBuilder(\Magento\ImportExport\Model\Import::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+
+        $this->model = (new ObjectManager($this))->getObject(
+            \Magento\CatalogImportExport\Model\Indexer\Product\Flat\Plugin\Import::class,
+            [
+                'productFlatIndexerProcessor' => $this->processorMock,
+                'flatState' => $this->flatStateMock
+            ]
         );
+    }
 
-        $subjectMock = $this->createMock(\Magento\ImportExport\Model\Import::class);
-        $processorMock->expects($this->once())->method('markIndexerAsInvalid');
+    public function testAfterImportSourceWithFlatEnabledAndIndexerScheduledDisabled()
+    {
+        $this->flatStateMock->expects($this->once())->method('isFlatEnabled')->willReturn(true);
+        $this->processorMock->expects($this->once())->method('isIndexerScheduled')->willReturn(false);
+        $this->processorMock->expects($this->once())->method('markIndexerAsInvalid');
+        $someData = [1, 2, 3];
+        $this->assertEquals($someData, $this->model->afterImportSource($this->subjectMock, $someData));
+    }
 
+    public function testAfterImportSourceWithFlatDisabledAndIndexerScheduledDisabled()
+    {
+        $this->flatStateMock->expects($this->once())->method('isFlatEnabled')->willReturn(false);
+        $this->processorMock->expects($this->never())->method('isIndexerScheduled')->willReturn(false);
+        $this->processorMock->expects($this->never())->method('markIndexerAsInvalid');
         $someData = [1, 2, 3];
+        $this->assertEquals($someData, $this->model->afterImportSource($this->subjectMock, $someData));
+    }
 
-        $model = new \Magento\CatalogImportExport\Model\Indexer\Product\Flat\Plugin\Import($processorMock);
-        $this->assertEquals($someData, $model->afterImportSource($subjectMock, $someData));
+    public function testAfterImportSourceWithFlatEnabledAndIndexerScheduledEnabled()
+    {
+        $this->flatStateMock->expects($this->once())->method('isFlatEnabled')->willReturn(true);
+        $this->processorMock->expects($this->once())->method('isIndexerScheduled')->willReturn(true);
+        $this->processorMock->expects($this->never())->method('markIndexerAsInvalid');
+        $someData = [1, 2, 3];
+        $this->assertEquals($someData, $this->model->afterImportSource($this->subjectMock, $someData));
     }
 }