TraceContextCorrelation.resolveRequestSource((HttpServletRequest) req, telemetry, instrumentationKey); } else { TelemetryCorrelationUtils.resolveRequestSource((HttpServletRequest) req, telemetry, instrumentationKey);
TraceContextCorrelation.resolveRequestSource((HttpServletRequest) req, telemetry, instrumentationKey); } else { TelemetryCorrelationUtils.resolveRequestSource((HttpServletRequest) req, telemetry, instrumentationKey);
@Test public void testTracestateIsNotResolvedWithNullRequest() { Map<String, String> headers = new HashMap<>(); String incomingTracestate = getTracestateHeaderValue("id1"); headers.put(TraceContextCorrelation.TRACESTATE_HEADER_NAME, incomingTracestate); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id2"); TraceContextCorrelation.resolveRequestSource(null, requestTelemetry, "ikey1"); Assert.assertNull(requestTelemetry.getSource()); }
@Test public void testTracestateIsNotResolvedIfHeaderIsEmpty() { Map<String, String> headers = new HashMap<>(); headers.put(TraceContextCorrelation.TRACESTATE_HEADER_NAME, ""); HttpServletRequest request = ServletUtils.createServletRequestWithHeaders(headers); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id1"); TraceContextCorrelation.resolveRequestSource(request, requestTelemetry, "ikey1"); Assert.assertNull(requestTelemetry.getSource()); }
@Test public void testTracestateIsNotResolvedWithNullIkey() { Map<String, String> headers = new HashMap<>(); String incomingTracestate = getTracestateHeaderValue("id1"); headers.put(TraceContextCorrelation.TRACESTATE_HEADER_NAME, incomingTracestate); HttpServletRequest request = ServletUtils.createServletRequestWithHeaders(headers); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id2"); TraceContextCorrelation.resolveRequestSource(request, requestTelemetry, null); Assert.assertNull(requestTelemetry.getSource()); }
@Test(expected = AssertionError.class) public void testTraceStateIsNotResolvedIfHeaderDoesntHaveAzureComponent() { Map<String, String> headers = new HashMap<>(); // get tracestate with non azure component String incomingTracestate = getTracestateHeaderValue(null); headers.put(TraceContextCorrelation.TRACESTATE_HEADER_NAME, incomingTracestate); HttpServletRequest request = ServletUtils.createServletRequestWithHeaders(headers); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id1"); TraceContextCorrelation.resolveRequestSource(request, requestTelemetry, "ikey1"); Assert.assertNull(requestTelemetry.getSource()); }
@Test public void testTracestateIsResolved() { Map<String, String> headers = new HashMap<>(); String incomingTracestate = getTracestateHeaderValue("id1"); headers.put(TraceContextCorrelation.TRACESTATE_HEADER_NAME, incomingTracestate); HttpServletRequest request = ServletUtils.createServletRequestWithHeaders(headers); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id2"); TraceContextCorrelation.resolveRequestSource(request, requestTelemetry, "ikey1"); Assert.assertEquals("cid-v1:id1", requestTelemetry.getSource()); }
@Test public void testTracestateIsNotResolvedWithNullRequestTelemetry() { Map<String, String> headers = new HashMap<>(); String incomingTracestate = getTracestateHeaderValue("id1"); headers.put(TraceContextCorrelation.TRACESTATE_HEADER_NAME, incomingTracestate); HttpServletRequest request = ServletUtils.createServletRequestWithHeaders(headers); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id2"); TraceContextCorrelation.resolveRequestSource(request, null, "ikey1"); Assert.assertNull(requestTelemetry.getSource()); }
@Test public void testTracestateIsNotResolvedWhenHeaderNotPresent() { Map<String, String> headers = new HashMap<>(); HttpServletRequest request = ServletUtils.createServletRequestWithHeaders(headers); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id1"); TraceContextCorrelation.resolveRequestSource(request, requestTelemetry, "ikey1"); Assert.assertNull(requestTelemetry.getSource()); }
@Test public void testSourceNotSetWhenIncomingAppIdInTraceStateIsSameAsCurrent() { Map<String, String> headers = new HashMap<>(); String incomingTracestate = getTracestateHeaderValue("id1"); headers.put(TraceContextCorrelation.TRACESTATE_HEADER_NAME, incomingTracestate); HttpServletRequest request = ServletUtils.createServletRequestWithHeaders(headers); RequestTelemetry requestTelemetry = new RequestTelemetry(); mockProfileFetcher.setResultStatus(ProfileFetcherResultTaskStatus.COMPLETE); mockProfileFetcher.setAppIdToReturn("id1"); TraceContextCorrelation.resolveRequestSource(request, requestTelemetry, "ikey1"); //source and target have same appId Assert.assertNull(requestTelemetry.getSource()); }