diff options
Diffstat (limited to 'tests/Unit/BackgroundJob/CleanUpJobTest.php')
-rw-r--r-- | tests/Unit/BackgroundJob/CleanUpJobTest.php | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/Unit/BackgroundJob/CleanUpJobTest.php b/tests/Unit/BackgroundJob/CleanUpJobTest.php index d5018c2..4b41370 100644 --- a/tests/Unit/BackgroundJob/CleanUpJobTest.php +++ b/tests/Unit/BackgroundJob/CleanUpJobTest.php @@ -23,6 +23,7 @@ namespace OCA\RansomwareDetection\tests\Unit\BackgroundJob; use OCA\RansomwareDetection\BackgroundJob\CleanUpJob; use OCA\RansomwareDetection\Db\FileOperationMapper; +use OCA\RansomwareDetection\Db\RecoveredFileOperationMapper; use OCA\RansomwareDetection\Service\FileOperationService; use OCA\RansomwareDetection\Tests\Unit\Db\MapperTestUtility; use OCP\IConfig; @@ -46,8 +47,12 @@ class CleanUpJobTest extends MapperTestUtility ->setMethods(['deleteFileOperationsBefore']) ->setConstructorArgs([$this->db]) ->getMock(); + $recoveredFileOperationMapper = $this->getMockBuilder(RecoveredFileOperationMapper::class) + ->setMethods(['deleteFileOperationsBefore']) + ->setConstructorArgs([$this->db]) + ->getMock(); $this->fileOperationService = $this->getMockBuilder(FileOperationService::class) - ->setConstructorArgs([$fileOperationMapper, 'john']) + ->setConstructorArgs([$fileOperationMapper, $recoveredFileOperationMapper, 'john']) ->getMock(); $this->config = $this->createMock(IConfig::class); $this->cleanUpJob = new CleanUpJob($this->fileOperationService, $this->config); |