feat: add tests for trace-to-profile correlation

* test/test_tracer.cpp
* test/test_tracer_config.cpp

Author: elsa

test/test_tracer.cpp

@@ -1566,6 +1566,48 @@ TEST_CASE("128-bit trace IDs") {
   REQUIRE(*high == trace_id.high);
 }
 
+#ifdef __linux__
+TEST_CASE("correlate full host profiles") {
+  TracerConfig config;
+  config.service = "testsvc";
+  config.collector = std::make_shared<NullCollector>();
+  config.logger = std::make_shared<NullLogger>();
+
+  SECTION("is off by default") {
+    auto finalized_config = finalize_config(config);
+    REQUIRE(finalized_config);
+    Tracer tracer{*finalized_config};
+    REQUIRE(elastic_apm_profiling_correlation_process_storage_v1 == nullptr);
+  }
+
+  SECTION("is available when enabled") {
+    config.correlate_full_host_profiles = true;
+    auto finalized_config = finalize_config(config);
+    REQUIRE(finalized_config);
+    Tracer tracer{*finalized_config};
+    REQUIRE(elastic_apm_profiling_correlation_process_storage_v1 != nullptr);
+    {
+      auto span = tracer.create_span();
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1 != nullptr);
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1->trace_present == 1);
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1->valid == 1);
+      REQUIRE(
+          elastic_apm_profiling_correlation_tls_v1->trace_flags ==
+          (span.trace_segment().sampling_decision().has_value() &&
+                   (span.trace_segment().sampling_decision().value().priority >
+                    0)
+               ? 1
+               : 0));
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1->span_id != 0);
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1->transaction_id != 0);
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1->trace_id_high != 0);
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1->trace_id_low != 0);
+    }
+    REQUIRE(elastic_apm_profiling_correlation_tls_v1->trace_present == 0);
+  }
+}
+#endif
+
 TEST_CASE(
     "_dd.p.tid invalid or inconsistent with trace ID results in error tag") {
   struct TestCase {

test/test_tracer_config.cpp

@@ -1316,6 +1316,85 @@ TRACER_CONFIG_TEST("configure 128-bit trace IDs") {
   }
 }
 
+TRACER_CONFIG_TEST("TracerConfig::correlate_full_host_profiles") {
+  TracerConfig config;
+  config.service = "testsvc";
+  config.logger = std::make_shared<NullLogger>();
+
+  SECTION("default is false") {
+    {
+      auto finalized = finalize_config(config);
+      REQUIRE(finalized);
+      Tracer tracer{*finalized};
+      auto span = tracer.create_span();
+    }
+    REQUIRE(elastic_apm_profiling_correlation_process_storage_v1 == nullptr);
+    REQUIRE(elastic_apm_profiling_correlation_tls_v1 == nullptr);
+  }
+
+  SECTION("true enables correlation") {
+    {
+      config.correlate_full_host_profiles = true;
+      auto finalized = finalize_config(config);
+      REQUIRE(finalized);
+      Tracer tracer{*finalized};
+      auto span = tracer.create_span();
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1 != nullptr);
+      REQUIRE(elastic_apm_profiling_correlation_tls_v1->trace_present == 1);
+    }
+    REQUIRE(elastic_apm_profiling_correlation_process_storage_v1 != nullptr);
+    REQUIRE(elastic_apm_profiling_correlation_tls_v1 != nullptr);
+    REQUIRE(elastic_apm_profiling_correlation_tls_v1->trace_present == 0);
+    // reset for next tests
+    elastic_apm_profiling_correlation_process_storage_v1 = nullptr;
+    elastic_apm_profiling_correlation_tls_v1 = nullptr;
+  }
+
+  SECTION("overridden by DD_TRACE_CORRELATE_FULL_HOST_PROFILES") {
+    struct TestCase {
+      std::string name;
+      std::string dd_trace_correlate_full_host_profiles;
+      bool original_value;
+      bool correlate;
+    };
+
+    auto test_case = GENERATE(values<TestCase>({
+        {"falsy override ('false')", "false", true, false},
+        {"falsy override ('0')", "0", true, false},
+        {"falsy consistent ('false')", "false", false, false},
+        {"falsy consistent ('0')", "0", false, false},
+        {"truthy override ('true')", "true", false, true},
+        {"truthy override ('1')", "1", false, true},
+        {"truthy consistent ('true')", "true", true, true},
+        {"truthy consistent ('1')", "1", true, true},
+    }));
+
+    CAPTURE(test_case.name);
+    const EnvGuard guard{"DD_TRACE_CORRELATE_FULL_HOST_PROFILES",
+                         test_case.dd_trace_correlate_full_host_profiles};
+    config.report_traces = test_case.original_value;
+    {
+      auto finalized = finalize_config(config);
+      REQUIRE(finalized);
+      Tracer tracer{*finalized};
+      auto span = tracer.create_span();
+      if (test_case.correlate) {
+        REQUIRE(elastic_apm_profiling_correlation_process_storage_v1 !=
+                nullptr);
+        REQUIRE(elastic_apm_profiling_correlation_tls_v1 != nullptr);
+        REQUIRE(elastic_apm_profiling_correlation_tls_v1->trace_present == 1);
+        // reset for next tests
+        elastic_apm_profiling_correlation_process_storage_v1 = nullptr;
+        elastic_apm_profiling_correlation_tls_v1 = nullptr;
+      } else {
+        REQUIRE(elastic_apm_profiling_correlation_process_storage_v1 ==
+                nullptr);
+        REQUIRE(elastic_apm_profiling_correlation_tls_v1 == nullptr);
+      }
+    }
+  }
+}
+
 TRACER_CONFIG_TEST("baggage") {
   TracerConfig config;