MAGETWO-32410: Grouped Product Integration API

 - fixes from code review

Author: Robert He

app/code/Magento/Catalog/Test/Unit/Model/ProductTest.php

@@ -766,15 +766,6 @@ public function testGetProductLinks()
         $inputRelatedLink->setData("type", "simple");
         $inputRelatedLink->setPosition(0);
 
-        $customData = ["attribute_code" => "qty", "value" => 1];
-        $inputGroupLink = $this->objectManagerHelper->getObject('Magento\Catalog\Model\ProductLink\Link');
-        $inputGroupLink->setProductSku("Simple Product 1");
-        $inputGroupLink->setLinkType("associated");
-        $inputGroupLink->setData("sku", "Simple Product 2");
-        $inputGroupLink->setData("type", "simple");
-        $inputGroupLink->setPosition(0);
-        $inputGroupLink["custom_attributes"] = [$customData];
-
         $outputRelatedLink = $this->objectManagerHelper->getObject('Magento\Catalog\Model\ProductLink\Link');
         $outputRelatedLink->setProductSku("Simple Product 1");
         $outputRelatedLink->setLinkType("related");

app/code/Magento/GroupedProduct/Test/Unit/Model/ProductTest.php

@@ -363,7 +363,11 @@ public function testGetProductLinks()
         $outputRelatedLink->setPosition(0);
 
         $groupExtension = $this->objectManagerHelper->getObject('Magento\Catalog\Api\Data\ProductLinkExtension');
-        $groupExtension->setQty(1);
+        $reflectionOfUser = new \ReflectionClass('Magento\Catalog\Api\Data\ProductLinkExtension');
+        $method = $reflectionOfUser->getMethod('setData');
+        $method->setAccessible(true);
+        $method->invokeArgs($groupExtension, array('qty', 1));
+
         $outputGroupLink = $this->objectManagerHelper->getObject('Magento\Catalog\Model\ProductLink\Link');
         $outputGroupLink->setProductSku("Simple Product 1");
         $outputGroupLink->setLinkType("associated");
@@ -409,7 +413,6 @@ public function testGetProductLinks()
 
         $extension = $this->objectManagerHelper->getObject('Magento\Catalog\Api\Data\ProductLinkExtension');
         $productLink2->setExtensionAttributes($extension);
-        $productLink2->setExtensionAttributes($extension);
 
         $links = $this->model->getProductLinks();
         $this->assertEquals($links, $expectedOutput);