5
5
*/
6
6
namespace Magento \Catalog \Test \Unit \Model \Category \Link ;
7
7
8
+ use Magento \Catalog \Api \Data \CategoryLinkInterface ;
8
9
use Magento \Catalog \Api \Data \CategoryLinkInterfaceFactory ;
10
+ use Magento \Catalog \Api \Data \ProductExtensionInterface ;
9
11
use Magento \Catalog \Model \Category \Link \ReadHandler ;
10
12
use Magento \Catalog \Model \ResourceModel \Product \CategoryLink ;
11
13
use Magento \Framework \Api \DataObjectHelper ;
@@ -66,10 +68,14 @@ public function testExecute()
66
68
['category_id ' => 3 , 'position ' => 10 ],
67
69
['category_id ' => 4 , 'position ' => 20 ]
68
70
];
71
+
69
72
$ dtoCategoryLinks = [];
70
73
foreach ($ categoryLinks as $ key => $ categoryLink ) {
71
- $ dtoCategoryLinks [$ key ] = $ this ->getMockBuilder (\ Magento \ Catalog \ Api \ Data \ CategoryLinkInterface::class)
74
+ $ dtoCategoryLinks [$ key ] = $ this ->getMockBuilder (CategoryLinkInterface::class)
72
75
->getMockForAbstractClass ();
76
+ $ this ->dataObjectHelper ->expects (static ::at ($ key ))
77
+ ->method ('populateWithArray ' )
78
+ ->with ($ dtoCategoryLinks [$ key ], $ categoryLink , CategoryLinkInterface::class);
73
79
$ this ->categoryLinkFactory ->expects (static ::at ($ key ))
74
80
->method ('create ' )
75
81
->willReturn ($ dtoCategoryLinks [$ key ]);
@@ -80,7 +86,7 @@ public function testExecute()
80
86
->setMethods (['getExtensionAttributes ' , 'setExtensionAttributes ' ])
81
87
->getMock ();
82
88
83
- $ extensionAttributes = $ this ->getMockBuilder (\ Magento \ Catalog \ Api \ Data \ ProductExtensionInterface::class)
89
+ $ extensionAttributes = $ this ->getMockBuilder (ProductExtensionInterface::class)
84
90
->disableOriginalConstructor ()
85
91
->setMethods (['setCategoryLinks ' ])
86
92
->getMockForAbstractClass ();
@@ -110,7 +116,7 @@ public function testExecuteNullExtensionAttributes()
110
116
->setMethods (['getExtensionAttributes ' , 'setExtensionAttributes ' ])
111
117
->getMock ();
112
118
113
- $ extensionAttributes = $ this ->getMockBuilder (\ Magento \ Catalog \ Api \ Data \ ProductExtensionInterface::class)
119
+ $ extensionAttributes = $ this ->getMockBuilder (ProductExtensionInterface::class)
114
120
->disableOriginalConstructor ()
115
121
->setMethods (['setCategoryLinks ' ])
116
122
->getMockForAbstractClass ();
0 commit comments