UploadFromUrlTest.php

Go to the documentation of this file.

<?php
 
class UploadFromUrlTest extends ApiTestCase {
    protected function setUp() {
        parent::setUp();
 
        $this->setMwGlobals( [
            'wgEnableUploads' => true,
            'wgAllowCopyUploads' => true,
        ] );
 
        if ( wfLocalFile( 'UploadFromUrlTest.png' )->exists() ) {
            $this->deleteFile( 'UploadFromUrlTest.png' );
        }
    }
 
    protected function doApiRequest( array $params, array $unused = null,
        $appendModule = false, User $user = null, $tokenType = null
    ) {
        global $wgRequest;
 
        $req = new FauxRequest( $params, true, $wgRequest->getSession() );
        $module = new ApiMain( $req, true );
        $module->execute();
 
        return [
            $module->getResult()->getResultData( null, [ 'Strip' => 'all' ] ),
            $req
        ];
    }
 
    public function testClearQueue() {
        $job = JobQueueGroup::singleton()->pop();
        while ( $job ) {
            $job = JobQueueGroup::singleton()->pop();
        }
        $this->assertFalse( $job );
    }
 
    public function testSetupUrlDownload( $data ) {
        $token = $this->user->getEditToken();
        $exception = false;
 
        try {
            $this->doApiRequest( [
                'action' => 'upload',
            ] );
        } catch ( ApiUsageException $e ) {
            $exception = true;
            $this->assertEquals( "The token parameter must be set", $e->getMessage() );
        }
        $this->assertTrue( $exception, "Got exception" );
 
        $exception = false;
        try {
            $this->doApiRequest( [
                'action' => 'upload',
                'token' => $token,
            ], $data );
        } catch ( ApiUsageException $e ) {
            $exception = true;
            $this->assertEquals( "One of the parameters sessionkey, file, url is required",
                $e->getMessage() );
        }
        $this->assertTrue( $exception, "Got exception" );
 
        $exception = false;
        try {
            $this->doApiRequest( [
                'action' => 'upload',
                'url' => 'http://www.example.com/test.png',
                'token' => $token,
            ], $data );
        } catch ( ApiUsageException $e ) {
            $exception = true;
            $this->assertEquals( "The filename parameter must be set", $e->getMessage() );
        }
        $this->assertTrue( $exception, "Got exception" );
 
        $this->user->removeGroup( 'sysop' );
        $exception = false;
        try {
            $this->doApiRequest( [
                'action' => 'upload',
                'url' => 'http://www.example.com/test.png',
                'filename' => 'UploadFromUrlTest.png',
                'token' => $token,
            ], $data );
        } catch ( ApiUsageException $e ) {
            $exception = true;
            $this->assertEquals( "Permission denied", $e->getMessage() );
        }
        $this->assertTrue( $exception, "Got exception" );
    }
 
    public function testSyncDownload( $data ) {
        $token = $this->user->getEditToken();
 
        $job = JobQueueGroup::singleton()->pop();
        $this->assertFalse( $job, 'Starting with an empty jobqueue' );
 
        $this->user->addGroup( 'users' );
        $data = $this->doApiRequest( [
            'action' => 'upload',
            'filename' => 'UploadFromUrlTest.png',
            'url' => 'http://upload.wikimedia.org/wikipedia/mediawiki/b/bc/Wiki.png',
            'ignorewarnings' => true,
            'token' => $token,
        ], $data );
 
        $job = JobQueueGroup::singleton()->pop();
        $this->assertFalse( $job );
 
        $this->assertEquals( 'Success', $data[0]['upload']['result'] );
        $this->deleteFile( 'UploadFromUrlTest.png' );
 
        return $data;
    }
 
    protected function deleteFile( $name ) {
        $t = Title::newFromText( $name, NS_FILE );
        $this->assertTrue( $t->exists(), "File '$name' exists" );
 
        if ( $t->exists() ) {
            $file = wfFindFile( $name, [ 'ignoreRedirect' => true ] );
            $empty = "";
            FileDeleteForm::doDelete( $t, $file, $empty, "none", true );
            $page = WikiPage::factory( $t );
            $page->doDeleteArticle( "testing" );
        }
        $t = Title::newFromText( $name, NS_FILE );
 
        $this->assertFalse( $t->exists(), "File '$name' was deleted" );
    }
}