From e1adcfb02c4c7d46140884b3d38f8daf1e1aae92 Mon Sep 17 00:00:00 2001 From: deiflaender Date: Tue, 29 Sep 2020 14:54:35 +0200 Subject: [PATCH] Added missing unittest --- .../service/RulingCleaningService.java | 4 ++-- .../v1/server/RedactionIntegrationTest.java | 22 +++++++++++++++++++ 2 files changed, 24 insertions(+), 2 deletions(-) diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/tableextraction/service/RulingCleaningService.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/tableextraction/service/RulingCleaningService.java index facef878..1d3d81f2 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/tableextraction/service/RulingCleaningService.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/tableextraction/service/RulingCleaningService.java @@ -18,9 +18,9 @@ import com.iqser.red.service.redaction.v1.server.tableextraction.utils.Utils; @Service public class RulingCleaningService { - public CleanRulings getCleanRulings(List rulings, float avgCharWidth, float avgCharHeight){ + public CleanRulings getCleanRulings(List rulings, float maxCharWidth, float maxCharHeight){ if (!rulings.isEmpty()) { - snapPoints(rulings, avgCharWidth , avgCharHeight); + snapPoints(rulings, maxCharWidth , maxCharHeight); } List vrs = new ArrayList<>(); diff --git a/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/RedactionIntegrationTest.java b/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/RedactionIntegrationTest.java index a1adb27d..4a6a3491 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/RedactionIntegrationTest.java +++ b/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/RedactionIntegrationTest.java @@ -1,5 +1,6 @@ package com.iqser.red.service.redaction.v1.server; +import static org.assertj.core.api.Assertions.assertThat; import static org.mockito.Mockito.when; import static org.springframework.boot.test.context.SpringBootTest.WebEnvironment.DEFINED_PORT; @@ -421,6 +422,27 @@ public class RedactionIntegrationTest { } + @Test + public void phantomCellsDocumentTest() throws IOException { + + ClassPathResource pdfFileResource = new ClassPathResource("files/Minimal Examples/Phantom Cells.pdf"); + + RedactionRequest request = RedactionRequest.builder() + .document(IOUtils.toByteArray(pdfFileResource.getInputStream())) + .build(); + request.setFlatRedaction(false); + + RedactionResult result = redactionController.redact(request); + + result.getRedactionLog().getRedactionLogEntry().forEach(entry -> { + if(!entry.isHint()){ + assertThat(entry.getReason()).isEqualTo("Not redacted because row is not a vertebrate study"); + } + }); + } + + + private static String loadFromClassPath(String path) { URL resource = ResourceLoader.class.getClassLoader().getResource(path);