Page MenuHomeWickedGov Phorge
Files F2753287

StoreBatchTest.php
No OneTemporary
Actions

Size
5 KB
Referenced Files
None
Subscribers
None

StoreBatchTest.php
View Options

<?php
use MediaWiki\Status\Status;
use MediaWiki\WikiMap\WikiMap;
use Wikimedia\FileBackend\FSFileBackend;
/**
* @group FileRepo
* @group medium
*/
class StoreBatchTest extends MediaWikiIntegrationTestCase {
/** @var string[] */
protected $createdFiles;
/** @var string */
protected $date;
/** @var FileRepo */
protected $repo;
protected function setUp(): void {
global $wgFileBackends;
parent::setUp();
# Forge a FileRepo object to not have to rely on local wiki settings
$tmpPrefix = $this->getNewTempDirectory();
if ( $this->getCliArg( 'use-filebackend' ) ) {
$name = $this->getCliArg( 'use-filebackend' );
$useConfig = [];
foreach ( $wgFileBackends as $conf ) {
if ( $conf['name'] == $name ) {
$useConfig = $conf;
}
}
$useConfig['lockManager'] = $this->getServiceContainer()->getLockManagerGroupFactory()
->getLockManagerGroup()->get( $useConfig['lockManager'] );
$useConfig['name'] = 'local-testing'; // swap name
$class = $useConfig['class'];
$backend = new $class( $useConfig );
} else {
$backend = new FSFileBackend( [
'name' => 'local-testing',
'wikiId' => WikiMap::getCurrentWikiId(),
'containerPaths' => [
'unittests-public' => "{$tmpPrefix}/public",
'unittests-thumb' => "{$tmpPrefix}/thumb",
'unittests-temp' => "{$tmpPrefix}/temp",
'unittests-deleted' => "{$tmpPrefix}/deleted",
]
] );
}
$this->repo = new FileRepo( [
'name' => 'unittests',
'backend' => $backend
] );
$this->date = gmdate( "YmdHis" );
$this->createdFiles = [];
}
protected function tearDown(): void {
// Delete files
$this->repo->cleanupBatch( $this->createdFiles );
parent::tearDown();
}
/**
* Store a file or virtual URL source into a media file name.
*
* @param string $originalName The title of the image
* @param string $srcPath The filepath or virtual URL
* @param int $flags Flags to pass into repo::store().
* @return Status
*/
private function storeit( $originalName, $srcPath, $flags ) {
$hashPath = $this->repo->getHashPath( $originalName );
$dstRel = "$hashPath{$this->date}!$originalName";
$dstUrlRel = $hashPath . $this->date . '!' . rawurlencode( $originalName );
$result = $this->repo->store( $srcPath, 'temp', $dstRel, $flags );
$result->value = $this->repo->getVirtualUrl( 'temp' ) . '/' . $dstUrlRel;
$this->createdFiles[] = $result->value;
return $result;
}
/**
* Test storing a file using different flags.
*
* @param string $fn The title of the image
* @param string $infn The name of the file (in the filesystem)
* @param string $otherfn The name of the different file (in the filesystem)
* @param bool $fromrepo 'true' if we want to copy from a virtual URL out of the Repo.
*/
private function storecohort( $fn, $infn, $otherfn, $fromrepo ) {
$f = $this->storeit( $fn, $infn, 0 );
$this->assertStatusGood( $f, 'failed to store a new file' );
$this->assertSame( 0, $f->failCount, "counts wrong {$f->successCount} {$f->failCount}" );
$this->assertSame( 1, $f->successCount, "counts wrong {$f->successCount} {$f->failCount}" );
if ( $fromrepo ) {
$f = $this->storeit( "Other-$fn", $infn, FileRepo::OVERWRITE );
$infn = $f->value;
}
// This should work because we're allowed to overwrite
$f = $this->storeit( $fn, $infn, FileRepo::OVERWRITE );
$this->assertStatusGood( $f, 'We should be allowed to overwrite' );
$this->assertSame( 0, $f->failCount, "counts wrong {$f->successCount} {$f->failCount}" );
$this->assertSame( 1, $f->successCount, "counts wrong {$f->successCount} {$f->failCount}" );
// This should fail because we're overwriting.
$f = $this->storeit( $fn, $infn, 0 );
$this->assertStatusError( 'backend-fail-alreadyexists', $f, 'We should not be allowed to overwrite' );
$this->assertSame( 1, $f->failCount, "counts wrong {$f->successCount} {$f->failCount}" );
$this->assertSame( 0, $f->successCount, "counts wrong {$f->successCount} {$f->failCount}" );
// This should succeed because we're overwriting the same content.
$f = $this->storeit( $fn, $infn, FileRepo::OVERWRITE_SAME );
$this->assertStatusGood( $f, 'We should be able to overwrite the same content' );
$this->assertSame( 0, $f->failCount, "counts wrong {$f->successCount} {$f->failCount}" );
$this->assertSame( 1, $f->successCount, "counts wrong {$f->successCount} {$f->failCount}" );
// This should fail because we're overwriting different content.
if ( $fromrepo ) {
$f = $this->storeit( "Other-$fn", $otherfn, FileRepo::OVERWRITE );
$otherfn = $f->value;
}
$f = $this->storeit( $fn, $otherfn, FileRepo::OVERWRITE_SAME );
$this->assertStatusError( 'backend-fail-notsame', $f, 'We should not be allowed to overwrite different content' );
$this->assertSame( 1, $f->failCount, "counts wrong {$f->successCount} {$f->failCount}" );
$this->assertSame( 0, $f->successCount, "counts wrong {$f->successCount} {$f->failCount}" );
}
/**
* @covers \FileRepo::store
*/
public function teststore() {
global $IP;
$this->storecohort(
"Test1.png",
"$IP/tests/phpunit/data/filerepo/wiki.png",
"$IP/tests/phpunit/data/filerepo/video.png",
false
);
$this->storecohort(
"Test2.png",
"$IP/tests/phpunit/data/filerepo/wiki.png",
"$IP/tests/phpunit/data/filerepo/video.png",
true
);
}
}

File Metadata

Mime Type
text/x-php
Expires
Fri, Jul 3, 21:02 (22 h, 40 m)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
8f/4b/fa756305b2da8b8396d7cda6c25e
Default Alt Text
StoreBatchTest.php (5 KB)

Event Timeline