9f9426697c
The global function wfWikiID() is deprecated since 1.35 and it's usages should be replaced with WikiMap::getCurrentWikiId(). Bug: T298059 Change-Id: I22d96b7aec17323d15a9bc401d4511ad2ee14165
140 lines
3.4 KiB
PHP
140 lines
3.4 KiB
PHP
<?php
|
|
|
|
/**
|
|
* @covers MigrateFileRepoLayout
|
|
*/
|
|
class MigrateFileRepoLayoutTest extends MediaWikiIntegrationTestCase {
|
|
protected $tmpPrefix;
|
|
protected $migratorMock;
|
|
protected $tmpFilepath;
|
|
protected $text = 'testing';
|
|
|
|
protected function setUp(): void {
|
|
parent::setUp();
|
|
|
|
$filename = 'Foo.png';
|
|
|
|
$this->tmpPrefix = $this->getNewTempDirectory();
|
|
|
|
$backend = new FSFileBackend( [
|
|
'name' => 'local-migratefilerepolayouttest',
|
|
'wikiId' => WikiMap::getCurrentWikiId(),
|
|
'containerPaths' => [
|
|
'migratefilerepolayouttest-original' => "{$this->tmpPrefix}-original",
|
|
'migratefilerepolayouttest-public' => "{$this->tmpPrefix}-public",
|
|
'migratefilerepolayouttest-thumb' => "{$this->tmpPrefix}-thumb",
|
|
'migratefilerepolayouttest-temp' => "{$this->tmpPrefix}-temp",
|
|
'migratefilerepolayouttest-deleted' => "{$this->tmpPrefix}-deleted",
|
|
]
|
|
] );
|
|
|
|
$dbMock = $this->getMockBuilder( Wikimedia\Rdbms\IDatabase::class )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$imageRow = (object)[
|
|
'img_name' => $filename,
|
|
'img_sha1' => sha1( $this->text ),
|
|
];
|
|
|
|
$dbMock->method( 'select' )
|
|
->will( $this->onConsecutiveCalls(
|
|
new FakeResultWrapper( [ $imageRow ] ), // image
|
|
new FakeResultWrapper( [] ), // image
|
|
new FakeResultWrapper( [] ) // filearchive
|
|
) );
|
|
|
|
$repoMock = $this->getMockBuilder( LocalRepo::class )
|
|
->onlyMethods( [ 'getPrimaryDB' ] )
|
|
->setConstructorArgs( [ [
|
|
'name' => 'migratefilerepolayouttest',
|
|
'backend' => $backend
|
|
] ] )
|
|
->getMock();
|
|
|
|
$repoMock
|
|
->method( 'getPrimaryDB' )
|
|
->willReturn( $dbMock );
|
|
|
|
$this->migratorMock = $this->getMockBuilder( MigrateFileRepoLayout::class )
|
|
->onlyMethods( [ 'getRepo' ] )->getMock();
|
|
$this->migratorMock
|
|
->method( 'getRepo' )
|
|
->willReturn( $repoMock );
|
|
|
|
$this->tmpFilepath = TempFSFile::factory(
|
|
'migratefilelayout-test-', 'png', wfTempDir() )->getPath();
|
|
|
|
file_put_contents( $this->tmpFilepath, $this->text );
|
|
|
|
$hashPath = $repoMock->getHashPath( $filename );
|
|
|
|
$status = $repoMock->store(
|
|
$this->tmpFilepath,
|
|
'public',
|
|
$hashPath . $filename,
|
|
FileRepo::OVERWRITE
|
|
);
|
|
}
|
|
|
|
protected function deleteFilesRecursively( $directory ) {
|
|
foreach ( glob( $directory . '/*' ) as $file ) {
|
|
if ( is_dir( $file ) ) {
|
|
$this->deleteFilesRecursively( $file );
|
|
} else {
|
|
unlink( $file );
|
|
}
|
|
}
|
|
|
|
rmdir( $directory );
|
|
}
|
|
|
|
protected function tearDown(): void {
|
|
foreach ( glob( $this->tmpPrefix . '*' ) as $directory ) {
|
|
$this->deleteFilesRecursively( $directory );
|
|
}
|
|
|
|
unlink( $this->tmpFilepath );
|
|
|
|
parent::tearDown();
|
|
}
|
|
|
|
public function testMigration() {
|
|
$this->migratorMock->loadParamsAndArgs(
|
|
null,
|
|
[ 'oldlayout' => 'name', 'newlayout' => 'sha1' ]
|
|
);
|
|
|
|
ob_start();
|
|
|
|
$this->migratorMock->execute();
|
|
|
|
ob_end_clean();
|
|
|
|
$sha1 = sha1( $this->text );
|
|
|
|
$expectedOriginalFilepath = $this->tmpPrefix
|
|
. '-original/'
|
|
. substr( $sha1, 0, 1 )
|
|
. '/'
|
|
. substr( $sha1, 1, 1 )
|
|
. '/'
|
|
. substr( $sha1, 2, 1 )
|
|
. '/'
|
|
. $sha1;
|
|
|
|
$this->assertEquals(
|
|
file_get_contents( $expectedOriginalFilepath ),
|
|
$this->text,
|
|
'New sha1 file should be exist and have the right contents'
|
|
);
|
|
|
|
$expectedPublicFilepath = $this->tmpPrefix . '-public/f/f8/Foo.png';
|
|
|
|
$this->assertEquals(
|
|
file_get_contents( $expectedPublicFilepath ),
|
|
$this->text,
|
|
'Existing name file should still and have the right contents'
|
|
);
|
|
}
|
|
}
|