UploadStashTest.php

Go to the documentation of this file.

<?php
class UploadStashTest extends MediaWikiTestCase {
    public static $users;

    protected function setUp() {
        parent::setUp();

        // Setup a file for bug 29408
        $this->bug29408File = __DIR__ . '/bug29408';
        file_put_contents( $this->bug29408File, "\x00" );

        self::$users = array(
            'sysop' => new TestUser(
                'Uploadstashtestsysop',
                'Upload Stash Test Sysop',
                '[email protected]',
                array( 'sysop' )
            ),
            'uploader' => new TestUser(
                'Uploadstashtestuser',
                'Upload Stash Test User',
                '[email protected]',
                array()
            )
        );
    }

    protected function tearDown() {
        if ( file_exists( $this->bug29408File . "." ) ) {
            unlink( $this->bug29408File . "." );
        }

        if ( file_exists( $this->bug29408File ) ) {
            unlink( $this->bug29408File );
        }

        parent::tearDown();
    }

    public function testBug29408() {
        $this->setMwGlobals( 'wgUser', self::$users['uploader']->user );

        $repo = RepoGroup::singleton()->getLocalRepo();
        $stash = new UploadStash( $repo );

        // Throws exception caught by PHPUnit on failure
        $file = $stash->stashFile( $this->bug29408File );
        // We'll never reach this point if we hit bug 29408
        $this->assertTrue( true, 'Unrecognized file without extension' );

        $stash->removeFile( $file->getFileKey() );
    }

    public function testValidRequest() {
        $request = new FauxRequest( array( 'wpFileKey' => 'foo' ) );
        $this->assertFalse( UploadFromStash::isValidRequest( $request ), 'Check failure on bad wpFileKey' );

        $request = new FauxRequest( array( 'wpSessionKey' => 'foo' ) );
        $this->assertFalse( UploadFromStash::isValidRequest( $request ), 'Check failure on bad wpSessionKey' );

        $request = new FauxRequest( array( 'wpFileKey' => 'testkey-test.test' ) );
        $this->assertTrue( UploadFromStash::isValidRequest( $request ), 'Check good wpFileKey' );

        $request = new FauxRequest( array( 'wpFileKey' => 'testkey-test.test' ) );
        $this->assertTrue( UploadFromStash::isValidRequest( $request ), 'Check good wpSessionKey' );

        $request = new FauxRequest( array( 'wpFileKey' => 'testkey-test.test', 'wpSessionKey' => 'foo' ) );
        $this->assertTrue( UploadFromStash::isValidRequest( $request ), 'Check key precedence' );
    }
}