private SessionCookie createSessionCookie(String cookieValue) { Cookie cookie = new Cookie(SessionCookie.COOKIE_NAME, cookieValue); return new SessionCookie(cookie); }
private static CookiesContainer getCookiesContainer(List<String> responseCookies) throws Exception { CookiesContainer cookiesContainer = new CookiesContainer(); if (responseCookies == null) { return cookiesContainer; } for (String formattedCookieWithExpiration : responseCookies) { if (formattedCookieWithExpiration.startsWith("ai_user")) { String formattedCookie = formattedCookieWithExpiration.split("=")[1].split(";")[0]; Cookie cookie = new Cookie(UserCookie.COOKIE_NAME, formattedCookie); UserCookie userCookie = new UserCookie(cookie); cookiesContainer.setUserCookie(userCookie); } else if(formattedCookieWithExpiration.startsWith("ai_session")) { String formattedCookie = formattedCookieWithExpiration.split("=")[1].split(";")[0]; Cookie cookie = new Cookie(SessionCookie.COOKIE_NAME, formattedCookie); SessionCookie sessionCookie = new SessionCookie(cookie); cookiesContainer.setSessionCookie(sessionCookie); } } return cookiesContainer; } }
@BeforeClass public static void initialize() throws Exception { sessionId = LocalStringsUtils.generateRandomId(true); String formattedCookie = SessionCookie.formatCookie(new String[] { sessionId }); defaultCookie = new Cookie(SessionCookie.COOKIE_NAME, formattedCookie); sessionContext = new SessionContext(new ConcurrentHashMap<String, String>()); sessionContext.setId(sessionId); SessionCookie sessionCookie = new SessionCookie(defaultCookie); requestTelemetryContextMock = mock(RequestTelemetryContext.class); when(requestTelemetryContextMock.getSessionCookie()).thenReturn(sessionCookie); }
@Test public void testCookieParsedSuccessfully() throws Exception { SessionCookie sessionCookie = new SessionCookie(defaultCookie); Assert.assertEquals("Wrong session ID", sessionId, sessionCookie.getSessionId()); }