| // http://rust-lang.org/COPYRIGHT. |
| // |
| |
| // #[test_case] is used by custom test authors to mark tests |
| // When building for test, it needs to make the item public and gensym the name |
| // Otherwise, we'll omit the item. This behavior means that any item annotated |
| // with #[test_case] is never addressable. |
| // |
| // We mark item with an inert attribute "rustc_test_marker" which the test generation |
| // logic will pick up on. |
| |
| use syntax::ext::base::*; |
| use syntax::ext::build::AstBuilder; |
| use syntax::ext::hygiene::{self, Mark, SyntaxContext}; |
| use syntax::ast; |
| use syntax::source_map::respan; |
| use syntax::symbol::Symbol; |
| use syntax_pos::{DUMMY_SP, Span}; |
| use syntax::source_map::{ExpnInfo, MacroAttribute}; |
| use syntax::feature_gate; |
| |
| pub fn expand( |
| ecx: &mut ExtCtxt, |
| attr_sp: Span, |
| _meta_item: &ast::MetaItem, |
| anno_item: Annotatable |
| ) -> Vec<Annotatable> { |
| if !ecx.ecfg.enable_custom_test_frameworks() { |
| feature_gate::emit_feature_err(&ecx.parse_sess, |
| "custom_test_frameworks", |
| attr_sp, |
| feature_gate::GateIssue::Language, |
| feature_gate::EXPLAIN_CUSTOM_TEST_FRAMEWORKS); |
| } |
| |
| if !ecx.ecfg.should_test { return vec![]; } |
| |
| let sp = { |
| let mark = Mark::fresh(Mark::root()); |
| mark.set_expn_info(ExpnInfo { |
| call_site: DUMMY_SP, |
| def_site: None, |
| format: MacroAttribute(Symbol::intern("test_case")), |
| allow_internal_unstable: true, |
| allow_internal_unsafe: false, |
| local_inner_macros: false, |
| edition: hygiene::default_edition(), |
| }); |
| attr_sp.with_ctxt(SyntaxContext::empty().apply_mark(mark)) |
| }; |
| |
| let mut item = anno_item.expect_item(); |
| |
| item = item.map(|mut item| { |
| item.vis = respan(item.vis.span, ast::VisibilityKind::Public); |
| item.ident = item.ident.gensym(); |
| item.attrs.push( |
| ecx.attribute(sp, |
| ecx.meta_word(sp, Symbol::intern("rustc_test_marker"))) |
| ); |
| item |
| }); |
| |
| return vec![Annotatable::Item(item)] |
| } |