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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\Setup\Test\Unit\Fixtures;
use Magento\Customer\Api\Data\GroupInterface;
use Magento\Customer\Api\Data\GroupInterfaceFactory;
use Magento\Customer\Api\GroupRepositoryInterface;
use Magento\Customer\Model\ResourceModel\Group\CollectionFactory;
use Magento\Setup\Fixtures\CustomerGroupsFixture;
/**
* Test Customer Groups generation
*/
class CustomerGroupsFixtureTest extends \PHPUnit\Framework\TestCase
{
/**
* @var \PHPUnit_Framework_MockObject_MockObject|\Magento\Setup\Fixtures\FixtureModel
*/
private $fixtureModelMock;
/**
* @var CollectionFactory
*/
private $groupCollectionFactoryMock;
/**
* @var GroupRepositoryInterface
*/
private $groupRepositoryMock;
/**
* @var GroupInterfaceFactory
*/
private $groupFactoryMock;
/**
* @var GroupInterface
*/
private $groupDataObjectMock;
/**
* @var \Magento\Setup\Fixtures\IndexersStatesApplyFixture
*/
private $model;
public function testExecute()
{
$this->fixtureModelMock = $this->getMockBuilder(\Magento\Setup\Fixtures\FixtureModel::class)
->disableOriginalConstructor()
->getMock();
//Mock repository for customer groups
$this->groupRepositoryMock = $this->getMockBuilder(GroupRepositoryInterface::class)
->disableOriginalConstructor()
->getMock();
//Mock for customer groups collection
$this->groupCollectionFactoryMock = $this->getMockBuilder(CollectionFactory::class)
->setMethods(['create', 'getSize'])
->disableOriginalConstructor()
->getMock();
$this->groupCollectionFactoryMock
->expects($this->once())
->method('create')
->willReturn($this->groupCollectionFactoryMock);
$this->groupCollectionFactoryMock
->expects($this->once())
->method('getSize')
->willReturn(0);
//Mock customer groups data object
$this->groupDataObjectMock = $this->getMockBuilder(GroupInterface::class)
->setMethods(['setCode', 'setTaxClassId', 'save'])
->disableOriginalConstructor()
->getMockForAbstractClass();
//Mock customer groups factory
$this->groupFactoryMock = $this->getMockBuilder(GroupInterfaceFactory::class)
->setMethods(['create'])
->disableOriginalConstructor()
->getMock();
$this->groupFactoryMock
->expects($this->once())
->method('create')
->willReturn($this->groupDataObjectMock);
$this->groupDataObjectMock
->expects($this->once())
->method('setCode')
->willReturn($this->groupDataObjectMock);
$this->groupDataObjectMock
->expects($this->once())
->method('setTaxClassId')
->willReturn($this->groupDataObjectMock);
$this->groupRepositoryMock
->expects($this->once())
->method('save')
->willReturn($this->groupDataObjectMock);
$this->fixtureModelMock
->expects($this->once())
->method('getValue')
->will($this->returnValue(1));
$this->model = new CustomerGroupsFixture(
$this->fixtureModelMock,
$this->groupCollectionFactoryMock,
$this->groupRepositoryMock,
$this->groupFactoryMock
);
$this->model->execute();
}
}