diff --git a/agent_api/src/test/java/context/JavalinContextObjectTest.java b/agent_api/src/test/java/context/JavalinContextObjectTest.java index f1d66d37..d57df22a 100644 --- 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> queryParams = new HashMap<>(); queryParams.put("param1", List.of("value1")); - Map cookies = new HashMap<>(); - cookies.put("sessionId", "abc123"); + HashMap> cookies = new HashMap<>(); + cookies.put("sessionId", List.of("abc123", "456")); Map 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 cookies = new HashMap<>(); - cookies.put("sessionId", "abc123"); - cookies.put("userId", "user456"); + HashMap> 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));