Fix unit test

Signed-off-by: Vitor Mattos <[email protected]>
LibreSign · Mar 28, 2024 · 322c2ae · 322c2ae
1 parent 63a7978
commit 322c2ae
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 33 deletions.
diff --git a/composer.lock b/composer.lock
diff --git a/lib/Handler/TCPDFLibresign.php b/lib/Handler/TCPDFLibresign.php
@@ -46,33 +46,4 @@ protected function _textstring($s, $n = 0) {
 		}
 		return parent::_textstring($s, $n);
 	}
-
-	/**
-	 * @psalm-return array{w?: mixed, h?: mixed}
-	 */
-	public function getPageTplDimension(int $pageNum): array {
-		if (!$this->tpls) {
-			return [];
-		}
-		return [
-			'w' => $this->tpls[$pageNum]['w'],
-			'h' => $this->tpls[$pageNum]['h']
-		];
-	}
-
-	public function getPagesMetadata(): array {
-		$pageCount = current($this->parsers)->getPageCount();
-		$data = [
-			'p' => $pageCount
-		];
-		for ($pageNo = 1; $pageNo <= $pageCount; $pageNo++) {
-			$dimensions = $this->getPageTplDimension($pageNo);
-			if (empty($dimensions['w'])) {
-				$this->importPage($pageNo);
-				$dimensions = $this->getPageTplDimension($pageNo);
-			}
-			$data['d'][] = $dimensions;
-		}
-		return $data;
-	}
 }
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) {