From 8344b7ccafcf98e7376c53a0a79708c51daa6d0a Mon Sep 17 00:00:00 2001
From: deiflaender <dominique.eiflaender@iqser.com>
Date: Wed, 12 Aug 2020 15:46:33 +0200
Subject: [PATCH] RED-242: Return sectionNumber for tests

---
 .../iqser/red/service/redaction/v1/model/RedactionLogEntry.java  | 1 +
 .../server/visualization/service/AnnotationHighlightService.java | 1 +
 2 files changed, 2 insertions(+)

diff --git a/redaction-service-v1/redaction-service-api-v1/src/main/java/com/iqser/red/service/redaction/v1/model/RedactionLogEntry.java b/redaction-service-v1/redaction-service-api-v1/src/main/java/com/iqser/red/service/redaction/v1/model/RedactionLogEntry.java
index e75b44e9..90c6c263 100644
--- a/redaction-service-v1/redaction-service-api-v1/src/main/java/com/iqser/red/service/redaction/v1/model/RedactionLogEntry.java
+++ b/redaction-service-v1/redaction-service-api-v1/src/main/java/com/iqser/red/service/redaction/v1/model/RedactionLogEntry.java
@@ -17,5 +17,6 @@ public class RedactionLogEntry {
     private String section;
     private float[] color;
     private List<Rectangle> positions = new ArrayList<>();
+    private int sectionNumber;
 
 }
diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java
index 0b7b8090..baa7bf86 100644
--- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java
+++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java
@@ -167,6 +167,7 @@ public class AnnotationHighlightService {
                 redactionLogEntry.setSection(entity.getHeadline());
                 redactionLogEntry.setHint(isHint(entity));
                 classifiedDoc.getRedactionLogEntities().add(redactionLogEntry);
+                redactionLogEntry.setSectionNumber(entity.getSectionNumber());
             }
 
         }