1818use Magento \Framework \TestFramework \Unit \Helper \ObjectManager ;
1919use Magento \MediaStorage \Helper \File \Storage \Database ;
2020use Magento \MediaStorage \Model \File \Storage ;
21+ use Magento \MediaStorage \Model \File \Storage \Database as StorageDatabase ;
2122use Magento \MediaStorage \Model \File \Storage \DatabaseFactory ;
2223use Magento \MediaStorage \Model \File \Storage \File ;
2324use PHPUnit \Framework \Attributes \DataProvider ;
@@ -107,7 +108,7 @@ public static function checkDbUsageDataProvider()
107108
108109 public function testGetStorageDatabaseModel ()
109110 {
110- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
111+ $ dbModelMock = $ this ->createMock (Database::class);
111112 $ this ->dbStorageFactoryMock ->expects ($ this ->once ())
112113 ->method ('create ' )
113114 ->willReturn ($ dbModelMock );
@@ -122,7 +123,10 @@ public function testGetStorageFileModel()
122123
123124 public function testGetResourceStorageModel ()
124125 {
125- $ dbModelMock = $ this ->createMock (\Magento \MediaStorage \Model \File \Storage \Database::class);
126+ $ dbModelMock = $ this ->createPartialMockWithReflection (
127+ StorageDatabase::class,
128+ ['getResource ' ]
129+ );
126130 $ this ->dbStorageFactoryMock ->expects ($ this ->once ())
127131 ->method ('create ' )
128132 ->willReturn ($ dbModelMock );
@@ -146,7 +150,7 @@ public function testSaveFile($storage, $callNum)
146150 ->method ('getValue ' )
147151 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
148152 ->willReturn ($ storage );
149- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
153+ $ dbModelMock = $ this ->createMock (Database::class);
150154 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
151155 ->method ('create ' )
152156 ->willReturn ($ dbModelMock );
@@ -179,7 +183,7 @@ public function testRenameFile($storage, $callNum)
179183 ->method ('getValue ' )
180184 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
181185 ->willReturn ($ storage );
182- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
186+ $ dbModelMock = $ this ->createMock (Database::class);
183187 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
184188 ->method ('create ' )
185189 ->willReturn ($ dbModelMock );
@@ -201,7 +205,7 @@ public function testCopyFile($storage, $callNum)
201205 ->method ('getValue ' )
202206 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
203207 ->willReturn ($ storage );
204- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
208+ $ dbModelMock = $ this ->createMock (Database::class);
205209 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
206210 ->method ('create ' )
207211 ->willReturn ($ dbModelMock );
@@ -224,7 +228,7 @@ public function testFileExists($storage, $callNum, $expected)
224228 ->method ('getValue ' )
225229 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
226230 ->willReturn ($ storage );
227- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
231+ $ dbModelMock = $ this ->createMock (Database::class);
228232 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
229233 ->method ('create ' )
230234 ->willReturn ($ dbModelMock );
@@ -259,7 +263,7 @@ public function testGetUniqueFilename($storage, $callNum, $expected)
259263 ->method ('getValue ' )
260264 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
261265 ->willReturn ($ storage );
262- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
266+ $ dbModelMock = $ this ->createMock (Database::class);
263267 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
264268 ->method ('create ' )
265269 ->willReturn ($ dbModelMock );
@@ -300,7 +304,10 @@ public function testSaveFileToFileSystem($expected, $storage, $callNum, $id = 0,
300304 ->method ('getValue ' )
301305 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
302306 ->willReturn ($ storage );
303- $ dbModelMock = $ this ->createMock (\Magento \MediaStorage \Model \File \Storage \Database::class);
307+ $ dbModelMock = $ this ->createPartialMock (
308+ StorageDatabase::class,
309+ ['loadByFilename ' , 'getId ' , 'getData ' ]
310+ );
304311 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
305312 ->method ('create ' )
306313 ->willReturn ($ dbModelMock );
@@ -357,7 +364,10 @@ public function testDeleteFolder($storage, $callNum)
357364 ->method ('getValue ' )
358365 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
359366 ->willReturn ($ storage );
360- $ dbModelMock = $ this ->createMock (\Magento \MediaStorage \Model \File \Storage \Database::class);
367+ $ dbModelMock = $ this ->createPartialMockWithReflection (
368+ StorageDatabase::class,
369+ ['getResource ' ]
370+ );
361371 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
362372 ->method ('create ' )
363373 ->willReturn ($ dbModelMock );
@@ -383,7 +393,7 @@ public function testDeleteFile($storage, $callNum)
383393 ->method ('getValue ' )
384394 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
385395 ->willReturn ($ storage );
386- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
396+ $ dbModelMock = $ this ->createMock (Database::class);
387397 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
388398 ->method ('create ' )
389399 ->willReturn ($ dbModelMock );
@@ -408,7 +418,7 @@ public function testSaveUploadedFile($result, $expected, $expectedFullPath, $sto
408418 ->method ('getValue ' )
409419 ->with (Storage::XML_PATH_STORAGE_MEDIA , 'default ' )
410420 ->willReturn ($ storage );
411- $ dbModelMock = $ this ->createMock (\ Magento \ MediaStorage \ Model \ File \ Storage \ Database::class);
421+ $ dbModelMock = $ this ->createMock (Database::class);
412422 $ this ->dbStorageFactoryMock ->expects ($ this ->exactly ($ callNum ))
413423 ->method ('create ' )
414424 ->willReturn ($ dbModelMock );
0 commit comments