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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#include <sstream>
#include "bpfmap.h"
#include "mocks.h"
#include "output/text.h"
#include "types_format.h"
#include "gtest/gtest.h"
namespace bpftrace::test::output {
static ast::CDefinitions no_c_defs; // Used for format below.
TEST(TextOutput, lhist_no_suffix)
{
std::stringstream out;
::bpftrace::output::TextOutput output(out, out);
auto bpftrace = get_mock_bpftrace();
bpftrace->resources.maps_info["@mymap"] = MapInfo{
.key_type = CreateInt64(),
.value_type = SizedType{ Type::lhist_t, 8 },
.detail = LinearHistogramArgs{ .min = 610000,
.max = 670000,
.step = 10000 },
.id = {},
.is_scalar = true,
};
HistogramMap values_by_key = {
{ OpaqueValue::from<uint64_t>(0), { 0, 1, 1, 1, 1, 1, 1, 0 } },
};
auto mock_map = std::make_unique<MockBpfMap>(libbpf::BPF_MAP_TYPE_HASH,
"@mymap");
EXPECT_CALL(*mock_map, collect_histogram_data(testing::_, testing::_))
.WillOnce(testing::Return(
testing::ByMove(Result<HistogramMap>(values_by_key))));
auto hist = format(*bpftrace, no_c_defs, *mock_map);
ASSERT_TRUE(bool(hist));
output.map(mock_map->name(), *hist);
// The buckets for this test case have been specifically chosen: 640000 can
// also be written as 625K, while the other bucket boundaries can not be
// expressed with a suffix. We should only use the suffix representation for a
// bucket if all buckets can be expressed with one.
EXPECT_EQ(R"(@mymap:
[610000, 620000) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[620000, 630000) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[630000, 640000) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[640000, 650000) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[650000, 660000) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[660000, 670000) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
)",
out.str());
}
TEST(TextOutput, lhist_suffix)
{
std::stringstream out;
::bpftrace::output::TextOutput output(out, out);
auto bpftrace = get_mock_bpftrace();
bpftrace->resources.maps_info["@mymap"] = MapInfo{
.key_type = CreateInt64(),
.value_type = SizedType{ Type::lhist_t, 8 },
.detail = LinearHistogramArgs{ .min = 0, .max = 5 * 1024, .step = 1024 },
.id = {},
.is_scalar = true,
};
const HistogramMap values_by_key = {
{ OpaqueValue::from<uint64_t>(0), { 0, 1, 1, 1, 1, 1, 0 } },
};
auto mock_map = std::make_unique<MockBpfMap>(libbpf::BPF_MAP_TYPE_HASH,
"@mymap");
EXPECT_CALL(*mock_map, collect_histogram_data(testing::_, testing::_))
.WillOnce(testing::Return(testing::ByMove(HistogramMap(values_by_key))));
auto hist = format(*bpftrace, no_c_defs, *mock_map);
ASSERT_TRUE(bool(hist));
output.map(mock_map->name(), *hist);
EXPECT_EQ(R"(@mymap:
[0, 1K) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[1K, 2K) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[2K, 3K) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[3K, 4K) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
[4K, 5K) 1 |@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@|
)",
out.str());
}
} // namespace bpftrace::test::output
|
