MAGETWO-28253: Downloadable Integration API

Yu Tang · Yu Tang · commit 8fa4fd63cd04 · 2015-04-19T19:12:48.000-05:00
- Fix build error
diff --git a/app/code/Magento/Downloadable/Test/Unit/Model/Plugin/AfterProductLoadTest.php b/app/code/Magento/Downloadable/Test/Unit/Model/Plugin/AfterProductLoadTest.php
@@ -49,7 +49,8 @@ protected function setUp()
         $this->productMock = $this->getMockBuilder('\Magento\Catalog\Model\Product')
             ->disableOriginalConstructor()
             ->getMock();
-        $this->productExtensionMock = $this->getMock('\Magento\Catalog\Api\Data\ProductExtensionInterface');
+        $this->productExtensionMock = $this->getMockBuilder('\Magento\Catalog\Api\Data\ProductExtension')
+            ->setMethods(['setDownloadableProductLinks', 'setDownloadableProductSamples'])->getMock();
     }
 
     public function testAfterLoad()
diff --git a/app/code/Magento/Downloadable/Test/Unit/Model/Plugin/AroundProductRepositorySaveTest.php b/app/code/Magento/Downloadable/Test/Unit/Model/Plugin/AroundProductRepositorySaveTest.php
@@ -71,8 +71,11 @@ protected function setUp()
         $this->model = new AroundProductRepositorySave(
             $this->linkRepositoryMock,
             $this->sampleRepositoryMock        );
-        $this->productExtensionMock = $this->getMock('Magento\Catalog\Api\Data\ProductExtensionInterface');
-        $this->existingProductExtensionMock = $this->getMock('Magento\Catalog\Api\Data\ProductExtensionInterface');
+        $this->productExtensionMock = $this->getMockBuilder('Magento\Catalog\Api\Data\ProductExtension')
+            ->setMethods(['getDownloadableProductLinks', 'getDownloadableProductSamples'])->getMock();
+        $this->existingProductExtensionMock = $this->getMockBuilder('Magento\Catalog\Api\Data\ProductExtension')
+            ->setMethods(['getDownloadableProductLinks', 'getDownloadableProductSamples'])
+            ->getMock();
     }
 
     public function testAroundSaveWhenProductIsSimple()

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,8 @@ protected function setUp()`
`49`	`49`	`$this->productMock = $this->getMockBuilder('\Magento\Catalog\Model\Product')`
`50`	`50`	`->disableOriginalConstructor()`
`51`	`51`	`->getMock();`
`52`		`- $this->productExtensionMock = $this->getMock('\Magento\Catalog\Api\Data\ProductExtensionInterface');`
	`52`	`+ $this->productExtensionMock = $this->getMockBuilder('\Magento\Catalog\Api\Data\ProductExtension')`
	`53`	`+ ->setMethods(['setDownloadableProductLinks', 'setDownloadableProductSamples'])->getMock();`
`53`	`54`	`}`
`54`	`55`
`55`	`56`	`public function testAfterLoad()`