VirusScannerPreProcessorTest.php

Go to the documentation of this file.

<?php
 
namespace ILIAS\FileUpload\Processor;
 
require_once('./libs/composer/vendor/autoload.php');
 
use ILIAS\Filesystem\Stream\Streams;
use ILIAS\FileUpload\DTO\Metadata;
use ILIAS\FileUpload\DTO\ProcessingStatus;
use Mockery;
use PHPUnit\Framework\TestCase;
 
class VirusScannerPreProcessorTest extends TestCase
{
    use Mockery\Adapter\Phpunit\MockeryPHPUnitIntegration;
 
 
    public function testVirusDetected()
    {
        $stream = Streams::ofString('Awesome stuff');
        $mock = Mockery::mock(\ilVirusScanner::class);
 
        $mock->shouldReceive("scanFile")->once()->withArgs(array( $stream->getMetadata('uri') ))->andReturn("Virus found!!!");
 
        $subject = new VirusScannerPreProcessor($mock);
        $result = $subject->process($stream, new Metadata("MyVirus.exe", $stream->getSize(), 'application/vnd.microsoft.portable-executable'));
        $this->assertSame(ProcessingStatus::REJECTED, $result->getCode());
        $this->assertSame('Virus detected.', $result->getMessage());
    }
 
 
    public function testNoVirusDetected()
    {
        $stream = Streams::ofString('Awesome stuff');
        $mock = Mockery::mock(\ilVirusScanner::class);
 
        $mock->shouldReceive("scanFile")->once()->withArgs(array( $stream->getMetadata('uri') ))->andReturn("");
 
        $subject = new VirusScannerPreProcessor($mock);
        $result = $subject->process($stream, new Metadata("MyVirus.exe", $stream->getSize(), 'application/vnd.microsoft.portable-executable'));
        $this->assertSame(ProcessingStatus::OK, $result->getCode());
    }
}