diff --git a/ocr-service-v1/ocr-service-server-v1/src/test/java/com/iqser/red/service/ocr/v1/server/service/WatermarkRemovalServiceTest.java b/ocr-service-v1/ocr-service-server-v1/src/test/java/com/iqser/red/service/ocr/v1/server/service/WatermarkRemovalServiceTest.java new file mode 100644 index 0000000..1fe26ff --- /dev/null +++ b/ocr-service-v1/ocr-service-server-v1/src/test/java/com/iqser/red/service/ocr/v1/server/service/WatermarkRemovalServiceTest.java @@ -0,0 +1,33 @@ +package com.iqser.red.service.ocr.v1.server.service; + +import static org.junit.jupiter.api.Assertions.*; + +import java.io.FileOutputStream; + +import org.junit.jupiter.api.Test; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.core.io.ClassPathResource; + +import com.iqser.red.service.ocr.v1.server.AbstractTest; +import com.iqser.red.service.ocr.v1.server.utils.OsUtils; + +import lombok.SneakyThrows; + +class WatermarkRemovalServiceTest extends AbstractTest { + + @Autowired + private WatermarkRemovalService watermarkRemovalService; + + + @Test + @SneakyThrows + public void removeWatermarksTest() { + + String filename = "files/403-17_Fantom_ToxicidadeInalatoriaAguda.pdf"; + try (var in = new ClassPathResource(filename).getInputStream(); var out = new FileOutputStream(OsUtils.createTmpFileName(filename, "WATERMARK_REMOVED"))) { + watermarkRemovalService.removeWatermarks(in, out); + } + } + + +} \ No newline at end of file diff --git a/ocr-service-v1/ocr-service-server-v1/src/test/java/com/iqser/red/service/ocr/v1/server/utils/OsUtils.java b/ocr-service-v1/ocr-service-server-v1/src/test/java/com/iqser/red/service/ocr/v1/server/utils/OsUtils.java index ccbff74..05cfab6 100644 --- a/ocr-service-v1/ocr-service-server-v1/src/test/java/com/iqser/red/service/ocr/v1/server/utils/OsUtils.java +++ b/ocr-service-v1/ocr-service-server-v1/src/test/java/com/iqser/red/service/ocr/v1/server/utils/OsUtils.java @@ -1,5 +1,8 @@ package com.iqser.red.service.ocr.v1.server.utils; +import java.nio.file.Path; +import java.util.stream.Stream; + import org.apache.commons.lang3.StringUtils; import lombok.SneakyThrows; @@ -55,4 +58,8 @@ public final class OsUtils { return "/tmp"; } + public static String createTmpFileName(String filename, String suffix) { + + return Path.of(OsUtils.getTemporaryDirectory()).resolve(Path.of(filename).getFileName()).toString().replace(".pdf", "_" + suffix + ".pdf"); + } }