|
| 1 | +// Copyright 2024-Present Datadog, Inc. https://www.datadoghq.com/ |
| 2 | +// SPDX-License-Identifier: Apache-2.0 |
| 3 | +#[cfg(test)] |
| 4 | +mod otlp_export_tests { |
| 5 | + use libdd_data_pipeline::trace_exporter::TraceExporter; |
| 6 | + use libdd_trace_utils::test_utils::create_test_json_span; |
| 7 | + use serde_json::json; |
| 8 | + use tokio::task; |
| 9 | + |
| 10 | + fn get_v04_trace_snapshot_test_payload(name_prefix: &str) -> Vec<u8> { |
| 11 | + let mut span_1 = create_test_json_span(1234, 12342, 12341, 1, false); |
| 12 | + span_1["name"] = json!(format!("{name_prefix}_01")); |
| 13 | + span_1["metrics"] = json!({ |
| 14 | + "_dd_metric1": 1.0, |
| 15 | + "_dd_metric2": 2.0 |
| 16 | + }); |
| 17 | + let mut span_2 = create_test_json_span(1234, 12343, 12341, 1, false); |
| 18 | + span_2["name"] = json!(format!("{name_prefix}_02")); |
| 19 | + rmp_serde::to_vec_named(&vec![vec![span_1, span_2]]).unwrap() |
| 20 | + } |
| 21 | + |
| 22 | + #[cfg_attr(miri, ignore)] |
| 23 | + #[tokio::test] |
| 24 | + async fn otlp_export_sends_correct_payload() { |
| 25 | + use httpmock::MockServer; |
| 26 | + |
| 27 | + let server = MockServer::start_async().await; |
| 28 | + |
| 29 | + // Assert the OTLP request structure using json_body_includes matchers. |
| 30 | + // resourceSpans must be present with the correct service.name and environment resource |
| 31 | + // attributes, and spans must contain the expected name prefix. |
| 32 | + let mut mock = server |
| 33 | + .mock_async(|when, then| { |
| 34 | + when.method("POST") |
| 35 | + .path("/v1/traces") |
| 36 | + .header("content-type", "application/json") |
| 37 | + .json_body_includes( |
| 38 | + serde_json::json!({ |
| 39 | + "resourceSpans": [{ |
| 40 | + "resource": { |
| 41 | + "attributes": [ |
| 42 | + {"key": "service.name", "value": {"stringValue": "test"}}, |
| 43 | + ] |
| 44 | + } |
| 45 | + }] |
| 46 | + }) |
| 47 | + .to_string(), |
| 48 | + ); |
| 49 | + then.status(200).body("{}"); |
| 50 | + }) |
| 51 | + .await; |
| 52 | + |
| 53 | + let otlp_endpoint = format!("http://localhost:{}/v1/traces", server.port()); |
| 54 | + |
| 55 | + let task_result = task::spawn_blocking(move || { |
| 56 | + let mut builder = TraceExporter::builder(); |
| 57 | + builder |
| 58 | + .set_otlp_endpoint(&otlp_endpoint) |
| 59 | + .set_language("test-lang") |
| 60 | + .set_language_version("2.0") |
| 61 | + .set_language_interpreter_vendor("vendor") |
| 62 | + .set_language_interpreter("interpreter") |
| 63 | + .set_tracer_version("1.0") |
| 64 | + .set_env("test_env") |
| 65 | + .set_service("test"); |
| 66 | + |
| 67 | + let trace_exporter = builder.build().expect("Unable to build TraceExporter"); |
| 68 | + let data = get_v04_trace_snapshot_test_payload("test_otlp_export"); |
| 69 | + let response = trace_exporter.send(data.as_ref()); |
| 70 | + assert!(response.is_ok(), "OTLP send failed: {:?}", response.err()); |
| 71 | + }) |
| 72 | + .await; |
| 73 | + |
| 74 | + assert!(task_result.is_ok()); |
| 75 | + assert_eq!(mock.calls_async().await, 1); |
| 76 | + mock.delete(); |
| 77 | + } |
| 78 | + |
| 79 | + #[cfg_attr(miri, ignore)] |
| 80 | + #[tokio::test] |
| 81 | + async fn otlp_export_unsampled_traces_not_exported() { |
| 82 | + use httpmock::MockServer; |
| 83 | + |
| 84 | + let server = MockServer::start_async().await; |
| 85 | + let mut mock = server |
| 86 | + .mock_async(|when, then| { |
| 87 | + when.method("POST").path("/v1/traces"); |
| 88 | + then.status(200).body("{}"); |
| 89 | + }) |
| 90 | + .await; |
| 91 | + |
| 92 | + let otlp_endpoint = format!("http://localhost:{}/v1/traces", server.port()); |
| 93 | + |
| 94 | + // Build a v04 payload where all spans have sampling priority -1 (drop). |
| 95 | + let data = { |
| 96 | + let mut span = create_test_json_span(1234, 12341, 0, 1, true); |
| 97 | + span["metrics"]["_sampling_priority_v1"] = serde_json::json!(-1.0); |
| 98 | + rmp_serde::to_vec_named(&vec![vec![span]]).unwrap() |
| 99 | + }; |
| 100 | + |
| 101 | + let task_result = task::spawn_blocking(move || { |
| 102 | + let mut builder = TraceExporter::builder(); |
| 103 | + builder |
| 104 | + .set_otlp_endpoint(&otlp_endpoint) |
| 105 | + .set_language("test-lang") |
| 106 | + .set_language_version("2.0") |
| 107 | + .set_tracer_version("1.0") |
| 108 | + .set_env("test_env") |
| 109 | + .set_service("test"); |
| 110 | + |
| 111 | + let trace_exporter = builder.build().expect("Unable to build TraceExporter"); |
| 112 | + let response = trace_exporter.send(data.as_ref()); |
| 113 | + assert!(response.is_ok(), "send failed: {:?}", response.err()); |
| 114 | + }) |
| 115 | + .await; |
| 116 | + |
| 117 | + assert!(task_result.is_ok()); |
| 118 | + // The mock must not have been called: unsampled traces should be dropped before export. |
| 119 | + assert_eq!( |
| 120 | + mock.calls_async().await, |
| 121 | + 0, |
| 122 | + "Unsampled trace was exported — sampling is not being respected" |
| 123 | + ); |
| 124 | + mock.delete(); |
| 125 | + } |
| 126 | +} |
0 commit comments