From ecb0c3b19bfaa31f124c8dbc6eb0ca36b29debd4 Mon Sep 17 00:00:00 2001
From: cigerhed <christoffer.sigerhed@soprasteria.com>
Date: Fri, 11 Oct 2024 11:16:26 +0200
Subject: [PATCH] K1J-330: CertificateStatusUpdateForCareType should include
 sent and received questions.

---
 .../services/NotificationTransformer.java     |  2 +-
 .../CertificateStatusUpdateFactory.java       |  8 +--
 .../NotificationMessageFactory.java           | 13 +++-
 .../ProcessIncomingMessageService.java        |  2 +-
 .../util/FragestallareConverterUtilTest.java  |  9 ++-
 .../NotificationMessageFactoryTest.java       | 62 ++++++++++++-------
 6 files changed, 63 insertions(+), 33 deletions(-)

diff --git a/notification-sender/src/main/java/se/inera/intyg/webcert/notification_sender/notifications/services/NotificationTransformer.java b/notification-sender/src/main/java/se/inera/intyg/webcert/notification_sender/notifications/services/NotificationTransformer.java
index 3d8d9cfa04..3811f420c5 100644
--- a/notification-sender/src/main/java/se/inera/intyg/webcert/notification_sender/notifications/services/NotificationTransformer.java
+++ b/notification-sender/src/main/java/se/inera/intyg/webcert/notification_sender/notifications/services/NotificationTransformer.java
@@ -136,4 +136,4 @@ private String resolveIntygTypeVersion(String intygsTyp, Message message, String
         message.setHeader(INTYG_TYPE_VERSION, certificateVersion);
         return certificateVersion;
     }
-}
+}
\ No newline at end of file
diff --git a/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/CertificateStatusUpdateFactory.java b/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/CertificateStatusUpdateFactory.java
index 82e6fa31a5..d572bcdbf8 100644
--- a/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/CertificateStatusUpdateFactory.java
+++ b/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/CertificateStatusUpdateFactory.java
@@ -50,7 +50,7 @@ private CertificateStatusUpdateFactory() {
     }
 
     public static byte[] create(String encodedXmlRepresentation, HandelsekodEnum eventType,
-        LocalDateTime now, String handledByHsaId, String reference) {
+        LocalDateTime now, String handledByHsaId, String reference, ArendeCount sentQuestions, ArendeCount recievedQuestions) {
         final var request = getRegisterCertificateType(encodedXmlRepresentation);
 
         final var certificateStatusUpdateForCareType = new CertificateStatusUpdateForCareType();
@@ -58,8 +58,8 @@ public static byte[] create(String encodedXmlRepresentation, HandelsekodEnum eve
         certificateStatusUpdateForCareType.setHandelse(
             NotificationRedeliveryUtil.getEventV3(eventType, now, null, null)
         );
-        certificateStatusUpdateForCareType.setSkickadeFragor(NotificationTypeConverter.toArenden(ArendeCount.getEmpty()));
-        certificateStatusUpdateForCareType.setMottagnaFragor(NotificationTypeConverter.toArenden(ArendeCount.getEmpty()));
+        certificateStatusUpdateForCareType.setSkickadeFragor(NotificationTypeConverter.toArenden(sentQuestions));
+        certificateStatusUpdateForCareType.setMottagnaFragor(NotificationTypeConverter.toArenden(recievedQuestions));
         certificateStatusUpdateForCareType.setHanteratAv(
             NotificationRedeliveryUtil.getIIType(new HsaId(), handledByHsaId, HSA_ID_OID)
         );
@@ -99,4 +99,4 @@ private static byte[] convertToByteArray(JAXBElement<CertificateStatusUpdateForC
             throw new IllegalStateException(e);
         }
     }
-}
+}
\ No newline at end of file
diff --git a/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactory.java b/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactory.java
index 807cb3c33d..cc935f378d 100644
--- a/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactory.java
+++ b/web/src/main/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactory.java
@@ -58,10 +58,17 @@ public NotificationMessage create(Certificate certificate, String encodedXmlRepr
         );
 
         notificationMessage.setStatusUpdateXml(
-            CertificateStatusUpdateFactory.create(encodedXmlRepresentation, eventType, now, handledByHsaId,
-                certificate.getMetadata().getExternalReference())
+            CertificateStatusUpdateFactory.create(
+                encodedXmlRepresentation,
+                eventType,
+                now,
+                handledByHsaId,
+                certificate.getMetadata().getExternalReference(),
+                notificationMessage.getSkickadeFragor(),
+                notificationMessage.getMottagnaFragor()
+            )
         );
 
         return notificationMessage;
     }
-}
+}
\ No newline at end of file
diff --git a/web/src/main/java/se/inera/intyg/webcert/web/csintegration/message/ProcessIncomingMessageService.java b/web/src/main/java/se/inera/intyg/webcert/web/csintegration/message/ProcessIncomingMessageService.java
index 8b8fc39630..1b091a87a3 100644
--- a/web/src/main/java/se/inera/intyg/webcert/web/csintegration/message/ProcessIncomingMessageService.java
+++ b/web/src/main/java/se/inera/intyg/webcert/web/csintegration/message/ProcessIncomingMessageService.java
@@ -104,4 +104,4 @@ private static HandelsekodEnum getEventType(ArendeAmne questionType, boolean isA
     private boolean unitIsIntegrated(Certificate certificate) {
         return integreradeEnheterRegistry.getIntegreradEnhet(certificate.getMetadata().getUnit().getUnitId()) != null;
     }
-}
+}
\ No newline at end of file
diff --git a/web/src/test/java/se/inera/intyg/webcert/web/converter/util/FragestallareConverterUtilTest.java b/web/src/test/java/se/inera/intyg/webcert/web/converter/util/FragestallareConverterUtilTest.java
index dddb8c9c9b..0eb5e18cc7 100644
--- a/web/src/test/java/se/inera/intyg/webcert/web/converter/util/FragestallareConverterUtilTest.java
+++ b/web/src/test/java/se/inera/intyg/webcert/web/converter/util/FragestallareConverterUtilTest.java
@@ -19,6 +19,7 @@
 package se.inera.intyg.webcert.web.converter.util;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 
 import org.junit.Test;
 import se.inera.intyg.webcert.common.service.exception.WebCertServiceException;
@@ -37,8 +38,14 @@ public void testPartCodeToFrageStallareKod() {
             .partToFrageStallarKod(PARTCODE_HSVARD));
     }
 
+    @Test
+    public void testPartNameToFrageStallareNamn() {
+        assertTrue(FrageStallare.FORSAKRINGSKASSAN.isNameEqual("Försäkringskassan"));
+        assertTrue(FrageStallare.WEBCERT.isNameEqual("Webcert"));
+    }
+
     @Test(expected = WebCertServiceException.class)
     public void testInvalidPartKod() {
         FragestallareConverterUtil.partToFrageStallarKod("INVALID");
     }
-}
+}
\ No newline at end of file
diff --git a/web/src/test/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactoryTest.java b/web/src/test/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactoryTest.java
index ef68132ed0..e39a20ed5f 100644
--- a/web/src/test/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactoryTest.java
+++ b/web/src/test/java/se/inera/intyg/webcert/web/csintegration/certificate/NotificationMessageFactoryTest.java
@@ -23,6 +23,7 @@
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.mockito.Mockito.doReturn;
 import static se.inera.intyg.common.support.Constants.KV_HANDELSE_CODE_SYSTEM;
+import static se.inera.intyg.webcert.notification_sender.notifications.services.NotificationTypeConverter.toArenden;
 
 import java.io.StringWriter;
 import java.nio.charset.StandardCharsets;
@@ -69,6 +70,7 @@ class NotificationMessageFactoryTest {
     private static final Intyg EXPECTED_INTYG = new Intyg();
     private static final String CERTIFICATE_ID = "certificateId";
     private static final String HSA_ID = "hsaId";
+    private static final ArendeCount ARENDE_COUNT = new ArendeCount(1, 1, 1, 1);
     private Certificate certificate;
     private String xmlRepresentation;
     private HandelsekodEnum eventType;
@@ -113,6 +115,12 @@ void setUp() {
         registerCertificateType.setIntyg(EXPECTED_INTYG);
         final var marshall = marshall(registerCertificateType);
         xmlRepresentation = Base64.getEncoder().encodeToString(marshall.getBytes(StandardCharsets.UTF_8));
+
+        final var questions = List.of(Question.builder().build());
+
+        doReturn(questions).when(csIntegrationService).getQuestions(ID);
+        doReturn(ARENDE_COUNT).when(questionCounter).calculateArendeCount(questions, FrageStallare.WEBCERT);
+        doReturn(ARENDE_COUNT).when(questionCounter).calculateArendeCount(questions, FrageStallare.FORSAKRINGSKASSAN);
     }
 
     @Test
@@ -156,34 +164,20 @@ void shallConvertFragorOchSvar() {
 
     @Test
     void shallConvertMottagnaFragor() {
-        final var questions = List.of(Question.builder().build());
-        final var expectedArendeCount = new ArendeCount(1, 1, 1, 1);
-
-        doReturn(questions).when(csIntegrationService).getQuestions(ID);
-        doReturn(new ArendeCount()).when(questionCounter).calculateArendeCount(questions, FrageStallare.WEBCERT);
-        doReturn(expectedArendeCount).when(questionCounter).calculateArendeCount(questions, FrageStallare.FORSAKRINGSKASSAN);
-
         final var result = converter.create(certificate, xmlRepresentation, eventType, HSA_ID);
-
-        assertEquals(expectedArendeCount.getBesvarade(), result.getMottagnaFragor().getBesvarade());
-        assertEquals(expectedArendeCount.getHanterade(), result.getMottagnaFragor().getHanterade());
-        assertEquals(expectedArendeCount.getEjBesvarade(), result.getMottagnaFragor().getEjBesvarade());
-        assertEquals(expectedArendeCount.getTotalt(), result.getMottagnaFragor().getTotalt());
+        assertEquals(ARENDE_COUNT.getBesvarade(), result.getSkickadeFragor().getBesvarade());
+        assertEquals(ARENDE_COUNT.getHanterade(), result.getSkickadeFragor().getHanterade());
+        assertEquals(ARENDE_COUNT.getEjBesvarade(), result.getSkickadeFragor().getEjBesvarade());
+        assertEquals(ARENDE_COUNT.getTotalt(), result.getSkickadeFragor().getTotalt());
     }
 
     @Test
     void shallConvertSkickadeFragor() {
-        final var questions = List.of(Question.builder().build());
-        final var expectedArendeCount = new ArendeCount(1, 1, 1, 1);
-
-        doReturn(questions).when(csIntegrationService).getQuestions(ID);
-        doReturn(expectedArendeCount).when(questionCounter).calculateArendeCount(questions, FrageStallare.WEBCERT);
-
         final var result = converter.create(certificate, xmlRepresentation, eventType, HSA_ID);
-        assertEquals(expectedArendeCount.getBesvarade(), result.getSkickadeFragor().getBesvarade());
-        assertEquals(expectedArendeCount.getHanterade(), result.getSkickadeFragor().getHanterade());
-        assertEquals(expectedArendeCount.getEjBesvarade(), result.getSkickadeFragor().getEjBesvarade());
-        assertEquals(expectedArendeCount.getTotalt(), result.getSkickadeFragor().getTotalt());
+        assertEquals(ARENDE_COUNT.getBesvarade(), result.getSkickadeFragor().getBesvarade());
+        assertEquals(ARENDE_COUNT.getHanterade(), result.getSkickadeFragor().getHanterade());
+        assertEquals(ARENDE_COUNT.getEjBesvarade(), result.getSkickadeFragor().getEjBesvarade());
+        assertEquals(ARENDE_COUNT.getTotalt(), result.getSkickadeFragor().getTotalt());
     }
 
     @Test
@@ -242,6 +236,28 @@ void shallIncludeRef() {
             final var careType = unmarshall(result.getStatusUpdateXml());
             assertEquals(EXTERNAL_REF, careType.getRef());
         }
+
+        @Test
+        void shallIncludeSentQuestions() {
+            final var arenden = toArenden(ARENDE_COUNT);
+            final var result = converter.create(certificate, xmlRepresentation, eventType, HSA_ID);
+            final var careType = unmarshall(result.getStatusUpdateXml());
+            assertEquals(arenden.getBesvarade(), careType.getSkickadeFragor().getBesvarade());
+            assertEquals(arenden.getHanterade(), careType.getSkickadeFragor().getHanterade());
+            assertEquals(arenden.getEjBesvarade(), careType.getSkickadeFragor().getEjBesvarade());
+            assertEquals(arenden.getTotalt(), careType.getSkickadeFragor().getTotalt());
+        }
+
+        @Test
+        void shallIncludeRecievedQuestions() {
+            final var arenden = toArenden(ARENDE_COUNT);
+            final var result = converter.create(certificate, xmlRepresentation, eventType, HSA_ID);
+            final var careType = unmarshall(result.getStatusUpdateXml());
+            assertEquals(arenden.getBesvarade(), careType.getSkickadeFragor().getBesvarade());
+            assertEquals(arenden.getHanterade(), careType.getSkickadeFragor().getHanterade());
+            assertEquals(arenden.getEjBesvarade(), careType.getSkickadeFragor().getEjBesvarade());
+            assertEquals(arenden.getTotalt(), careType.getSkickadeFragor().getTotalt());
+        }
     }
 
 
@@ -265,4 +281,4 @@ private static CertificateStatusUpdateForCareType unmarshall(byte[] registerCert
         final var type = XmlMarshallerHelper.unmarshal(new String(registerCertificateType, StandardCharsets.UTF_8));
         return (CertificateStatusUpdateForCareType) type.getValue();
     }
-}
+}
\ No newline at end of file