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; }