VirusScannerPreProcessorTestTBD.php

Go to the documentation of this file.

<?php

declare(strict_types=1);

namespace ILIAS\VirusScanner\tests;

use ILIAS\Filesystem\Stream\Streams;
use ILIAS\FileUpload\DTO\Metadata;
use ILIAS\FileUpload\DTO\ProcessingStatus;
use PHPUnit\Framework\TestCase;
use ilVirusScannerPreProcessor;
use Mockery;

class VirusScannerPreProcessorTest extends TestCase
{
    use Mockery\Adapter\Phpunit\MockeryPHPUnitIntegration;

    public function testVirusDetected(): void
    {
        $stream = Streams::ofString('Awesome stuff');
        $mock = $this->getMockBuilder(\ilVirusScanner::class)
                     ->disableOriginalConstructor()
                     ->getMock();
        $mock->expects($this->once())->method('scanFile')->with($stream->getMetadata('uri'))->willReturn(
            'Virus found!!!'
        );

        $subject = new ilVirusScannerPreProcessor($mock);
        $result = $subject->process(
            $stream,
            new Metadata('MyVirus.exe', $stream->getSize(), 'application/vnd.microsoft.portable-executable')
        );
        $this->assertSame(ProcessingStatus::DENIED, $result->getCode());
        $this->assertSame('Virus detected.', $result->getMessage());
    }

    public function testNoVirusDetected(): void
    {
        $stream = Streams::ofString('Awesome stuff');

        $mock = $this->getMockBuilder(\ilVirusScanner::class)
                     ->disableOriginalConstructor()
                     ->getMock();
        $mock->expects($this->once())->method('scanFile')->with($stream->getMetadata('uri'))->willReturn('');

        $subject = new ilVirusScannerPreProcessor($mock);
        $result = $subject->process(
            $stream,
            new Metadata('MyVirus.exe', $stream->getSize(), 'application/vnd.microsoft.portable-executable')
        );
        $this->assertSame(ProcessingStatus::OK, $result->getCode());
    }
}