ApiTestCaseUpload.php

Go to the documentation of this file.

<?php

abstract class ApiTestCaseUpload extends ApiTestCase {
        public function setUp() {
                global $wgEnableUploads, $wgEnableAPI;
                parent::setUp();

                $wgEnableUploads = true;
                $wgEnableAPI = true;
                wfSetupSession();

                $this->clearFakeUploads();
        }

        public function tearDown() {
                $this->clearTempUpload();
        }

        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();
        }




}