dd-trace-py/ddtrace/internal/datadog/profiling/stack/test/test_sampling_cycle_state.cpp at 93702898f8e8da2d7d33e20fcd3666f3ebea6e0a · DataDog/dd-trace-py · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#include "echion/echion_sampler.h"
#include "echion/task_name.h"
#include "sampler.hpp"

#include <gtest/gtest.h>

#include <memory>

TEST(SamplingCycleState, UnwindReplacesTaskAndGreenletStacksFromPriorCycle)
{
    EchionSampler echion;
#if defined PL_LINUX
    ThreadInfo thread(1, 1, "test-thread", CLOCK_THREAD_CPUTIME_ID);
#elif defined PL_DARWIN
    ThreadInfo thread(1, 1, "test-thread", mach_thread_self());
#endif
    PyThreadState empty_tstate{};

    thread.current_tasks.push_back(std::make_unique<StackInfo>(TaskName::from_literal("stale-task"), false, 1));
    thread.current_greenlets.push_back(std::make_unique<StackInfo>(TaskName::from_literal("stale-greenlet"), false, 2));

    thread.unwind(echion, &empty_tstate);

    EXPECT_TRUE(thread.current_tasks.empty());
    EXPECT_TRUE(thread.current_greenlets.empty());
}

TEST(SamplingCycleState, GreenletSwitchPreservesLinkedParentFrame)
{
    constexpr GreenletInfo::ID child_id = 101;
    constexpr GreenletInfo::ID parent_id = 102;
    PyObject child_running_frame{};
    PyObject child_suspended_frame{};
    PyObject parent_suspended_frame{};
    PyObject parent_resumed_frame{};

    Datadog::Sampler& sampler = Datadog::Sampler::get();
    EchionSampler& echion = sampler.get_echion();
    {
        std::lock_guard<std::mutex> guard(echion.greenlet_info_map_lock());
        auto& greenlets = echion.greenlet_info_map();
        greenlets.emplace(
          child_id, std::make_unique<GreenletInfo>(child_id, &child_running_frame, TaskName::from_literal("child")));
        greenlets.emplace(
          parent_id,
          std::make_unique<GreenletInfo>(parent_id, &parent_suspended_frame, TaskName::from_literal("parent")));
    }
    sampler.link_greenlets(parent_id, child_id);

    sampler.record_greenlet_switch(child_id, &child_suspended_frame, parent_id, &parent_resumed_frame, false);
    {
        std::lock_guard<std::mutex> guard(echion.greenlet_info_map_lock());
        EXPECT_EQ(echion.greenlet_parent_map().at(child_id), parent_id);
        EXPECT_EQ(echion.greenlet_info_map().at(child_id)->frame, &child_suspended_frame);
        EXPECT_EQ(echion.greenlet_info_map().at(parent_id)->frame, &parent_suspended_frame);
    }

    sampler.record_greenlet_switch(child_id, &child_running_frame, parent_id, &parent_resumed_frame, true);
    {
        std::lock_guard<std::mutex> guard(echion.greenlet_info_map_lock());
        EXPECT_EQ(echion.greenlet_info_map().at(child_id)->frame, &child_running_frame);
        EXPECT_EQ(echion.greenlet_info_map().at(parent_id)->frame, &parent_resumed_frame);
        echion.greenlet_info_map().erase(child_id);
        echion.greenlet_info_map().erase(parent_id);
        echion.greenlet_parent_map().erase(child_id);
    }
}