MAGETWO-92402: Write tests for price indexer parallelization

Author: duhon

dev/tests/integration/framework/Magento/TestFramework/Annotation/IndexerDimensionMode.php

@@ -111,9 +111,11 @@ public function startTest(TestCase $test)
             $this->fail("Invalid @magentoDbIsolation declaration: $dbIsolation[0]", $test);
         }
 
-        if ($annotations[0] == Processor::INDEXER_ID) {
+        list($indexerType, $indexerMode) = explode(' ', $annotations[0]);
+
+        if ($indexerType == Processor::INDEXER_ID) {
             $this->isDimensionMode = true;
-            $this->setDimensionMode(DimensionModeConfiguration::DIMENSION_WEBSITE_AND_CUSTOMER_GROUP, $test);
+            $this->setDimensionMode($indexerMode, $test);
         }
     }
 

dev/tests/integration/testsuite/Magento/Catalog/Model/Product/Type/PriceWithDimensionTest.php

@@ -6,6 +6,14 @@
 namespace Magento\Catalog\Model\Product\Type;
 
 use Magento\Catalog\Model\Product;
+use Magento\Catalog\Model\ProductRepository;
+use Magento\Framework\App\ResourceConnection;
+use Magento\Framework\DataObject;
+use Magento\Catalog\Model\Indexer\Product\Price\PriceTableResolver;
+use Magento\Framework\Indexer\DimensionFactory;
+use Magento\Store\Model\Indexer\WebsiteDimensionProvider;
+use Magento\Customer\Model\Indexer\CustomerGroupDimensionProvider;
+use Magento\TestFramework\Helper\Bootstrap;
 
 /**
  * @magentoDbIsolation disabled
@@ -22,20 +30,49 @@ class PriceWithDimensionTest extends \PHPUnit\Framework\TestCase
 
     protected function setUp()
     {
-        $this->_model = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+        $this->_model = Bootstrap::getObjectManager()->create(
             \Magento\Catalog\Model\Product\Type\Price::class
         );
     }
 
+    public function testGetPriceFromIndexer()
+    {
+        /** @var PriceTableResolver $tableResolver */
+        $tableResolver = Bootstrap::getObjectManager()->create(PriceTableResolver::class);
+
+        /** @var ResourceConnection $resourceConnection */
+        $resourceConnection = Bootstrap::getObjectManager()->create(ResourceConnection::class);
+
+        /** @var DimensionFactory $dimensionFactory */
+        $dimensionFactory = Bootstrap::getObjectManager()->create(DimensionFactory::class);
+        $dimension = [
+            $dimensionFactory->create(CustomerGroupDimensionProvider::DIMENSION_NAME, (string)0),
+            $dimensionFactory->create(WebsiteDimensionProvider::DIMENSION_NAME, (string)1)
+        ];
+        $connection = $resourceConnection->getConnection();
+        $priceTable = $connection->getTableName(
+            $tableResolver->resolve('catalog_product_index_price', $dimension)
+        );
+
+        $select = $connection->select()->from($priceTable)->where('entity_id = 1');
+
+        $return = $connection->fetchAll($select);
+
+        $this->assertEquals('10', $return[0]['price']);
+        $this->assertEquals('10', $return[0]['final_price']);
+        $this->assertEquals('19', $return[0]['min_price']);
+        $this->assertEquals('19', $return[0]['max_price']);
+    }
+
     public function testGetPrice()
     {
-        $this->assertEquals('test', $this->_model->getPrice(new \Magento\Framework\DataObject(['price' => 'test'])));
+        $this->assertEquals('test', $this->_model->getPrice(new DataObject(['price' => 'test'])));
     }
 
     public function testGetFinalPrice()
     {
-        $repository = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
-            \Magento\Catalog\Model\ProductRepository::class
+        $repository = Bootstrap::getObjectManager()->create(
+            ProductRepository::class
         );
         $product = $repository->get('simple');
         // fixture
@@ -58,8 +95,8 @@ public function testGetFinalPrice()
 
     public function testGetFormatedPrice()
     {
-        $repository = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
-            \Magento\Catalog\Model\ProductRepository::class
+        $repository = Bootstrap::getObjectManager()->create(
+            ProductRepository::class
         );
         $product = $repository->get('simple');
         // fixture
@@ -93,7 +130,7 @@ public function testIsTierPriceFixed()
      * Build buy request based on product custom options
      *
      * @param Product $product
-     * @return \Magento\Framework\DataObject
+     * @return DataObject
      */
     private function prepareBuyRequest(Product $product)
     {
@@ -114,6 +151,6 @@ private function prepareBuyRequest(Product $product)
             $options[$option->getId()] = $value;
         }
 
-        return new \Magento\Framework\DataObject(['qty' => 1, 'options' => $options]);
+        return new DataObject(['qty' => 1, 'options' => $options]);
     }
 }