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() {
                global $wgUser;
                $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' );
        }
}