From beb1e8b6b143a4ee99b602bf15607358a3245a29 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kilian=20Sch=C3=BCttler?= Date: Thu, 22 Jun 2023 18:05:08 +0200 Subject: [PATCH] RED-6929: fix acceptance tests/rules --- .../graph/entity/RedactionEntity.java | 2 +- .../document/entity/RedactionEntityTest.java | 25 +++++++++++++++++++ 2 files changed, 26 insertions(+), 1 deletion(-) create mode 100644 redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/document/entity/RedactionEntityTest.java diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/layoutparsing/document/graph/entity/RedactionEntity.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/layoutparsing/document/graph/entity/RedactionEntity.java index 51ce6c09..a6c6db6d 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/layoutparsing/document/graph/entity/RedactionEntity.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/layoutparsing/document/graph/entity/RedactionEntity.java @@ -141,7 +141,7 @@ public class RedactionEntity { if (matchedRules.isEmpty()) { return ""; } - return matchedRules.getFirst(); + return matchedRules.getLast(); } diff --git a/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/document/entity/RedactionEntityTest.java b/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/document/entity/RedactionEntityTest.java new file mode 100644 index 00000000..18b7c3d8 --- /dev/null +++ b/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/document/entity/RedactionEntityTest.java @@ -0,0 +1,25 @@ +package com.iqser.red.service.redaction.v1.server.document.entity; + +import static org.assertj.core.api.Assertions.assertThat; + +import org.junit.jupiter.api.Test; + +import com.iqser.red.service.redaction.v1.server.layoutparsing.document.graph.Boundary; +import com.iqser.red.service.redaction.v1.server.layoutparsing.document.graph.entity.EntityType; +import com.iqser.red.service.redaction.v1.server.layoutparsing.document.graph.entity.RedactionEntity; + +public class RedactionEntityTest { + + @Test + public void testMatchedRule() { + + RedactionEntity entity = RedactionEntity.initialEntityNode(new Boundary(1, 100), "PII", EntityType.ENTITY); + entity.addMatchedRule("CBI.1.0"); + entity.addMatchedRule("CBI.2.0"); + entity.addMatchedRule("CBI.3.0"); + entity.addMatchedRule("CBI.4.0"); + assertThat(entity.getMatchedRule()).isEqualTo("CBI.4.0"); + assertThat(entity.getMatchedRuleUnit()).isEqualTo(4); + } + +}