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
|
#include "ast/passes/unstable_feature.h"
#include "ast/passes/config_analyser.h"
#include "ast/passes/parser.h"
#include "ast/passes/printer.h"
#include "mocks.h"
#include "gtest/gtest.h"
namespace bpftrace::test::unstable_feature {
using ::testing::HasSubstr;
void test(const std::string& input, const std::string& error = "")
{
auto mock_bpftrace = get_mock_bpftrace();
BPFtrace& bpftrace = *mock_bpftrace;
// The input provided here is embedded into an expression.
ast::ASTContext ast("stdin", input);
std::stringstream msg;
msg << "\nInput:\n" << input << "\n\nOutput:\n";
auto ok = ast::PassManager()
.put(ast)
.put(bpftrace)
.add(CreateParsePass())
.add(ast::CreateConfigPass())
.add(ast::CreateUnstableFeaturePass())
.run();
std::ostringstream out;
ast::Printer printer(out);
printer.visit(ast.root);
ast.diagnostics().emit(out);
if (error.empty()) {
ASSERT_TRUE(ok && ast.diagnostics().ok()) << msg.str() << out.str();
} else {
ASSERT_FALSE(ok && ast.diagnostics().ok()) << msg.str() << out.str();
EXPECT_THAT(out.str(), HasSubstr(error)) << msg.str() << out.str();
}
}
void test_error(const std::string& input, const std::string& error)
{
test(input, error);
}
TEST(unstable_feature, check_error)
{
test_error("config = { unstable_map_decl=0 } let @a = lruhash(5); begin { "
"@a[0] = 0; }",
"map declarations feature is not enabled by default. To enable "
"this unstable "
"feature, "
"set the config flag to enable. unstable_map_decl=enable");
test_error("config = { unstable_macro=0 } macro add_one($x) { $x } begin { "
"@a[0] = 0; add_one(1); }",
"macros feature is not enabled by default. To enable this "
"unstable feature, "
"set the config flag to enable. unstable_macro=enable");
test_error(
"config = { unstable_macro=error } macro add_one($x) { $x } begin { "
"@a[0] = 0; add_one(1); }",
"macros feature is not enabled by default. To enable this unstable "
"feature, "
"set the config flag to enable. unstable_macro=enable");
test("config = { unstable_macro=warn } macro add_one($x) { $x } begin { "
"@a[0] = 0; add_one(1); }");
// Macros have to be called to get the error/warning
test("config = { unstable_macro=error } macro add_one($x) { $x } begin { "
"@a[0] = 0; }");
test("config = { unstable_tseries=warn } begin { @ = tseries(4, 1s, 10); }");
test_error(
"config = { unstable_tseries=error } begin { @ = tseries(4, 1s, 10); }",
"tseries feature is not enabled by default. To enable this unstable "
"feature, set the config flag to enable. unstable_tseries=enable");
test("config = { unstable_addr=warn } begin { $x = 1; &$x; }");
test_error("config = { unstable_addr=error } begin { $x = 1; &$x; }",
"address-of operator (&) feature is not enabled by default. To "
"enable this unstable "
"feature, set the config flag to enable. unstable_addr=enable");
test("config = { unstable_addr=warn } begin { @x = 1; &@x; }");
test_error("config = { unstable_addr=error } begin { @x = 1; &@x; }",
"address-of operator (&) feature is not enabled by default. To "
"enable this unstable "
"feature, set the config flag to enable. unstable_addr=enable");
}
} // namespace bpftrace::test::unstable_feature
|
