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
|
use rustc_ast as ast;
use rustc_span::{DUMMY_SP, Ident, create_default_session_globals_then};
use thin_vec::ThinVec;
use super::*;
fn fun_to_string(
decl: &ast::FnDecl,
header: ast::FnHeader,
ident: Ident,
generics: &ast::Generics,
) -> String {
to_string(|s| {
let (cb, ib) = s.head("");
s.print_fn(decl, header, Some(ident), generics);
s.end(ib);
s.end(cb);
})
}
fn variant_to_string(var: &ast::Variant) -> String {
to_string(|s| s.print_variant(var))
}
#[test]
fn test_fun_to_string() {
create_default_session_globals_then(|| {
let abba_ident = Ident::from_str("abba");
let decl = ast::FnDecl { inputs: ThinVec::new(), output: ast::FnRetTy::Default(DUMMY_SP) };
let generics = ast::Generics::default();
assert_eq!(
fun_to_string(&decl, ast::FnHeader::default(), abba_ident, &generics),
"fn abba()"
);
})
}
#[test]
fn test_variant_to_string() {
create_default_session_globals_then(|| {
let ident = Ident::from_str("principal_skinner");
let var = ast::Variant {
ident,
vis: ast::Visibility {
span: DUMMY_SP,
kind: ast::VisibilityKind::Inherited,
tokens: None,
},
attrs: ast::AttrVec::new(),
id: ast::DUMMY_NODE_ID,
data: ast::VariantData::Unit(ast::DUMMY_NODE_ID),
disr_expr: None,
span: DUMMY_SP,
is_placeholder: false,
};
let varstr = variant_to_string(&var);
assert_eq!(varstr, "principal_skinner");
})
}
|
