3 require_once dirname( __FILE__
) . '/api/ApiSetup.php';
8 class UploadFromUrlTest
extends ApiTestSetup
{
10 public function setUp() {
11 global $wgEnableUploads, $wgAllowCopyUploads, $wgAllowAsyncCopyUploads;
14 $wgEnableUploads = true;
15 $wgAllowCopyUploads = true;
16 $wgAllowAsyncCopyUploads = true;
19 ini_set( 'log_errors', 1 );
20 ini_set( 'error_reporting', 1 );
21 ini_set( 'display_errors', 1 );
23 if ( wfLocalFile( 'UploadFromUrlTest.png' )->exists() ) {
24 $this->deleteFile( 'UploadFromUrlTest.png' );
28 protected function doApiRequest( $params, $unused = null ) {
29 $sessionId = session_id();
30 session_write_close();
32 $req = new FauxRequest( $params, true, $_SESSION );
33 $module = new ApiMain( $req, true );
36 wfSetupSession( $sessionId );
37 return array( $module->getResultData(), $req );
41 * Ensure that the job queue is empty before continuing
43 public function testClearQueue() {
44 while ( $job = Job
::pop() ) { }
45 $this->assertFalse( $job );
49 * @todo Document why we test login, since the $wgUser hack used doesn't
52 public function testLogin() {
53 $data = $this->doApiRequest( array(
55 'lgname' => self
::$userName,
56 'lgpassword' => self
::$passWord ) );
57 $this->assertArrayHasKey( "login", $data[0] );
58 $this->assertArrayHasKey( "result", $data[0]['login'] );
59 $this->assertEquals( "NeedToken", $data[0]['login']['result'] );
60 $token = $data[0]['login']['token'];
62 $data = $this->doApiRequest( array(
65 "lgname" => self
::$userName,
66 "lgpassword" => self
::$passWord ) );
68 $this->assertArrayHasKey( "login", $data[0] );
69 $this->assertArrayHasKey( "result", $data[0]['login'] );
70 $this->assertEquals( "Success", $data[0]['login']['result'] );
71 $this->assertArrayHasKey( 'lgtoken', $data[0]['login'] );
78 * @depends testClearQueue
80 public function testSetupUrlDownload( $data ) {
81 $token = self
::$user->editToken();
85 $this->doApiRequest( array(
88 } catch ( UsageException
$e ) {
90 $this->assertEquals( "The token parameter must be set", $e->getMessage() );
92 $this->assertTrue( $exception, "Got exception" );
96 $this->doApiRequest( array(
100 } catch ( UsageException
$e ) {
102 $this->assertEquals( "One of the parameters sessionkey, file, url, statuskey is required",
105 $this->assertTrue( $exception, "Got exception" );
109 $this->doApiRequest( array(
110 'action' => 'upload',
111 'url' => 'http://www.example.com/test.png',
114 } catch ( UsageException
$e ) {
116 $this->assertEquals( "The filename parameter must be set", $e->getMessage() );
118 $this->assertTrue( $exception, "Got exception" );
120 self
::$user->removeGroup( 'sysop' );
123 $this->doApiRequest( array(
124 'action' => 'upload',
125 'url' => 'http://www.example.com/test.png',
126 'filename' => 'UploadFromUrlTest.png',
129 } catch ( UsageException
$e ) {
131 $this->assertEquals( "Permission denied", $e->getMessage() );
133 $this->assertTrue( $exception, "Got exception" );
135 self
::$user->addGroup( 'sysop' );
137 $data = $this->doApiRequest( array(
138 'action' => 'upload',
139 'url' => 'http://bits.wikimedia.org/skins-1.5/common/images/poweredby_mediawiki_88x31.png',
140 'asyncdownload' => 1,
141 'filename' => 'UploadFromUrlTest.png',
145 $this->assertEquals( $data[0]['upload']['result'], 'Queued', 'Queued upload' );
148 $this->assertThat( $job, $this->isInstanceOf( 'UploadFromUrlJob' ), 'Queued upload inserted' );
153 * @depends testClearQueue
155 public function testAsyncUpload( $data ) {
156 $token = self
::$user->editToken();
158 self
::$user->addGroup( 'users' );
160 $data = $this->doAsyncUpload( $token, true );
161 $this->assertEquals( $data[0]['upload']['result'], 'Success' );
162 $this->assertEquals( $data[0]['upload']['filename'], 'UploadFromUrlTest.png' );
163 $this->assertTrue( wfLocalFile( $data[0]['upload']['filename'] )->exists() );
165 $this->deleteFile( 'UploadFromUrlTest.png' );
172 * @depends testClearQueue
174 public function testAsyncUploadWarning( $data ) {
175 $token = self
::$user->editToken();
177 self
::$user->addGroup( 'users' );
180 $data = $this->doAsyncUpload( $token );
182 $this->assertEquals( $data[0]['upload']['result'], 'Warning' );
183 $this->assertTrue( isset( $data[0]['upload']['sessionkey'] ) );
185 $data = $this->doApiRequest( array(
186 'action' => 'upload',
187 'sessionkey' => $data[0]['upload']['sessionkey'],
188 'filename' => 'UploadFromUrlTest.png',
189 'ignorewarnings' => 1,
192 $this->assertEquals( $data[0]['upload']['result'], 'Success' );
193 $this->assertEquals( $data[0]['upload']['filename'], 'UploadFromUrlTest.png' );
194 $this->assertTrue( wfLocalFile( $data[0]['upload']['filename'] )->exists() );
196 $this->deleteFile( 'UploadFromUrlTest.png' );
203 * @depends testClearQueue
205 public function testSyncDownload( $data ) {
206 $token = self
::$user->editToken();
209 $this->assertFalse( $job, 'Starting with an empty jobqueue' );
211 self
::$user->addGroup( 'users' );
212 $data = $this->doApiRequest( array(
213 'action' => 'upload',
214 'filename' => 'UploadFromUrlTest.png',
215 'url' => 'http://bits.wikimedia.org/skins-1.5/common/images/poweredby_mediawiki_88x31.png',
216 'ignorewarnings' => true,
221 $this->assertFalse( $job );
223 $this->assertEquals( 'Success', $data[0]['upload']['result'] );
224 $this->deleteFile( 'UploadFromUrlTest.png' );
229 public function testLeaveMessage() {
230 $token = self
::$user->editToken();
232 $talk = self
::$user->getTalkPage();
233 if ( $talk->exists() ) {
234 $a = new Article( $talk );
235 $a->doDeleteArticle( '' );
238 $this->assertFalse( (bool)$talk->getArticleId( Title
::GAID_FOR_UPDATE
), 'User talk does not exist' );
240 $data = $this->doApiRequest( array(
241 'action' => 'upload',
242 'filename' => 'UploadFromUrlTest.png',
243 'url' => 'http://bits.wikimedia.org/skins-1.5/common/images/poweredby_mediawiki_88x31.png',
244 'asyncdownload' => 1,
247 'ignorewarnings' => 1,
251 $this->assertEquals( 'UploadFromUrlJob', get_class( $job ) );
254 $this->assertTrue( wfLocalFile( 'UploadFromUrlTest.png' )->exists() );
255 $this->assertTrue( (bool)$talk->getArticleId( Title
::GAID_FOR_UPDATE
), 'User talk exists' );
257 $this->deleteFile( 'UploadFromUrlTest.png' );
259 $talkRev = Revision
::newFromTitle( $talk );
260 $talkSize = $talkRev->getSize();
264 $data = $this->doApiRequest( array(
265 'action' => 'upload',
266 'filename' => 'UploadFromUrlTest.png',
267 'url' => 'http://bits.wikimedia.org/skins-1.5/common/images/poweredby_mediawiki_88x31.png',
268 'asyncdownload' => 1,
272 } catch ( UsageException
$e ) {
274 $this->assertEquals( 'Using leavemessage without ignorewarnings is not supported', $e->getMessage() );
276 $this->assertTrue( $exception );
279 $this->assertFalse( $job );
284 // Broken until using leavemessage with ignorewarnings is supported
287 $this->assertFalse( wfLocalFile( 'UploadFromUrlTest.png' )->exists() );
289 $talkRev = Revision::newFromTitle( $talk );
290 $this->assertTrue( $talkRev->getSize() > $talkSize, 'New message left' );
295 * Helper function to perform an async upload, execute the job and fetch
298 * @return array The result of action=upload&statuskey=key
300 private function doAsyncUpload( $token, $ignoreWarnings = false, $leaveMessage = false ) {
302 'action' => 'upload',
303 'filename' => 'UploadFromUrlTest.png',
304 'url' => 'http://bits.wikimedia.org/skins-1.5/common/images/poweredby_mediawiki_88x31.png',
305 'asyncdownload' => 1,
308 if ( $ignoreWarnings ) {
309 $params['ignorewarnings'] = 1;
311 if ( $leaveMessage ) {
312 $params['leavemessage'] = 1;
315 $data = $this->doApiRequest( $params );
316 $this->assertEquals( $data[0]['upload']['result'], 'Queued' );
317 $this->assertTrue( isset( $data[0]['upload']['statuskey'] ) );
318 $statusKey = $data[0]['upload']['statuskey'];
321 $this->assertEquals( 'UploadFromUrlJob', get_class( $job ) );
323 $status = $job->run();
324 $this->assertTrue( $status );
326 $data = $this->doApiRequest( array(
327 'action' => 'upload',
328 'statuskey' => $statusKey,
339 protected function deleteFile( $name ) {
340 $t = Title
::newFromText( $name, NS_FILE
);
341 $this->assertTrue($t->exists(), "File '$name' exists");
343 if ( $t->exists() ) {
344 $file = wfFindFile( $name, array( 'ignoreRedirect' => true ) );
346 FileDeleteForm
::doDelete( $t, $file, $empty, "none", true );
347 $a = new Article ( $t );
348 $a->doDeleteArticle( "testing" );
350 $t = Title
::newFromText( $name, NS_FILE
);
352 $this->assertFalse($t->exists(), "File '$name' was deleted");