Javalin context object test cases use HashMap<String, List<String>>

AikidoSec · Feb 28, 2025 · 069d3e7 · 069d3e7
1 parent 7603bd1
commit 069d3e7
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/agent_api/src/test/java/context/JavalinContextObjectTest.java b/agent_api/src/test/java/context/JavalinContextObjectTest.java
@@ -4,6 +4,7 @@
 import dev.aikido.agent_api.context.RouteMetadata;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import org.springframework.expression.spel.ast.Literal;
 
 import java.util.*;
 
@@ -20,8 +21,8 @@ void setUp() {
         String rawIp = "192.168.1.1";
         Map<String, List<String>> queryParams = new HashMap<>();
         queryParams.put("param1", List.of("value1"));
-        Map<String, String> cookies = new HashMap<>();
-        cookies.put("sessionId", "abc123");
+        HashMap<String, List<String>> cookies = new HashMap<>();
+        cookies.put("sessionId", List.of("abc123", "456"));
         Map<String, String> headers = new HashMap<>();
         headers.put("Content-Type", "application/json");
 
@@ -38,6 +39,8 @@ void testConstructor() {
         assertEquals("value1", contextObject.getQuery().get("param1").get(0));
         assertEquals(1, contextObject.getCookies().size());
         assertEquals("abc123", contextObject.getCookies().get("sessionId").get(0));
+        assertEquals("456", contextObject.getCookies().get("sessionId").get(1));
+
     }
 
     @Test
@@ -100,9 +103,9 @@ void testCookiesExtraction() {
     @Test
     void testMultipleCookiesExtraction() {
         // Test with multiple cookies
-        Map<String, String> cookies = new HashMap<>();
-        cookies.put("sessionId", "abc123");
-        cookies.put("userId", "user456");
+        HashMap<String, List<String>> cookies = new HashMap<>();
+        cookies.put("sessionId", List.of("abc123"));
+        cookies.put("userId", List.of("user456"));
         contextObject = new JavalinContextObject("GET", "http://example.com", "192.168.1.1", new HashMap<>(), cookies, new HashMap<>());
 
         assertEquals("abc123", contextObject.getCookies().get("sessionId").get(0));