SilverStripe\MimeValidator\Tests\MimeUploadValidatorTest

Complexity

Total Complexity

5

Size/Duplication

Total Lines	93
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	testMimeComparison()	0	12	1
B	testInvalidFileExtensionValidatingMimeType()	0	31	2
B	testGetExpectedMimeTypes()	0	44	2

<?php

namespace SilverStripe\MimeValidator\Tests;

use SilverStripe\Assets\Upload;
use SilverStripe\MimeValidator\MimeUploadValidator;
use SilverStripe\Dev\SapphireTest;
use SilverStripe\Core\Config\Config;

/**
 * Class MimeUploadValidatorTest
 */
class MimeUploadValidatorTest extends SapphireTest
{
    public function testInvalidFileExtensionValidatingMimeType()
    {
        // setup plaintext file with invalid extension
        $tmpFileName = 'UploadTest-testUpload.jpg';
        $tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
        $tmpFileContent = '';

        for ($i = 0; $i < 10000; $i++) {
            $tmpFileContent .= '0';
        }

        file_put_contents($tmpFilePath, $tmpFileContent);

        // emulates the $_FILES array
        $tmpFile = [
            'name' => $tmpFileName,
            'size' => filesize($tmpFilePath),
            'tmp_name' => $tmpFilePath,
            'extension' => 'jpg',
            'error' => UPLOAD_ERR_OK,
        ];

        $upload = Upload::create();
        $upload->setValidator(MimeUploadValidator::create());
        $result = $upload->load($tmpFile);
        $errors = $upload->getErrors();

        $this->assertFalse($result, 'Load failed because file extension does not match excepted MIME type');
        $this->assertEquals('File is not a valid upload', $errors[0]);

        unlink($tmpFilePath);
    }

    public function testGetExpectedMimeTypes()
    {
        // Setup a file with a capitalised extension and try to match it against a lowercase file.
        $tmpFileName = 'text.TXT';
        $tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
        $tmpFileContent = '';

        for ($i = 0; $i < 10000; $i++) {
            $tmpFileContent .= '0';
        }

        file_put_contents($tmpFilePath, $tmpFileContent);

        $validator = MimeUploadValidator::create();
        $tmpFile = [
            'name' => $tmpFileName,
            'tmp_name' => $tmpFilePath,
        ];

        $expected = $validator->getExpectedMimeTypes($tmpFile);
        $this->assertCount(1, $expected);
        $this->assertContains('text/plain', $expected);

        unlink($tmpFilePath);

        // Test a physical ico file with capitalised extension
        $tmpFile = [
            'name' => 'favicon.ICO',
            'tmp_name' => 'assets/favicon.ICO',
        ];

        // Ensure that site configuration doesn't interfere with the test
        $icoMimes = [
            'ico' => [
                'image/vnd.microsoft.icon',
                'image/x-icon',
                'image/x-ico'
            ]
        ];
        Config::modify()->set(MimeUploadValidator::class, 'MimeTypes', $icoMimes);

        $expected = $validator->getExpectedMimeTypes($tmpFile);
        $this->assertCount(3, $expected);
        $this->assertContains('image/x-icon', $expected);
    }

    public function testMimeComparison()
    {
        $validator = MimeUploadValidator::create();

        $this->assertTrue($validator->compareMime('application/xhtml+xml', 'application/xml'));
        $this->assertTrue($validator->compareMime('application/vnd.text', 'application/text'));
        $this->assertTrue($validator->compareMime('application/vnd.vnd.text', 'application/text'));
        $this->assertTrue($validator->compareMime('application/x-text', 'application/text'));
        $this->assertTrue($validator->compareMime('application/gzip', 'application/gzip'));
        $this->assertTrue($validator->compareMime('application/x-gzip', 'application/gzip'));
        $this->assertFalse($validator->compareMime('application/png', 'application/json'));
        $this->assertFalse($validator->compareMime('text/plain', 'text/json'));
    }
}