9c7b66984c
File extensions can also be added to the upload type whitelist; they'll be normalized to types for the actual comparison, so only known extensions will work.
97 lines
3.2 KiB
PHP
97 lines
3.2 KiB
PHP
<?php
|
|
|
|
if (isset($_SERVER) && array_key_exists('REQUEST_METHOD', $_SERVER)) {
|
|
print "This script must be run from the command line\n";
|
|
exit();
|
|
}
|
|
|
|
define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
|
|
define('STATUSNET', true);
|
|
define('LACONICA', true);
|
|
|
|
require_once INSTALLDIR . '/lib/common.php';
|
|
|
|
class MediaFileTest extends PHPUnit_Framework_TestCase
|
|
{
|
|
|
|
public function setup()
|
|
{
|
|
$this->old_attachments_supported = common_config('attachments', 'supported');
|
|
$GLOBALS['config']['attachments']['supported'] = true;
|
|
}
|
|
|
|
public function tearDown()
|
|
{
|
|
$GLOBALS['config']['attachments']['supported'] = $this->old_attachments_supported;
|
|
}
|
|
|
|
/**
|
|
* @dataProvider fileTypeCases
|
|
*
|
|
*/
|
|
public function testFileType($filename, $expectedType)
|
|
{
|
|
if (!file_exists($filename)) {
|
|
throw new Exception("WTF? $filename test file missing");
|
|
}
|
|
|
|
$type = MediaFile::getUploadedFileType($filename, basename($filename));
|
|
$this->assertEquals($expectedType, $type);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider fileTypeCases
|
|
*
|
|
*/
|
|
public function testUploadedFileType($filename, $expectedType)
|
|
{
|
|
if (!file_exists($filename)) {
|
|
throw new Exception("WTF? $filename test file missing");
|
|
}
|
|
$tmp = tmpfile();
|
|
fwrite($tmp, file_get_contents($filename));
|
|
|
|
$type = MediaFile::getUploadedFileType($tmp, basename($filename));
|
|
$this->assertEquals($expectedType, $type);
|
|
}
|
|
|
|
static public function fileTypeCases()
|
|
{
|
|
$base = dirname(__FILE__);
|
|
$dir = "$base/sample-uploads";
|
|
$files = array(
|
|
"image.png" => "image/png",
|
|
"image.gif" => "image/gif",
|
|
"image.jpg" => "image/jpeg",
|
|
"image.jpeg" => "image/jpeg",
|
|
|
|
"office.pdf" => "application/pdf",
|
|
|
|
"wordproc.odt" => "application/vnd.oasis.opendocument.text",
|
|
"wordproc.ott" => "application/vnd.oasis.opendocument.text-template",
|
|
"wordproc.doc" => "application/msword",
|
|
"wordproc.docx" => "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
|
"wordproc.rtf" => "text/rtf",
|
|
|
|
"spreadsheet.ods" => "application/vnd.oasis.opendocument.spreadsheet",
|
|
"spreadsheet.ots" => "application/vnd.oasis.opendocument.spreadsheet-template",
|
|
"spreadsheet.xls" => "application/vnd.ms-excel",
|
|
"spreadsheet.xlt" => "application/vnd.ms-excel",
|
|
"spreadsheet.xlsx" => "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|
|
|
"presentation.odp" => "application/vnd.oasis.opendocument.presentation",
|
|
"presentation.otp" => "application/vnd.oasis.opendocument.presentation-template",
|
|
"presentation.ppt" => "application/vnd.ms-powerpoint",
|
|
"presentation.pptx" => "application/vnd.openxmlformats-officedocument.presentationml.presentation",
|
|
);
|
|
|
|
$dataset = array();
|
|
foreach ($files as $file => $type) {
|
|
$dataset[] = array("$dir/$file", $type);
|
|
}
|
|
return $dataset;
|
|
}
|
|
|
|
}
|
|
|