Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/hir_expand/src/builtin_attr_macro.rs')
| -rw-r--r-- | crates/hir_expand/src/builtin_attr_macro.rs | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/crates/hir_expand/src/builtin_attr_macro.rs b/crates/hir_expand/src/builtin_attr_macro.rs index 2e461864a0..ec587daf9b 100644 --- a/crates/hir_expand/src/builtin_attr_macro.rs +++ b/crates/hir_expand/src/builtin_attr_macro.rs @@ -46,6 +46,16 @@ register_builtin! { (test_case, TestCase) => dummy_attr_expand } +pub fn is_builtin_test_or_bench_attr(makro: MacroDefId) -> bool { + match makro.kind { + MacroDefKind::BuiltInAttr(expander, ..) => { + BuiltinAttrExpander::find_by_name(&name!(test)) == Some(expander) + || BuiltinAttrExpander::find_by_name(&name!(bench)) == Some(expander) + } + _ => false, + } +} + pub fn find_builtin_attr( ident: &name::Name, krate: CrateId, |