ApiTestCaseUpload.php

Go to the documentation of this file.

<?php

abstract class ApiTestCaseUpload extends ApiTestCase {
        protected function setUp() {
                parent::setUp();

                $this->setMwGlobals( array(
                        'wgEnableUploads' => true,
                        'wgEnableAPI' => true,
                ) );

                wfSetupSession();

                $this->clearFakeUploads();
        }

        protected function tearDown() {
                $this->clearTempUpload();

                parent::tearDown();
        }

        public function deleteFileByTitle( $title ) {
                if ( $title->exists() ) {
                        $file = wfFindFile( $title, array( 'ignoreRedirect' => true ) );
                        $noOldArchive = ""; // yes this really needs to be set this way
                        $comment = "removing for test";
                        $restrictDeletedVersions = false;
                        $status = FileDeleteForm::doDelete( $title, $file, $noOldArchive, $comment, $restrictDeletedVersions );
                        if ( !$status->isGood() ) {
                                return false;
                        }
                        $page = WikiPage::factory( $title );
                        $page->doDeleteArticle( "removing for test" );

                        // see if it now doesn't exist; reload
                        $title = Title::newFromText( $title->getText(), NS_FILE );
                }
                return !( $title && $title instanceof Title && $title->exists() );
        }

        public function deleteFileByFileName( $fileName ) {
                return $this->deleteFileByTitle( Title::newFromText( $fileName, NS_FILE ) );
        }

        public function deleteFileByContent( $filePath ) {
                $hash = FSFile::getSha1Base36FromPath( $filePath );
                $dupes = RepoGroup::singleton()->findBySha1( $hash );
                $success = true;
                foreach ( $dupes as $dupe ) {
                        $success &= $this->deleteFileByTitle( $dupe->getTitle() );
                }
                return $success;
        }

        function fakeUploadFile( $fieldName, $fileName, $type, $filePath ) {
                $tmpName = tempnam( wfTempDir(), "" );
                if ( !file_exists( $filePath ) ) {
                        throw new Exception( "$filePath doesn't exist!" );
                }

                if ( !copy( $filePath, $tmpName ) ) {
                        throw new Exception( "couldn't copy $filePath to $tmpName" );
                }

                clearstatcache();
                $size = filesize( $tmpName );
                if ( $size === false ) {
                        throw new Exception( "couldn't stat $tmpName" );
                }

                $_FILES[$fieldName] = array(
                        'name' => $fileName,
                        'type' => $type,
                        'tmp_name' => $tmpName,
                        'size' => $size,
                        'error' => null
                );

                return true;

        }

        function fakeUploadChunk( $fieldName, $fileName, $type, & $chunkData ) {
                $tmpName = tempnam( wfTempDir(), "" );
                // copy the chunk data to temp location:
                if ( !file_put_contents( $tmpName, $chunkData ) ) {
                        throw new Exception( "couldn't copy chunk data to $tmpName" );
                }

                clearstatcache();
                $size = filesize( $tmpName );
                if ( $size === false ) {
                        throw new Exception( "couldn't stat $tmpName" );
                }

                $_FILES[$fieldName] = array(
                        'name' => $fileName,
                        'type' => $type,
                        'tmp_name' => $tmpName,
                        'size' => $size,
                        'error' => null
                );
        }

        function clearTempUpload() {
                if ( isset( $_FILES['file']['tmp_name'] ) ) {
                        $tmp = $_FILES['file']['tmp_name'];
                        if ( file_exists( $tmp ) ) {
                                unlink( $tmp );
                        }
                }
        }

        function clearFakeUploads() {
                $_FILES = array();
        }

}