55 */
66namespace Magento \CatalogUrlRewrite \Test \Unit \Model \Map ;
77
8- use Magento \Catalog \Model \ResourceModel \Category as CategoryResource ;
8+ use Magento \Catalog \Model \ResourceModel \CategoryFactory ;
9+ use Magento \Catalog \Model \ResourceModel \Category ;
910use Magento \Framework \DB \Select ;
1011use Magento \Catalog \Model \CategoryRepository ;
1112use Magento \Catalog \Api \Data \CategoryInterface ;
@@ -21,7 +22,10 @@ class DataCategoryHashMapTest extends \PHPUnit_Framework_TestCase
2122 /** @var CategoryRepository|\PHPUnit_Framework_MockObject_MockObject */
2223 private $ categoryRepository ;
2324
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 */
2529 private $ categoryResource ;
2630
2731 /** @var DataCategoryHashMap|\PHPUnit_Framework_MockObject_MockObject */
@@ -30,19 +34,24 @@ class DataCategoryHashMapTest extends \PHPUnit_Framework_TestCase
3034 protected function setUp ()
3135 {
3236 $ this ->categoryRepository = $ this ->getMock (CategoryRepository::class, [], [], '' , false );
37+ $ this ->categoryResourceFactory = $ this ->getMock (CategoryFactory::class, ['create ' ], [], '' , false );
3338 $ this ->categoryResource = $ this ->getMock (
34- CategoryResource ::class,
39+ Category ::class,
3540 ['getConnection ' , 'getEntityTable ' ],
3641 [],
3742 '' ,
3843 false
3944 );
4045
46+ $ this ->categoryResourceFactory ->expects ($ this ->any ())
47+ ->method ('create ' )
48+ ->willReturn ($ this ->categoryResource );
49+
4150 $ this ->model = (new ObjectManager ($ this ))->getObject (
4251 DataCategoryHashMap::class,
4352 [
4453 'categoryRepository ' => $ this ->categoryRepository ,
45- 'categoryResource ' => $ this ->categoryResource
54+ 'categoryResourceFactory ' => $ this ->categoryResourceFactory
4655 ]
4756 );
4857 }
0 commit comments