private void createUserCookie(String cookieValue) throws Exception { Cookie corruptedCookie = new Cookie(UserCookie.COOKIE_NAME, cookieValue); new UserCookie(corruptedCookie); }
@BeforeClass public static void initialize() throws Exception { userId = LocalStringsUtils.generateRandomId(true); acquisitionTime = new Date(); String formattedCookie = UserCookie.formatCookie(new String[] { userId, DateTimeUtils.formatAsRoundTripDate(acquisitionTime), }); defaultCookie = new Cookie(UserCookie.COOKIE_NAME, formattedCookie); UserCookie userCookie = new UserCookie(defaultCookie); requestTelemetryContextMock = mock(RequestTelemetryContext.class); when(requestTelemetryContextMock.getUserCookie()).thenReturn(userCookie); }
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; } }
@Test public void testCookieParsedSuccessfully() throws Exception { UserCookie userCookie = new UserCookie(defaultCookie); Date expectedAcquisitionTime = DateTimeUtils.parseRoundTripDateString(DateTimeUtils.formatAsRoundTripDate(acquisitionTime)); Assert.assertEquals("Wrong user ID", userId, userCookie.getUserId()); Assert.assertEquals("Wrong acquisition time", expectedAcquisitionTime, userCookie.getAcquisitionDate()); }