1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\Framework\Search\Test\Unit;
use Magento\Framework\Api\Search\SearchResultInterface;
use Magento\Framework\Search\Response\QueryResponse;
use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
class SearchResponseBuilderTest extends \PHPUnit\Framework\TestCase
{
/**
* @var \Magento\Framework\Search\SearchResponseBuilder
*/
private $model;
/**
* @var \Magento\Framework\Api\Search\SearchResultFactory|\PHPUnit_Framework_MockObject_MockObject
*/
private $searchResultFactory;
/**
* @var \Magento\Framework\Api\Search\DocumentFactory|\PHPUnit_Framework_MockObject_MockObject
*/
private $documentFactory;
protected function setUp()
{
$this->searchResultFactory = $this->getMockBuilder(\Magento\Framework\Api\Search\SearchResultFactory::class)
->disableOriginalConstructor()
->getMock();
$this->documentFactory = $this->getMockBuilder(\Magento\Framework\Api\Search\DocumentFactory::class)
->disableOriginalConstructor()
->getMock();
$this->model = (new ObjectManager($this))->getObject(
\Magento\Framework\Search\SearchResponseBuilder::class,
['searchResultFactory' => $this->searchResultFactory]
);
}
public function testBuild()
{
$aggregations = ['aggregations'];
$document = $this->getMockBuilder(\Magento\Framework\Api\Search\DocumentInterface::class)
->disableOriginalConstructor()
->getMockForAbstractClass();
/** @var SearchResultInterface|\PHPUnit_Framework_MockObject_MockObject $searchResult */
$searchResult = $this->getMockBuilder(\Magento\Framework\Api\Search\SearchResultInterface::class)
->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);
/** @var QueryResponse|\PHPUnit_Framework_MockObject_MockObject $response */
$response = $this->getMockBuilder(\Magento\Framework\Search\Response\QueryResponse::class)
->setMethods(['getIterator', 'getAggregations'])
->disableOriginalConstructor()
->getMockForAbstractClass();
$response->expects($this->any())
->method('getIterator')
->willReturn(new \ArrayIterator([$document]));
$response->expects($this->once())
->method('getAggregations')
->willReturn($aggregations);
$result = $this->model->build($response);
$this->assertInstanceOf(\Magento\Framework\Api\Search\SearchResultInterface::class, $result);
}
}