69 lines
1.7 KiB
Rust
69 lines
1.7 KiB
Rust
use super::*;
|
|
|
|
use rustc_span;
|
|
use rustc_span::source_map::{dummy_spanned, respan};
|
|
use syntax::ast;
|
|
use syntax::attr::with_default_globals;
|
|
|
|
fn fun_to_string(
|
|
decl: &ast::FnDecl,
|
|
header: ast::FnHeader,
|
|
name: ast::Ident,
|
|
generics: &ast::Generics,
|
|
) -> String {
|
|
to_string(|s| {
|
|
s.head("");
|
|
s.print_fn(
|
|
decl,
|
|
header,
|
|
Some(name),
|
|
generics,
|
|
&dummy_spanned(ast::VisibilityKind::Inherited),
|
|
);
|
|
s.end(); // Close the head box.
|
|
s.end(); // Close the outer box.
|
|
})
|
|
}
|
|
|
|
fn variant_to_string(var: &ast::Variant) -> String {
|
|
to_string(|s| s.print_variant(var))
|
|
}
|
|
|
|
#[test]
|
|
fn test_fun_to_string() {
|
|
with_default_globals(|| {
|
|
let abba_ident = ast::Ident::from_str("abba");
|
|
|
|
let decl = ast::FnDecl {
|
|
inputs: Vec::new(),
|
|
output: ast::FunctionRetTy::Default(rustc_span::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() {
|
|
with_default_globals(|| {
|
|
let ident = ast::Ident::from_str("principal_skinner");
|
|
|
|
let var = ast::Variant {
|
|
ident,
|
|
vis: respan(rustc_span::DUMMY_SP, ast::VisibilityKind::Inherited),
|
|
attrs: Vec::new(),
|
|
id: ast::DUMMY_NODE_ID,
|
|
data: ast::VariantData::Unit(ast::DUMMY_NODE_ID),
|
|
disr_expr: None,
|
|
span: rustc_span::DUMMY_SP,
|
|
is_placeholder: false,
|
|
};
|
|
|
|
let varstr = variant_to_string(&var);
|
|
assert_eq!(varstr, "principal_skinner");
|
|
})
|
|
}
|