5
5
*/
6
6
namespace Magento \CatalogUrlRewrite \Test \Unit \Model \Map ;
7
7
8
- use Magento \Catalog \Model \ResourceModel \Category as CategoryResource ;
8
+ use Magento \Catalog \Model \ResourceModel \CategoryFactory ;
9
+ use Magento \Catalog \Model \ResourceModel \Category ;
9
10
use Magento \Framework \DB \Select ;
10
11
use Magento \Catalog \Model \CategoryRepository ;
11
12
use Magento \Catalog \Api \Data \CategoryInterface ;
@@ -21,7 +22,10 @@ class DataCategoryHashMapTest extends \PHPUnit_Framework_TestCase
21
22
/** @var CategoryRepository|\PHPUnit_Framework_MockObject_MockObject */
22
23
private $ categoryRepository ;
23
24
24
- /** @var CategoryResource|\PHPUnit_Framework_MockObject_MockObject */
25
+ /** @var CategoryResourceFactory|\PHPUnit_Framework_MockObject_MockObject */
26
+ private $ categoryResourceFactory ;
27
+
28
+ /** @var Category|\PHPUnit_Framework_MockObject_MockObject */
25
29
private $ categoryResource ;
26
30
27
31
/** @var DataCategoryHashMap|\PHPUnit_Framework_MockObject_MockObject */
@@ -30,19 +34,24 @@ class DataCategoryHashMapTest extends \PHPUnit_Framework_TestCase
30
34
protected function setUp ()
31
35
{
32
36
$ this ->categoryRepository = $ this ->getMock (CategoryRepository::class, [], [], '' , false );
37
+ $ this ->categoryResourceFactory = $ this ->getMock (CategoryFactory::class, ['create ' ], [], '' , false );
33
38
$ this ->categoryResource = $ this ->getMock (
34
- CategoryResource ::class,
39
+ Category ::class,
35
40
['getConnection ' , 'getEntityTable ' ],
36
41
[],
37
42
'' ,
38
43
false
39
44
);
40
45
46
+ $ this ->categoryResourceFactory ->expects ($ this ->any ())
47
+ ->method ('create ' )
48
+ ->willReturn ($ this ->categoryResource );
49
+
41
50
$ this ->model = (new ObjectManager ($ this ))->getObject (
42
51
DataCategoryHashMap::class,
43
52
[
44
53
'categoryRepository ' => $ this ->categoryRepository ,
45
- 'categoryResource ' => $ this ->categoryResource
54
+ 'categoryResourceFactory ' => $ this ->categoryResourceFactory
46
55
]
47
56
);
48
57
}
0 commit comments