MAGETWO-59130: [Github] REST API with searchCriteria does not return category_ids #6127

Author: irenelagno

app/code/Magento/Catalog/Model/ProductRepository.php

@@ -640,6 +640,7 @@ public function getList(\Magento\Framework\Api\SearchCriteriaInterface $searchCr
 
         $collection->load();
 
+        $collection->addCategoryIds();
         $searchResult = $this->searchResultsFactory->create();
         $searchResult->setSearchCriteria($searchCriteria);
         $searchResult->setItems($collection->getItems());

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

@@ -715,6 +715,7 @@ public function testGetList()
             ->method('process')
             ->with($searchCriteriaMock, $collectionMock);
         $collectionMock->expects($this->once())->method('load');
+        $collectionMock->expects($this->once())->method('addCategoryIds');
         $collectionMock->expects($this->once())->method('getItems')->willReturn([$itemsMock]);
         $collectionMock->expects($this->once())->method('getSize')->willReturn(128);
         $searchResultsMock = $this->getMock(

dev/tests/api-functional/testsuite/Magento/Catalog/Api/ProductRepositoryInterfaceTest.php

@@ -713,6 +713,16 @@ public function testGetList()
 
         $this->assertNotNull($response['items'][0]['sku']);
         $this->assertEquals('simple', $response['items'][0]['sku']);
+        $isCategoryInformationExist = false;
+        $index = null;
+        foreach($response['items'][0]['custom_attributes'] as $key => $customAttribute) {
+            if ($customAttribute['attribute_code'] == 'category_ids') {
+                $isCategoryInformationExist = true;
+                $index = $key;
+            }
+        }
+        $this->assertTrue($isCategoryInformationExist, 'Category information wasn\'t set');
+        $this->assertEquals([2], $response['items'][0]['custom_attributes'][$index]['value']);
     }
 
     /**