public function testBuild()
{
$documentId = 333;
$fieldName = 'fieldName';
$fieldValue = 'fieldValue';
$aggregations = ['aggregations'];
$document = $this->getMockBuilder('Magento\\Framework\\Api\\Search\\DocumentInterface')->disableOriginalConstructor()->getMockForAbstractClass();
$document->expects($this->once())->method('setCustomAttribute')->with($fieldName, $fieldValue);
$document->expects($this->once())->method('setId')->with($documentId);
$this->documentFactory->expects($this->once())->method('create')->willReturn($document);
/** @var SearchResultInterface|\PHPUnit_Framework_MockObject_MockObject $searchResult */
$searchResult = $this->getMockBuilder('Magento\\Framework\\Api\\Search\\SearchResultInterface')->disableOriginalConstructor()->getMockForAbstractClass();
$searchResult->expects($this->once())->method('setItems')->with([$document]);
$searchResult->expects($this->once())->method('setAggregations')->with($aggregations);
$this->searchResultFactory->expects($this->once())->method('create')->willReturn($searchResult);
$field = $this->getMockBuilder('Magento\\Framework\\Search\\DocumentField')->disableOriginalConstructor()->getMock();
$field->expects($this->once())->method('getName')->willReturn($fieldName);
$field->expects($this->once())->method('getValue')->willReturn($fieldValue);
$responseDocument = $this->getMockBuilder('Magento\\Framework\\Search\\Document')->disableOriginalConstructor()->getMock();
$responseDocument->expects($this->any())->method('getIterator')->willReturn(new \ArrayIterator([$field]));
$responseDocument->expects($this->once())->method('getId')->willReturn($documentId);
/** @var QueryResponse|\PHPUnit_Framework_MockObject_MockObject $response */
$response = $this->getMockBuilder('Magento\\Framework\\Search\\Response\\QueryResponse')->setMethods(['getIterator', 'getAggregations'])->disableOriginalConstructor()->getMockForAbstractClass();
$response->expects($this->any())->method('getIterator')->willReturn(new \ArrayIterator([$responseDocument]));
$response->expects($this->once())->method('getAggregations')->willReturn($aggregations);
$result = $this->model->build($response);
$this->assertInstanceOf('Magento\\Framework\\Api\\Search\\SearchResultInterface', $result);
}
/**
* @param ResponseInterface $response
* @return SearchResultInterface
*/
public function build(ResponseInterface $response)
{
/** @var \Magento\Framework\Api\Search\SearchResult $searchResult */
$searchResult = $this->searchResultFactory->create();
/** @var \Magento\Framework\Api\Search\DocumentInterface[] $documents */
$documents = [];
/** @var \Magento\Framework\Search\Document $responseDocument */
foreach ($response as $responseDocument) {
$document = $this->documentFactory->create();
/** @var \Magento\Framework\Search\DocumentField $field */
foreach ($responseDocument as $field) {
$document->setCustomAttribute($field->getName(), $field->getValue());
}
$document->setId($responseDocument->getId());
$documents[] = $document;
}
$searchResult->setItems($documents);
$searchResult->setAggregations($response->getAggregations());
$searchResult->setTotalCount(count($documents));
return $searchResult;
}
