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
|
/*
* MRustC - Rust Compiler
* - By John Hodge (Mutabah/thePowersGang)
*
* expand/test.cpp
* - #[test] handling
*/
#include <synext_decorator.hpp>
#include <ast/ast.hpp>
#include <ast/crate.hpp>
class CTestHandler:
public ExpandDecorator
{
AttrStage stage() const override { return AttrStage::Post; }
void handle(const Span& sp, const AST::Attribute& mi, ::AST::Crate& crate, const AST::Path& path, AST::Module& mod, AST::Item&i) const override {
if( ! i.is_Function() ) {
ERROR(sp, E0000, "#[test] can only be put on functions - found on " << i.tag_str());
}
if( crate.m_test_harness )
{
::AST::TestDesc td;
for(const auto& node : path.nodes())
{
td.name += "::";
td.name += node.name();
}
td.path = ::AST::Path(path);
crate.m_tests.push_back( mv$(td) );
}
else
{
i = AST::Item::make_None({});
}
}
};
class CTestHandler_SP:
public ExpandDecorator
{
AttrStage stage() const override { return AttrStage::Pre; }
void handle(const Span& sp, const AST::Attribute& mi, ::AST::Crate& crate, const AST::Path& path, AST::Module& mod, AST::Item&i) const override {
if( ! i.is_Function() ) {
ERROR(sp, E0000, "#[should_panic] can only be put on functions - found on " << i.tag_str());
}
if( crate.m_test_harness )
{
for(auto& td : crate.m_tests)
{
if( td.path != path )
continue ;
if( mi.has_sub_items() )
{
td.panic_type = ::AST::TestDesc::ShouldPanic::YesWithMessage;
// TODO: Check that name is correct and that it is a string
td.expected_panic_message = mi.items().at(0).string();
}
else
{
td.panic_type = ::AST::TestDesc::ShouldPanic::Yes;
}
return ;
}
//ERROR()
}
}
};
class CTestHandler_Ignore:
public ExpandDecorator
{
AttrStage stage() const override { return AttrStage::Pre; }
void handle(const Span& sp, const AST::Attribute& mi, ::AST::Crate& crate, const AST::Path& path, AST::Module& mod, AST::Item&i) const override {
if( ! i.is_Function() ) {
ERROR(sp, E0000, "#[should_panic] can only be put on functions - found on " << i.tag_str());
}
if( crate.m_test_harness )
{
for(auto& td : crate.m_tests)
{
if( td.path != path )
continue ;
td.ignore = true;
return ;
}
//ERROR()
}
}
};
STATIC_DECORATOR("test", CTestHandler);
STATIC_DECORATOR("should_panic", CTestHandler_SP);
STATIC_DECORATOR("ignore", CTestHandler_Ignore);
|
