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