Fix unit test

Signed-off-by: Vitor Mattos <[email protected]>
LibreSign · Mar 27, 2024 · c71d917 · c71d917
1 parent d0932e9
commit c71d917
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/tests/Unit/Handler/Pkcs12HandlerTest.php b/tests/Unit/Handler/Pkcs12HandlerTest.php
@@ -107,7 +107,8 @@ public function testGetFooterWithoutValidationSite() {
 			$this->pdfParserService,
 		);
 		$file = $this->createMock(\OCP\Files\File::class);
-		$actual = $this->pkcs12Handler->getFooter($file, 'uuid');
+		$libresignFile = $this->createMock(\OCA\Libresign\Db\File::class);
+		$actual = $this->pkcs12Handler->getFooter($file, $libresignFile);
 		$this->assertEmpty($actual);
 	}
 
@@ -141,8 +142,24 @@ public function testGetFooterWithSuccess() {
 			->willReturn('small_valid.pdf');
 		$file->method('getContent')
 			->willReturn(file_get_contents(__DIR__ . '/../../fixtures/small_valid.pdf'));
-		$actual = $this->pkcs12Handler->getFooter($file, 'uuid');
-		$this->assertEquals(18615, strlen($actual));
+		$libresignFile = $this->createMock(\OCA\Libresign\Db\File::class);
+		$libresignFile
+			->method('__call')
+			->willReturnCallback(function ($key, $default) {
+				switch ($key) {
+					case 'getMetadata': return [
+						'd' => [
+							[
+								'w' => 100,
+								'h' => 100,
+							],
+						],
+					];
+					case 'getUuid': return 'uuid';
+				}
+			});
+		$actual = $this->pkcs12Handler->getFooter($file, $libresignFile);
+		$this->assertEquals(7655, strlen($actual));
 	}
 
 	public function cfsslHandlerCallbackToGetSetArguments($functionName, $value = null) {