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
85
86
87
88
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\Customer\Test\Unit\CustomerData;
use Magento\Customer\CustomerData\SectionPool;
class SectionPoolTest extends \PHPUnit\Framework\TestCase
{
/**
* @var \PHPUnit_Framework_MockObject_MockObject
*/
protected $objectManagerMock;
/**
* @var \PHPUnit_Framework_MockObject_MockObject
*/
protected $identifierMock;
/**
* @var array|null
*/
protected $sectionSourceMap;
/**
* @var SectionPool
*/
protected $model;
protected function setUp()
{
$this->objectManagerMock = $this->createMock(\Magento\Framework\ObjectManagerInterface::class);
$this->identifierMock = $this->createMock(\Magento\Customer\CustomerData\Section\Identifier::class);
$this->sectionSourceMap = ['section1' => 'b'];
$this->model = new SectionPool(
$this->objectManagerMock,
$this->identifierMock,
$this->sectionSourceMap
);
}
public function testGetSectionsDataAllSections()
{
$sectionNames = ['section1'];
$sectionsData = ['data1', 'data2'];
$allSectionsData = [
'section1' => [
'data1',
'data2'
]
];
$identifierResult = [1, 2, 3];
$sectionSourceMock = $this->createMock(\Magento\Customer\CustomerData\SectionSourceInterface::class);
$this->objectManagerMock->expects($this->once())
->method('get')
->with('b')
->willReturn($sectionSourceMock);
$sectionSourceMock->expects($this->once())->method('getSectionData')->willReturn($sectionsData);
$this->identifierMock->expects($this->once())
->method('markSections')
//check also default value for $forceTimestamp = false
->with($allSectionsData, $sectionNames, false)
->willReturn($identifierResult);
$modelResult = $this->model->getSectionsData($sectionNames);
$this->assertEquals($identifierResult, $modelResult);
}
/**
* @expectedException \Magento\Framework\Exception\LocalizedException
* @expectedExceptionMessage b doesn't extend \Magento\Customer\CustomerData\SectionSourceInterface
*/
public function testGetSectionsDataAllSectionsException()
{
$sectionNames = [];
$identifierResult = [1, 2, 3];
$this->objectManagerMock->expects($this->once())
->method('get')
->with('b')
->willReturn($this->model);
$modelResult = $this->model->getSectionsData($sectionNames);
$this->assertEquals($identifierResult, $modelResult);
}
}