/** * Initializes the properties of the given telemetry. * * @param telemetry The {@link com.microsoft.applicationinsights.telemetry.Telemetry} to initialize. */ @Override protected void onInitializeTelemetry(Telemetry telemetry) { SessionContext session = telemetry.getContext().getSession(); if (!CommonUtils.isNullOrEmpty(session.getId())) { return; } RequestTelemetry requestTelemetry = ThreadContext.getRequestTelemetryContext().getHttpRequestTelemetry(); SessionContext requestSessionContext = requestTelemetry.getContext().getSession(); if (requestSessionContext == null) { return; } session.setId(requestSessionContext.getId()); Boolean isFirst = requestSessionContext.getIsFirst(); session.setIsFirst(isFirst != null ? isFirst : false); } }
@Test public void testSetIsFirstTrue() { ConcurrentHashMap<String, String> map = new ConcurrentHashMap<String, String>(); SessionContext context = new SessionContext(map); context.setIsFirst(true); assertEquals(context.getIsFirst(), true); assertEquals(map.size(), 1); assertEquals(Boolean.valueOf(map.get(ContextTagKeys.getKeys().getSessionIsFirst())), true); }
@Test public void testSetIsNewSessionTrue() { ConcurrentHashMap<String, String> map = new ConcurrentHashMap<String, String>(); SessionContext context = new SessionContext(map); context.setIsNewSession(true); assertEquals(context.getIsNewSession(), true); assertEquals(map.size(), 1); assertEquals(Boolean.valueOf(map.get(ContextTagKeys.getKeys().getSessionIsNew())), true); } }
private void handlePossibleGSMSyntheticRequest(Telemetry telemetry, HttpServletRequest request) { String gsmSyntheticTestRunId = request.getHeader(SYNTHETIC_TEST_RUN_ID); if (CommonUtils.isNullOrEmpty(gsmSyntheticTestRunId)) { return; } String syntheticSource = telemetry.getContext().getOperation().getSyntheticSource(); if (CommonUtils.isNullOrEmpty(syntheticSource)) { telemetry.getContext().getOperation().setSyntheticSource(SYNTHETIC_SOURCE_NAME); } String sessionId = telemetry.getContext().getSession().getId(); if (CommonUtils.isNullOrEmpty(sessionId)) { telemetry.getContext().getSession().setId(gsmSyntheticTestRunId); } String userId = telemetry.getContext().getUser().getId(); if (CommonUtils.isNullOrEmpty(userId)) { String header = request.getHeader(SYNTHETIC_TEST_LOCATION); telemetry.getContext().getUser().setId(header); } }
@Test public void testSetId() { ConcurrentHashMap<String, String> map = new ConcurrentHashMap<String, String>(); SessionContext context = new SessionContext(map); context.setId("mock"); assertEquals(context.getId(), "mock"); assertEquals(map.size(), 1); assertEquals(map.get(ContextTagKeys.getKeys().getSessionId()), "mock"); }
@Test public void testWhenCookieExistCorrectSessionIdAttachedToSentTelemetry() throws Exception { HttpHelper.sendRequestAndGetResponseCookie(server.getPortNumber(), sessionCookieFormatted); RequestTelemetry requestTelemetry = channel.getTelemetryItems(RequestTelemetry.class).get(0); Assert.assertTrue(sessionCookieFormatted.contains(requestTelemetry.getContext().getSession().getId())); Assert.assertEquals(requestTelemetry.getContext().getSession().getId(), HttpHelper.getCookie()); }
@Before public void classInitialize() { RequestTelemetryContext context = new RequestTelemetryContext(DateTimeUtils.getDateTimeNow().getTime()); ThreadContext.setRequestTelemetryContext(context); // Set session ID for the http request. RequestTelemetry requestTelemetry = context.getHttpRequestTelemetry(); requestTelemetry.getContext().getSession().setId(REQUEST_SESSION_ID); requestTelemetry.getContext().getSession().setIsFirst(true); }
private void initializeSession(final SessionContext sessionContext) { sessionContext.setId(UUID.randomUUID().toString()); }
@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); }
@Override public void initialize(Telemetry telemetry) { RequestTelemetry requestTelemetry = ThreadContext.getRequestTelemetryContext().getHttpRequestTelemetry(); SessionContext requestSessionContext = requestTelemetry.getContext().getSession(); if (expectedSessionId == null) { //No longer testing for UUID //Assert.assertTrue(Sanitizer.isUUID(requestSessionContext.getId())); Assert.assertNotEquals(requestSessionContext.getId(), HttpHelper.getCookie()); } else { Assert.assertEquals(expectedSessionId, requestSessionContext.getId()); } Assert.assertEquals(expectedIsFirst, requestSessionContext.getIsFirst()); } }
/** * Gets the object describing a user session tracked by this instance. * @return The user's session. */ public SessionContext getSession() { if (session == null) { session = new SessionContext(tags); } return session; }
@Test public void testIsFirstSessionSetWhenRequired() { TraceTelemetry telemetry = new TraceTelemetry(); sessionTelemetryInitializer.onInitializeTelemetry(telemetry); Assert.assertEquals("First session expected.", true, telemetry.getContext().getSession().getIsFirst()); }
private void handlePossibleGSMSyntheticRequest(Telemetry telemetry, HttpServletRequest request) { String gsmSyntheticTestRunId = request.getHeader(SYNTHETIC_TEST_RUN_ID); if (CommonUtils.isNullOrEmpty(gsmSyntheticTestRunId)) { return; } String syntheticSource = telemetry.getContext().getOperation().getSyntheticSource(); if (CommonUtils.isNullOrEmpty(syntheticSource)) { telemetry.getContext().getOperation().setSyntheticSource(SYNTHETIC_SOURCE_NAME); } String sessionId = telemetry.getContext().getSession().getId(); if (CommonUtils.isNullOrEmpty(sessionId)) { telemetry.getContext().getSession().setId(gsmSyntheticTestRunId); } String userId = telemetry.getContext().getUser().getId(); if (CommonUtils.isNullOrEmpty(userId)) { String header = request.getHeader(SYNTHETIC_TEST_LOCATION); telemetry.getContext().getUser().setId(header); } }
@Test public void gsmSyntheticHeadersTest() { WebSyntheticRequestTelemetryInitializer tested = new WebSyntheticRequestTelemetryInitializer(); HttpServletRequest mockRequest = Mockito.mock(HttpServletRequest.class); Mockito.doReturn("").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_SOURCE); Mockito.doReturn("A1").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_RUN_ID); Mockito.doReturn("A2").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_LOCATION); RequestTelemetryContext context = new RequestTelemetryContext(DateTimeUtils.getDateTimeNow().getTime(), mockRequest); ThreadContext.setRequestTelemetryContext(context); TraceTelemetry telemetry = new TraceTelemetry(); tested.initialize(telemetry); assertEquals(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_SOURCE_NAME, telemetry.getContext().getOperation().getSyntheticSource()); assertEquals("A1", telemetry.getContext().getSession().getId()); assertEquals("A2", telemetry.getContext().getUser().getId()); }
/** * Begin request processing. * * @param req The request to process * @param res The response to modify */ @Override public void onBeginRequest(ServletRequest req, ServletResponse res) { HttpServletRequest request = (HttpServletRequest)req; RequestTelemetryContext context = ThreadContext.getRequestTelemetryContext(); SessionCookie sessionCookie = com.microsoft.applicationinsights.web.internal.cookies.Cookie.getCookie( SessionCookie.class, request, SessionCookie.COOKIE_NAME); if (sessionCookie == null) { return; } context.setSessionCookie(sessionCookie); String sessionId = sessionCookie.getSessionId(); getTelemetrySessionContext(context).setId(sessionId); }
/** * Initializes the properties of the given telemetry. * * @param telemetry The {@link com.microsoft.applicationinsights.telemetry.Telemetry} to initialize. */ @Override protected void onInitializeTelemetry(Telemetry telemetry) { SessionContext session = telemetry.getContext().getSession(); if (!CommonUtils.isNullOrEmpty(session.getId())) { return; } RequestTelemetry requestTelemetry = ThreadContext.getRequestTelemetryContext().getHttpRequestTelemetry(); SessionContext requestSessionContext = requestTelemetry.getContext().getSession(); if (requestSessionContext == null) { return; } session.setId(requestSessionContext.getId()); Boolean isFirst = requestSessionContext.getIsFirst(); session.setIsFirst(isFirst != null ? isFirst : false); } }
@Test public void testSessionIdNotOverriddenIfAlreadySet() { String expectedSessionId = "SOME_ID"; TraceTelemetry telemetry = new TraceTelemetry(); telemetry.getContext().getSession().setId(expectedSessionId); sessionTelemetryInitializer.onInitializeTelemetry(telemetry); Assert.assertEquals(expectedSessionId, telemetry.getContext().getSession().getId()); }
@Test public void testSetIsFirstNull() { ConcurrentHashMap<String, String> map = new ConcurrentHashMap<String, String>(); SessionContext context = new SessionContext(map); context.setIsFirst(null); assertNull(context.getIsFirst()); assertEquals(map.size(), 0); assertNull(map.get(ContextTagKeys.getKeys().getSessionIsFirst())); }
@Test public void commonSyntheticHeadersTest() { WebSyntheticRequestTelemetryInitializer tested = new WebSyntheticRequestTelemetryInitializer(); HttpServletRequest mockRequest = Mockito.mock(HttpServletRequest.class); Mockito.doReturn("A").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_SOURCE); Mockito.doReturn("A1").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_USER_ID); Mockito.doReturn("A2").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_SESSION_ID); Mockito.doReturn("A3").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_OPERATION_ID); Mockito.doReturn("A4").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_TEST_NAME); Mockito.doReturn("A5").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_RUN_ID); Mockito.doReturn("A6").when(mockRequest).getHeader(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_LOCATION); RequestTelemetryContext context = new RequestTelemetryContext(DateTimeUtils.getDateTimeNow().getTime(), mockRequest); ThreadContext.setRequestTelemetryContext(context); TraceTelemetry telemetry = new TraceTelemetry(); tested.initialize(telemetry); assertEquals("A", telemetry.getContext().getOperation().getSyntheticSource()); assertEquals("A1", telemetry.getContext().getUser().getId()); assertEquals("A2", telemetry.getContext().getSession().getId()); assertEquals("A3", telemetry.getContext().getOperation().getId()); assertEquals("A4", telemetry.getContext().getProperties().get(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_TEST_NAME)); assertEquals("A5", telemetry.getContext().getProperties().get(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_RUN_ID)); assertEquals("A6", telemetry.getContext().getProperties().get(WebSyntheticRequestTelemetryInitializer.SYNTHETIC_TEST_LOCATION)); } }
/** * Begin request processing. * * @param req The request to process * @param res The response to modify */ @Override public void onBeginRequest(ServletRequest req, ServletResponse res) { HttpServletRequest request = (HttpServletRequest)req; RequestTelemetryContext context = ThreadContext.getRequestTelemetryContext(); SessionCookie sessionCookie = com.microsoft.applicationinsights.web.internal.cookies.Cookie.getCookie( SessionCookie.class, request, SessionCookie.COOKIE_NAME); if (sessionCookie == null) { return; } context.setSessionCookie(sessionCookie); String sessionId = sessionCookie.getSessionId(); getTelemetrySessionContext(context).setId(sessionId); }