wiki.techinc.nl/tests/phpunit/includes/libs/ParamValidator/Util/UploadedFileTestBase.php
DannyS712 6a93b0ca93 More misc test cleanup
* parent::setUp() should be first, and ::tearDown()
  should be last
* Move tests that directly extend PHPUnit\Framework\TestCase
  to /unit

Change-Id: I1172855c58f4f52a8f624e6d596ec43beb8c93ff
2020-12-24 00:52:06 +00:00

81 lines
2.1 KiB
PHP

<?php
namespace Wikimedia\ParamValidator\Util;
use RecursiveDirectoryIterator;
use RecursiveIteratorIterator;
use Wikimedia\AtEase\AtEase;
abstract class UploadedFileTestBase extends \PHPUnit\Framework\TestCase {
/** @var string|null */
protected static $tmpdir;
public static function setUpBeforeClass() : void {
parent::setUpBeforeClass();
// Create a temporary directory for this test's files.
self::$tmpdir = null;
$base = sys_get_temp_dir() . DIRECTORY_SEPARATOR .
'phpunit-ParamValidator-UploadedFileTest-' . time() . '-' . getmypid() . '-';
for ( $i = 0; $i < 10000; $i++ ) {
$dir = $base . sprintf( '%04d', $i );
if ( AtEase::quietCall( 'mkdir', $dir, 0700, false ) === true ) {
self::$tmpdir = $dir;
break;
}
}
if ( self::$tmpdir === null ) {
self::fail( "Could not create temporary directory '{$base}XXXX'" );
}
}
public static function tearDownAfterClass() : void {
// Clean up temporary directory.
if ( self::$tmpdir !== null ) {
$iter = new RecursiveIteratorIterator(
new RecursiveDirectoryIterator( self::$tmpdir, RecursiveDirectoryIterator::SKIP_DOTS ),
RecursiveIteratorIterator::CHILD_FIRST
);
foreach ( $iter as $file ) {
if ( $file->isDir() ) {
rmdir( $file->getRealPath() );
} else {
unlink( $file->getRealPath() );
}
}
rmdir( self::$tmpdir );
self::$tmpdir = null;
}
parent::tearDownAfterClass();
}
protected static function assertTmpdir() {
if ( self::$tmpdir === null || !is_dir( self::$tmpdir ) ) {
self::fail( 'No temporary directory for ' . static::class );
}
}
/**
* @param string $prefix For tempnam()
* @param string $content Contents of the file
* @return string Filename
*/
protected function makeTemp( $prefix, $content = 'foobar' ) {
self::assertTmpdir();
$filename = tempnam( self::$tmpdir, $prefix );
if ( $filename === false ) {
self::fail( 'Failed to create temporary file' );
}
self::assertSame(
strlen( $content ),
file_put_contents( $filename, $content ),
'Writing test temporary file'
);
return $filename;
}
}