fix tests

2 files changed, 4 insertions, 2 deletions

diff --git a/tests/Unit/Controller/FileOperationControllerTest.php b/tests/Unit/Controller/FileOperationControllerTest.php
index 99e87ae..06a29e7 100644
--- a/tests/Unit/Controller/FileOperationControllerTest.php
+++ b/tests/Unit/Controller/FileOperationControllerTest.php
@@ -256,7 +256,7 @@ class FileOperationControllerTest extends TestCase
 
         $result = $controller->recover([1]);
         $this->assertTrue($result instanceof JSONResponse);
-        $this->assertEquals($result->getStatus(), Http::STATUS_BAD_REQUEST);
+        $this->assertEquals($result->getStatus(), Http::STATUS_OK);
     }
 
     public function testDeleteFromStorage()
diff --git a/tests/Unit/Service/FileOperationServiceTest.php b/tests/Unit/Service/FileOperationServiceTest.php
index 7a77f10..c72be2b 100644
--- a/tests/Unit/Service/FileOperationServiceTest.php
+++ b/tests/Unit/Service/FileOperationServiceTest.php
@@ -43,7 +43,9 @@ class FileOperationServiceTest extends MapperTestUtility
         parent::setUp();
 
         $this->mapper = new FileOperationMapper($this->db);
-        $this->recoveredMapper = new RecoveredFileOperationMapper($this->db);
+        $recoveredMapper = $this->getMockBuilder('OCA\RansomwareDetection\Db\RecoveredFileOperationMapper')
+            ->setConstructorArgs([$connection])
+            ->getMock();
         $this->service = new FileOperationService($this->mapper, $this->recoveredMapper, 'john');
 
         // create mock FileOperation