/** * 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); } }
/** * 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 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()); }
@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()); }
@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)); } }
@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()); } }
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); } }
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 testSessionIdIsUpdatedFromRequest() { TraceTelemetry telemetry = new TraceTelemetry(); sessionTelemetryInitializer.onInitializeTelemetry(telemetry); Assert.assertEquals(REQUEST_SESSION_ID, telemetry.getContext().getSession().getId()); }
@Test public void noSyntheticHeadersTest() { WebSyntheticRequestTelemetryInitializer tested = new WebSyntheticRequestTelemetryInitializer(); HttpServletRequest mockRequest = Mockito.mock(HttpServletRequest.class); Mockito.doReturn("").when(mockRequest).getHeader(anyString()); RequestTelemetryContext context = new RequestTelemetryContext(DateTimeUtils.getDateTimeNow().getTime(), mockRequest); ThreadContext.setRequestTelemetryContext(context); TraceTelemetry telemetry = new TraceTelemetry(); tested.initialize(telemetry); assertTrue(CommonUtils.isNullOrEmpty(telemetry.getContext().getSession().getId())); assertTrue(CommonUtils.isNullOrEmpty(telemetry.getContext().getUser().getId())); assertTrue(CommonUtils.isNullOrEmpty(telemetry.getContext().getOperation().getId())); }
private void handleCommonSyntheticRequest(String syntheticSourceHeader, Telemetry telemetry, HttpServletRequest request) { String syntheticSource = telemetry.getContext().getOperation().getSyntheticSource(); if (CommonUtils.isNullOrEmpty(syntheticSource)) { telemetry.getContext().getOperation().setSyntheticSource(syntheticSourceHeader); } String userId = telemetry.getContext().getUser().getId(); if (CommonUtils.isNullOrEmpty(userId)) { String header = request.getHeader(SYNTHETIC_TEST_USER_ID); telemetry.getContext().getUser().setId(header); } String sessionId = telemetry.getContext().getSession().getId(); if (CommonUtils.isNullOrEmpty(sessionId)) { String header = request.getHeader(SYNTHETIC_TEST_SESSION_ID); telemetry.getContext().getSession().setId(header); } String operationId = telemetry.getContext().getOperation().getId(); if (CommonUtils.isNullOrEmpty(operationId)) { String header = request.getHeader(SYNTHETIC_TEST_OPERATION_ID); telemetry.getContext().getOperation().setId(header); } putInProperties(telemetry, request, SYNTHETIC_TEST_TEST_NAME, SYNTHETIC_TEST_RUN_ID, SYNTHETIC_TEST_LOCATION); }
private void handleCommonSyntheticRequest(String syntheticSourceHeader, Telemetry telemetry, HttpServletRequest request) { String syntheticSource = telemetry.getContext().getOperation().getSyntheticSource(); if (CommonUtils.isNullOrEmpty(syntheticSource)) { telemetry.getContext().getOperation().setSyntheticSource(syntheticSourceHeader); } String userId = telemetry.getContext().getUser().getId(); if (CommonUtils.isNullOrEmpty(userId)) { String header = request.getHeader(SYNTHETIC_TEST_USER_ID); telemetry.getContext().getUser().setId(header); } String sessionId = telemetry.getContext().getSession().getId(); if (CommonUtils.isNullOrEmpty(sessionId)) { String header = request.getHeader(SYNTHETIC_TEST_SESSION_ID); telemetry.getContext().getSession().setId(header); } String operationId = telemetry.getContext().getOperation().getId(); if (CommonUtils.isNullOrEmpty(operationId)) { String header = request.getHeader(SYNTHETIC_TEST_OPERATION_ID); telemetry.getContext().getOperation().setId(header); } putInProperties(telemetry, request, SYNTHETIC_TEST_TEST_NAME, SYNTHETIC_TEST_RUN_ID, SYNTHETIC_TEST_LOCATION); }
@Test @Ignore public void testSessionStateTelemetryContainsSessionIdOnEndState() throws Exception { sessionCookieFormatted = HttpHelper.getFormattedSessionCookieHeader(true); HttpHelper.sendRequestAndGetResponseCookie(server.getPortNumber(), sessionCookieFormatted); SessionStateTelemetry telemetry = getSessionStateTelemetryWithState(SessionState.End); Assert.assertNotNull("Session ID shouldn't be null", telemetry.getContext().getSession().getId()); }
@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 @Ignore public void testSessionStateTelemetryEndStateContainsExpiredSessionId() throws Exception { sessionCookieFormatted = HttpHelper.getFormattedSessionCookieHeader(true); HttpHelper.sendRequestAndGetResponseCookie(server.getPortNumber(), sessionCookieFormatted); SessionStateTelemetry telemetry = getSessionStateTelemetryWithState(SessionState.End); String expectedSessionId = HttpHelper.getSessionIdFromCookie(sessionCookieFormatted); Assert.assertEquals( "Expected session ID of the expired session cookie", expectedSessionId, telemetry.getContext().getSession().getId()); }
private void testNoSessionCookie(Boolean isFirstValue) { RequestTelemetryContext requestTelemetryContext = ThreadContext.getRequestTelemetryContext(); requestTelemetryContext.getHttpRequestTelemetry().getContext().getSession().setIsFirst(isFirstValue); requestTelemetryContext.getHttpRequestTelemetry().getContext().getSession().setId(null); TraceTelemetry telemetry = new TraceTelemetry(); sessionTelemetryInitializer.onInitializeTelemetry(telemetry); Assert.assertNull(telemetry.getContext().getSession().getId()); Assert.assertFalse(telemetry.getContext().getSession().getIsFirst()); } }