tests/phpunit/suites/UploadFromUrlTestSuite.php

   1 <?php
   2
   3 require_once dirname( __DIR__ ) . '/includes/upload/UploadFromUrlTest.php';
   4
   5 class UploadFromUrlTestSuite extends PHPUnit_Framework_TestSuite {
   6         public $savedGlobals = array();
   7
   8         public static function addTables( &$tables ) {
   9                 $tables[] = 'user_properties';
  10                 $tables[] = 'filearchive';
  11                 $tables[] = 'logging';
  12                 $tables[] = 'updatelog';
  13                 $tables[] = 'iwlinks';
  14
  15                 return true;
  16         }
  17
  18         protected function setUp() {
  19                 global $wgParser, $wgParserConf, $IP, $messageMemc, $wgMemc, $wgUser,
  20                         $wgLang, $wgOut, $wgRequest, $wgStyleDirectory,
  21                         $wgParserCacheType, $wgNamespaceAliases, $wgNamespaceProtection,
  22                         $parserMemc;
  23
  24                 $tmpDir = $this->getNewTempDirectory();
  25                 $tmpGlobals = array();
  26
  27                 $tmpGlobals['wgScript'] = '/index.php';
  28                 $tmpGlobals['wgScriptPath'] = '/';
  29                 $tmpGlobals['wgArticlePath'] = '/wiki/$1';
  30                 $tmpGlobals['wgStylePath'] = '/skins';
  31                 $tmpGlobals['wgThumbnailScriptPath'] = false;
  32                 $tmpGlobals['wgLocalFileRepo'] = array(
  33                         'class' => 'LocalRepo',
  34                         'name' => 'local',
  35                         'url' => 'http://example.com/images',
  36                         'hashLevels' => 2,
  37                         'transformVia404' => false,
  38                         'backend' => new FSFileBackend( array(
  39                                 'name' => 'local-backend',
  40                                 'wikiId' => wfWikiId(),
  41                                 'containerPaths' => array(
  42                                         'local-public' => "{$tmpDir}/test-repo/public",
  43                                         'local-thumb' => "{$tmpDir}/test-repo/thumb",
  44                                         'local-temp' => "{$tmpDir}/test-repo/temp",
  45                                         'local-deleted' => "{$tmpDir}/test-repo/delete",
  46                                 )
  47                         ) ),
  48                 );
  49                 foreach ( $tmpGlobals as $var => $val ) {
  50                         if ( array_key_exists( $var, $GLOBALS ) ) {
  51                                 $this->savedGlobals[$var] = $GLOBALS[$var];
  52                         }
  53                         $GLOBALS[$var] = $val;
  54                 }
  55
  56                 $wgNamespaceProtection[NS_MEDIAWIKI] = 'editinterface';
  57                 $wgNamespaceAliases['Image'] = NS_FILE;
  58                 $wgNamespaceAliases['Image_talk'] = NS_FILE_TALK;
  59
  60                 $wgParserCacheType = CACHE_NONE;
  61                 DeferredUpdates::clearPendingUpdates();
  62                 $wgMemc = wfGetMainCache();
  63                 $messageMemc = wfGetMessageCacheStorage();
  64                 $parserMemc = wfGetParserCacheStorage();
  65
  66                 $wgUser = new User;
  67                 $context = new RequestContext;
  68                 $wgLang = $context->getLanguage();
  69                 $wgOut = $context->getOutput();
  70                 $wgParser = new StubObject( 'wgParser', $wgParserConf['class'], array( $wgParserConf ) );
  71                 $wgRequest = $context->getRequest();
  72
  73                 if ( $wgStyleDirectory === false ) {
  74                         $wgStyleDirectory = "$IP/skins";
  75                 }
  76
  77                 RepoGroup::destroySingleton();
  78                 FileBackendGroup::destroySingleton();
  79         }
  80
  81         protected function tearDown() {
  82                 foreach ( $this->savedGlobals as $var => $val ) {
  83                         $GLOBALS[$var] = $val;
  84                 }
  85                 // Restore backends
  86                 RepoGroup::destroySingleton();
  87                 FileBackendGroup::destroySingleton();
  88
  89                 $this->teardownUploadDir( $this->uploadDir );
  90
  91                 parent::tearDown();
  92         }
  93
  94         private $uploadDir;
  95         private $keepUploads;
  96
  97         /**
  98          * Remove the dummy uploads directory
  99          * @param string $dir
 100          */
 101         private function teardownUploadDir( $dir ) {
 102                 if ( $this->keepUploads ) {
 103                         return;
 104                 }
 105
 106                 // delete the files first, then the dirs.
 107                 self::deleteFiles(
 108                         array(
 109                                 "$dir/3/3a/Foobar.jpg",
 110                                 "$dir/thumb/3/3a/Foobar.jpg/180px-Foobar.jpg",
 111                                 "$dir/thumb/3/3a/Foobar.jpg/200px-Foobar.jpg",
 112                                 "$dir/thumb/3/3a/Foobar.jpg/640px-Foobar.jpg",
 113                                 "$dir/thumb/3/3a/Foobar.jpg/120px-Foobar.jpg",
 114
 115                                 "$dir/0/09/Bad.jpg",
 116                         )
 117                 );
 118
 119                 self::deleteDirs(
 120                         array(
 121                                 "$dir/3/3a",
 122                                 "$dir/3",
 123                                 "$dir/thumb/6/65",
 124                                 "$dir/thumb/6",
 125                                 "$dir/thumb/3/3a/Foobar.jpg",
 126                                 "$dir/thumb/3/3a",
 127                                 "$dir/thumb/3",
 128
 129                                 "$dir/0/09/",
 130                                 "$dir/0/",
 131
 132                                 "$dir/thumb",
 133                                 "$dir",
 134                         )
 135                 );
 136         }
 137
 138         /**
 139          * Delete the specified files, if they exist.
 140          *
 141          * @param array $files Full paths to files to delete.
 142          */
 143         private static function deleteFiles( $files ) {
 144                 foreach ( $files as $file ) {
 145                         if ( file_exists( $file ) ) {
 146                                 unlink( $file );
 147                         }
 148                 }
 149         }
 150
 151         /**
 152          * Delete the specified directories, if they exist. Must be empty.
 153          *
 154          * @param array $dirs Full paths to directories to delete.
 155          */
 156         private static function deleteDirs( $dirs ) {
 157                 foreach ( $dirs as $dir ) {
 158                         if ( is_dir( $dir ) ) {
 159                                 rmdir( $dir );
 160                         }
 161                 }
 162         }
 163
 164         /**
 165          * Create a dummy uploads directory which will contain a couple
 166          * of files in order to pass existence tests.
 167          *
 168          * @return string The directory
 169          */
 170         private function setupUploadDir() {
 171                 global $IP;
 172
 173                 if ( $this->keepUploads ) {
 174                         $dir = wfTempDir() . '/mwParser-images';
 175
 176                         if ( is_dir( $dir ) ) {
 177                                 return $dir;
 178                         }
 179                 } else {
 180                         $dir = $this->getNewTempDirectory();
 181                 }
 182
 183                 wfDebug( "Creating upload directory $dir\n" );
 184
 185                 wfMkdirParents( $dir . '/3/3a', null, __METHOD__ );
 186                 copy( "$IP/tests/phpunit/data/upload/headbg.jpg", "$dir/3/3a/Foobar.jpg" );
 187
 188                 wfMkdirParents( $dir . '/0/09', null, __METHOD__ );
 189                 copy( "$IP/tests/phpunit/data/upload/headbg.jpg", "$dir/0/09/Bad.jpg" );
 190
 191                 return $dir;
 192         }
 193
 194         public static function suite() {
 195                 // Hack to invoke the autoloader required to get phpunit to recognize
 196                 // the UploadFromUrlTest class
 197                 class_exists( 'UploadFromUrlTest' );
 198                 $suite = new UploadFromUrlTestSuite( 'UploadFromUrlTest' );
 199
 200                 return $suite;
 201         }
 202 }