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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
#include <iostream>
#include "log.h"
#include "gtest/gtest.h"
namespace bpftrace::test::log {
TEST(LogStream, basic)
{
std::ostringstream ss;
const std::string content_1 = "hello world";
const std::string content_2 = "some messages 100###**";
LOG(WARNING, ss) << content_1 << content_2;
EXPECT_EQ(ss.str(), "WARNING: " + content_1 + content_2 + "\n");
ss.str({});
LOG(ERROR, ss) << content_1 << content_2 << content_1 << content_2 << "\n"
<< content_1 << content_2 << content_1 << content_2 << "\n";
const std::string content_long = content_1 + content_2 + content_1 +
content_2;
EXPECT_EQ(ss.str(), "ERROR: " + content_long + "\n" + content_long + "\n");
ss.str({});
// test macro with 1 argument
ENABLE_LOG(V1);
auto *cerr_buf = std::cerr.rdbuf(ss.rdbuf());
LOG(V1) << content_1 << content_2;
EXPECT_EQ(ss.str(), content_1 + content_2 + "\n");
std::cerr.rdbuf(cerr_buf);
DISABLE_LOG(V1);
}
TEST(LogStream, basic_colorized)
{
std::ostringstream ss;
const std::string content_1 = "hello world";
const std::string content_2 = "some messages 100###**";
const std::string warning_color = std::string(bpftrace::LogColor::YELLOW);
const std::string error_color = std::string(bpftrace::LogColor::RED);
const std::string default_color = std::string(bpftrace::LogColor::RESET);
Log::get().set_colorize(true);
LOG(WARNING, ss) << content_1 << content_2;
EXPECT_EQ(ss.str(),
warning_color + "WARNING: " + content_1 + content_2 +
default_color + "\n");
ss.str({});
LOG(ERROR, ss) << content_1 << content_2 << content_1 << content_2 << "\n"
<< content_1 << content_2 << content_1 << content_2 << "\n";
const std::string content_long = content_1 + content_2 + content_1 +
content_2;
EXPECT_EQ(ss.str(),
error_color + "ERROR: " + content_long + "\n" + content_long +
default_color + "\n");
Log::get().set_colorize(false);
ss.str({});
// test macro with 1 argument
ENABLE_LOG(V1);
auto *cerr_buf = std::cerr.rdbuf(ss.rdbuf());
LOG(V1) << content_1 << content_2;
EXPECT_EQ(ss.str(), content_1 + content_2 + "\n");
std::cerr.rdbuf(cerr_buf);
DISABLE_LOG(V1);
}
TEST(LogStream, with_location)
{
std::string location = "stdin:xyz";
std::vector<std::string> context({ "ctx1", "ctx2" });
std::ostringstream ss;
const std::string expected = "stdin:xyz: ERROR: test error\n"
"ctx1\n"
"ctx2\n";
LOG(ERROR, std::move(location), std::move(context), ss) << "test error";
EXPECT_EQ(ss.str(), expected);
}
TEST(LogStream, with_location_colorized)
{
std::string location = "stdin:xyz";
std::vector<std::string> context({ "ctx1", "ctx2" });
std::ostringstream ss;
const std::string error_color = std::string(bpftrace::LogColor::RED);
const std::string default_color = std::string(bpftrace::LogColor::RESET);
const std::string expected = error_color + "stdin:xyz: ERROR: test error" +
default_color + "\n" + "ctx1\n" + "ctx2\n";
Log::get().set_colorize(true);
LOG(ERROR, std::move(location), std::move(context), ss) << "test error";
Log::get().set_colorize(false);
EXPECT_EQ(ss.str(), expected);
}
TEST(Log, disable_log_type)
{
std::ostringstream ss;
const std::string content = "This is the warning message";
LOG(WARNING, ss) << content;
EXPECT_EQ(ss.str(), "WARNING: " + content + "\n");
ss.str({});
Log::get().disable(LogType::WARNING);
LOG(WARNING, ss) << content;
EXPECT_EQ(ss.str(), "");
// make sure other log types are not affected
LOG(ERROR, ss) << content;
EXPECT_EQ(ss.str(), "ERROR: " + content + "\n");
ss.str({});
Log::get().enable(LogType::WARNING);
LOG(WARNING, ss) << content;
EXPECT_EQ(ss.str(), "WARNING: " + content + "\n");
ss.str({});
}
TEST(LogStreamBug, log_stream_bug_should_be_aborted)
{
const std::string content = "I'm gonna die";
auto output = [&content](int line) {
const std::string filename = __FILE__;
return "BUG: \\[" + filename + ":" + std::to_string(line) + "\\] " +
content;
};
EXPECT_DEATH(LOG(BUG) << content, output(__LINE__));
}
TEST(LogStreamBug, log_stream_bug_colorized)
{
Log::get().set_colorize(true);
const std::string content = "I'm gonna die";
auto output = [&content](int line) {
// Escape [ to avoid being captured by regex
const std::string bug_color = "\033\\[31m";
const std::string default_color = "\033\\[0m";
const std::string filename = __FILE__;
return bug_color + "BUG: \\[" + filename + ":" + std::to_string(line) +
"\\] " + content + default_color;
};
EXPECT_DEATH(LOG(BUG) << content, output(__LINE__));
Log::get().set_colorize(false);
}
} // namespace bpftrace::test::log
|
