1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
<?php
class MigrateFileRepoLayoutTest extends MediaWikiTestCase {
protected $tmpPrefix;
protected $migratorMock;
protected $tmpFilepath;
protected $text = 'testing';
protected function setUp() {
parent::setUp();
$filename = 'Foo.png';
$this->tmpPrefix = wfTempDir() . '/migratefilelayout-test-' . time() . '-' . mt_rand();
$backend = new FSFileBackend( array(
'name' => 'local-migratefilerepolayouttest',
'wikiId' => wfWikiID(),
'containerPaths' => array(
'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( 'DatabaseMysql' )
->disableOriginalConstructor()
->getMock();
$imageRow = new stdClass;
$imageRow->img_name = $filename;
$imageRow->img_sha1 = sha1( $this->text );
$dbMock->expects( $this->any() )
->method( 'select' )
->will( $this->onConsecutiveCalls(
new FakeResultWrapper( array( $imageRow ) ), // image
new FakeResultWrapper( array() ), // image
new FakeResultWrapper( array() ) // filearchive
) );
$repoMock = $this->getMock( 'LocalRepo',
array( 'getMasterDB' ),
array( array(
'name' => 'migratefilerepolayouttest',
'backend' => $backend
) ) );
$repoMock->expects( $this->any() )->method( 'getMasterDB' )->will( $this->returnValue( $dbMock ) );
$this->migratorMock = $this->getMock( 'MigrateFileRepoLayout', array( 'getRepo' ) );
$this->migratorMock->expects( $this->any() )->method( 'getRepo' )->will( $this->returnValue( $repoMock ) );
$this->tmpFilepath = TempFSFile::factory( 'migratefilelayout-test-', 'png' )->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() {
foreach ( glob( $this->tmpPrefix . '*' ) as $directory ) {
$this->deleteFilesRecursively( $directory );
}
unlink( $this->tmpFilepath );
parent::tearDown();
}
public function testMigration() {
$this->migratorMock->loadParamsAndArgs( null, array( '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' );
}
}
|
