MAGETWO-54257: "Stock Status" attribute is not changed for Variation Simple Product

Author: mslabko

dev/tests/integration/testsuite/Magento/ConfigurableProduct/Model/Product/VariationHandlerTest.php

@@ -8,6 +8,8 @@
 
 namespace Magento\ConfigurableProduct\Model\Product;
 
+use Magento\TestFramework\Helper\Bootstrap;
+
 /**
  * @magentoAppIsolation enabled
  * @magentoDataFixture Magento/ConfigurableProduct/_files/product_configurable.php
@@ -26,19 +28,22 @@ class VariationHandlerTest extends \PHPUnit_Framework_TestCase
      */
     protected $_product;
 
+    /** @var \Magento\CatalogInventory\Api\StockRegistryInterface */
+    protected $stockRegistry;
+
     protected function setUp()
     {
-        $this->_product = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+        $this->_product = Bootstrap::getObjectManager()->create(
             'Magento\Catalog\Model\Product'
         );
         $this->_product->load(1);
-        // fixture
 
-        $this->_model = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+        $this->_model = Bootstrap::getObjectManager()->create(
             'Magento\ConfigurableProduct\Model\Product\VariationHandler'
         );
         // prevent fatal errors by assigning proper "singleton" of type instance to the product
         $this->_product->setTypeInstance($this->_model);
+        $this->stockRegistry = Bootstrap::getObjectManager()->get('Magento\CatalogInventory\Api\StockRegistryInterface');
     }
 
     /**
@@ -52,15 +57,18 @@ public function testGenerateSimpleProducts($productsData)
         $generatedProducts = $this->_model->generateSimpleProducts($this->_product, $productsData);
         $this->assertEquals(3, count($generatedProducts));
         foreach ($generatedProducts as $productId) {
+            $stockItem = $this->stockRegistry->getStockItem($productId);
             /** @var $product \Magento\Catalog\Model\Product */
-            $product = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
+            $product = Bootstrap::getObjectManager()->create(
                 'Magento\Catalog\Model\Product'
             );
             $product->load($productId);
             $this->assertNotNull($product->getName());
             $this->assertNotNull($product->getSku());
             $this->assertNotNull($product->getPrice());
             $this->assertNotNull($product->getWeight());
+            $this->assertEquals('1', $stockItem->getIsInStock());
+
         }
     }
 
@@ -71,14 +79,12 @@ public function testGenerateSimpleProducts($productsData)
      */
     public function testGenerateSimpleProductsWithPartialData($productsData)
     {
-        $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();
-        /** @var \Magento\CatalogInventory\Api\StockRegistryInterface $stockRegistry */
-        $stockRegistry = $objectManager->get('Magento\CatalogInventory\Api\StockRegistryInterface');
         $this->_product->setNewVariationsAttributeSetId(4);
         $generatedProducts = $this->_model->generateSimpleProducts($this->_product, $productsData);
+        $parentStockItem = $this->stockRegistry->getStockItem($this->_product->getId());
         foreach ($generatedProducts as $productId) {
-            $stockItem = $stockRegistry->getStockItem($productId);
-            $this->assertEquals('0', $stockItem->getManageStock());
+            $stockItem = $this->stockRegistry->getStockItem($productId);
+            $this->assertEquals($parentStockItem->getManageStock(), $stockItem->getManageStock());
             $this->assertEquals('1', $stockItem->getIsInStock());
         }
     }