update native test

DataDog · Oct 30, 2024 · 3cdff6c · 3cdff6c
1 parent 26037b4
commit 3cdff6c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/ddtrace/internal/datadog/profiling/stack_v2/test/CMakeLists.txt b/ddtrace/internal/datadog/profiling/stack_v2/test/CMakeLists.txt
@@ -14,7 +14,7 @@ include(AnalysisFunc)
 
 function(dd_wrapper_add_test name)
     add_executable(${name} ${ARGN})
-    target_include_directories(${name} PRIVATE ../include)
+    target_include_directories(${name} PRIVATE ../include ../..)
     target_link_libraries(${name} PRIVATE gmock gtest_main _stack_v2)
     add_ddup_config(${name})
 

diff --git a/ddtrace/internal/datadog/profiling/stack_v2/test/thread_span_links.cpp b/ddtrace/internal/datadog/profiling/stack_v2/test/thread_span_links.cpp
@@ -1,3 +1,7 @@
+#include "thread_span_links.hpp"
+
+#include "constants.hpp"
+
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>
 
@@ -7,8 +11,6 @@
 #include <thread>
 #include <unordered_set>
 
-#include "thread_span_links.hpp"
-
 static void
 get()
 {
@@ -72,7 +74,9 @@ TEST(ThreadSpanLinks, ClearUnSeen)
         }
     }
 
-    Datadog::ThreadSpanLinks::get_instance().clear_unseen(seen_thread_ids);
+    for (unsigned int i = 0; i < g_default_unseen_count; ++i) {
+        Datadog::ThreadSpanLinks::get_instance().clear_unseen(seen_thread_ids);
+    }
 
     // Check that the unseen ids are removed
     for (auto thread_id : thread_ids) {