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); + } + +}