diff --git a/app/code/Magento/Bundle/Test/Unit/Model/Product/TypeTest.php b/app/code/Magento/Bundle/Test/Unit/Model/Product/TypeTest.php
index 0188e74e4b7c1106f29342acde3f6f53173a5780..7456a045f376f9153349938831fb7ac1ccb879ea 100644
--- a/app/code/Magento/Bundle/Test/Unit/Model/Product/TypeTest.php
+++ b/app/code/Magento/Bundle/Test/Unit/Model/Product/TypeTest.php
@@ -2347,13 +2347,34 @@ class TypeTest extends \PHPUnit_Framework_TestCase
 
     /**
      * @param bool $isManageStock
-     * @return \Magento\CatalogInventory\Api\Data\StockItem|\PHPUnit_Framework_MockObject_MockObject
+     * @return \Magento\CatalogInventory\Api\Data\StockItemInterface|\PHPUnit_Framework_MockObject_MockObject
      */
     protected function getStockItem($isManageStock)
     {
-        $result = $this->getMockBuilder('\Magento\CatalogInventory\Api\Data\StockItem')
-            ->setMethods(['getManageStock'])
-            ->disableOriginalConstructor()
+        $result = $this->getMockBuilder('Magento\CatalogInventory\Api\Data\StockItemInterface')
+            ->setMethods(
+                [
+                    'getManageStock', 'getItemId', 'setItemId',
+                    'getProductId', 'setProductId', 'getWebsiteId',
+                    'setWebsiteId', 'getStockId', 'setStockId',
+                    'getQty', 'setQty', 'getIsInStock',
+                    'setIsInStock', 'getIsQtyDecimal', 'setIsQtyDecimal',
+                    'getShowDefaultNotificationMessage', 'getUseConfigMinQty', 'setUseConfigMinQty',
+                    'getMinQty', 'setMinQty', 'getUseConfigMinSaleQty',
+                    'setUseConfigMinSaleQty', 'getMinSaleQty', 'setMinSaleQty',
+                    'getUseConfigMaxSaleQty', 'setUseConfigMaxSaleQty', 'getMaxSaleQty',
+                    'setMaxSaleQty', 'getUseConfigBackorders', 'setUseConfigBackorders',
+                    'getBackorders', 'setBackorders', 'getUseConfigNotifyStockQty',
+                    'setUseConfigNotifyStockQty', 'getNotifyStockQty', 'setNotifyStockQty',
+                    'getUseConfigQtyIncrements', 'setUseConfigQtyIncrements', 'getQtyIncrements',
+                    'setQtyIncrements', 'getUseConfigEnableQtyInc', 'setUseConfigEnableQtyInc',
+                    'getEnableQtyIncrements', 'setEnableQtyIncrements', 'getUseConfigManageStock',
+                    'setUseConfigManageStock', 'setManageStock', 'getLowStockDate',
+                    'setLowStockDate', 'getIsDecimalDivided', 'setIsDecimalDivided',
+                    'getStockStatusChangedAuto', 'setStockStatusChangedAuto', 'getExtensionAttributes',
+                    'setExtensionAttributes'
+                ]
+            )
             ->getMock();
         $result->method('getManageStock')
             ->willReturn($isManageStock);