From a19e52ffd70e06c78e94eac3fee17198de24b666 Mon Sep 17 00:00:00 2001
From: Milen Marinov <milen.danchev.marinov@gmail.com>
Date: Tue, 23 Apr 2024 18:38:25 +0300
Subject: [PATCH] [MS-425] Return JSON path for attendantId and moduleId
 tokenizedFields

---
 .../ApiEventDownSyncRequestPayload.kt         |  7 +++-
 .../ApiEventDownSyncRequestPayloadTest.kt     | 35 +++++++++++++++++++
 2 files changed, 41 insertions(+), 1 deletion(-)
 create mode 100644 infra/event-sync/src/test/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayloadTest.kt

diff --git a/infra/event-sync/src/main/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayload.kt b/infra/event-sync/src/main/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayload.kt
index 61fbfa4de4..936f9bf7fc 100644
--- a/infra/event-sync/src/main/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayload.kt
+++ b/infra/event-sync/src/main/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayload.kt
@@ -45,5 +45,10 @@ internal data class ApiEventDownSyncRequestPayload(
         val lastEventId: String?,
     )
 
-    override fun getTokenizedFieldJsonPath(tokenKeyType: TokenKeyType): String? = null
+    override fun getTokenizedFieldJsonPath(tokenKeyType: TokenKeyType): String? =
+        when (tokenKeyType) {
+            TokenKeyType.AttendantId -> "queryParameters.attendantId"
+            TokenKeyType.ModuleId -> "queryParameters.moduleId"
+            else -> null
+        }
 }
diff --git a/infra/event-sync/src/test/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayloadTest.kt b/infra/event-sync/src/test/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayloadTest.kt
new file mode 100644
index 0000000000..c6c9e70ab1
--- /dev/null
+++ b/infra/event-sync/src/test/java/com/simprints/infra/eventsync/event/remote/models/downsync/ApiEventDownSyncRequestPayloadTest.kt
@@ -0,0 +1,35 @@
+package com.simprints.infra.eventsync.event.remote.models.downsync
+
+import com.simprints.infra.config.store.models.TokenKeyType
+import io.mockk.mockk
+import org.junit.Assert.assertEquals
+import org.junit.Test
+
+class ApiEventDownSyncRequestPayloadTest {
+
+    @Test
+    fun testGetTokenizedFieldJsonPath() {
+        // Arrange
+        val payload = ApiEventDownSyncRequestPayload(
+            startTime = mockk(),
+            endTime = mockk(),
+            requestId = "requestId",
+            queryParameters = ApiEventDownSyncRequestPayload.ApiQueryParameters(
+                moduleId = "moduleId",
+                attendantId = "attendantId",
+                subjectId = null,
+                modes = null,
+                lastEventId = null,
+            ),
+            responseStatus = 200,
+            errorType = null,
+            msToFirstResponseByte = 1000L,
+            eventsRead = 10
+        )
+
+        // Act & Assert
+        assertEquals("queryParameters.attendantId", payload.getTokenizedFieldJsonPath(TokenKeyType.AttendantId))
+        assertEquals("queryParameters.moduleId", payload.getTokenizedFieldJsonPath(TokenKeyType.ModuleId))
+        assertEquals(null, payload.getTokenizedFieldJsonPath(TokenKeyType.Unknown))
+    }
+}
\ No newline at end of file