1 files changed, 29 insertions, 16 deletions

diff --git a/crates/hir_expand/src/builtin_derive.rs b/crates/hir_expand/src/builtin_derive.rs
index 5e908b223..60fd2ebdd 100644
--- a/crates/hir_expand/src/builtin_derive.rs
+++ b/crates/hir_expand/src/builtin_derive.rs
@@ -61,8 +61,7 @@ pub fn find_builtin_derive(
     let expander = BuiltinDeriveExpander::find_by_name(ident)?;
     Some(MacroDefId {
         krate,
-        ast_id: Some(ast_id),
-        kind: MacroDefKind::BuiltInDerive(expander),
+        kind: MacroDefKind::BuiltInDerive(expander, ast_id),
         local_inner: false,
     })
 }
@@ -268,14 +267,13 @@ fn partial_ord_expand(
 mod tests {
     use base_db::{fixture::WithFixture, CrateId, SourceDatabase};
     use expect_test::{expect, Expect};
-    use name::{known, Name};
+    use name::AsName;
 
     use crate::{test_db::TestDB, AstId, MacroCallId, MacroCallKind, MacroCallLoc};
 
     use super::*;
 
-    fn expand_builtin_derive(ra_fixture: &str, name: Name) -> String {
-        let expander = BuiltinDeriveExpander::find_by_name(&name).unwrap();
+    fn expand_builtin_derive(ra_fixture: &str) -> String {
         let fixture = format!(
             r#"//- /main.rs crate:main deps:core
 $0
@@ -288,19 +286,34 @@ $0
 
         let (db, file_pos) = TestDB::with_position(&fixture);
         let file_id = file_pos.file_id;
+        let ast_id_map = db.ast_id_map(file_id.into());
         let parsed = db.parse(file_id);
-        let items: Vec<_> =
-            parsed.syntax_node().descendants().filter_map(ast::Item::cast).collect();
+        let macros: Vec<_> =
+            parsed.syntax_node().descendants().filter_map(ast::Macro::cast).collect();
+        let items: Vec<_> = parsed
+            .syntax_node()
+            .descendants()
+            .filter(|node| !ast::Macro::can_cast(node.kind()))
+            .filter_map(ast::Item::cast)
+            .collect();
+
+        assert_eq!(macros.len(), 1, "test must contain exactly 1 macro definition");
+        assert_eq!(items.len(), 1, "test must contain exactly 1 item");
+
+        let macro_ast_id = AstId::new(file_id.into(), ast_id_map.ast_id(&macros[0]));
+        let name = match &macros[0] {
+            ast::Macro::MacroRules(rules) => rules.name().unwrap().as_name(),
+            ast::Macro::MacroDef(def) => def.name().unwrap().as_name(),
+        };
 
-        let ast_id_map = db.ast_id_map(file_id.into());
+        let expander = BuiltinDeriveExpander::find_by_name(&name).unwrap();
 
         let attr_id = AstId::new(file_id.into(), ast_id_map.ast_id(&items[0]));
 
         let loc = MacroCallLoc {
             def: MacroDefId {
                 krate: CrateId(0),
-                ast_id: None,
-                kind: MacroDefKind::BuiltInDerive(expander),
+                kind: MacroDefKind::BuiltInDerive(expander, macro_ast_id),
                 local_inner: false,
             },
             krate: CrateId(0),
@@ -315,8 +328,8 @@ $0
         parsed.text().to_string()
     }
 
-    fn check_derive(ra_fixture: &str, name: Name, expected: Expect) {
-        let expanded = expand_builtin_derive(ra_fixture, name);
+    fn check_derive(ra_fixture: &str, expected: Expect) {
+        let expanded = expand_builtin_derive(ra_fixture);
         expected.assert_eq(&expanded);
     }
 
@@ -324,10 +337,10 @@ $0
     fn test_copy_expand_simple() {
         check_derive(
             r#"
+            macro Copy {}
             #[derive(Copy)]
             struct Foo;
             "#,
-            known::Copy,
             expect![["impl< >core::marker::CopyforFoo< >{}"]],
         );
     }
@@ -336,10 +349,10 @@ $0
     fn test_copy_expand_with_type_params() {
         check_derive(
             r#"
+            macro Copy {}
             #[derive(Copy)]
             struct Foo<A, B>;
             "#,
-            known::Copy,
             expect![["impl<T0:core::marker::Copy,T1:core::marker::Copy>core::marker::CopyforFoo<T0,T1>{}"]],
         );
     }
@@ -348,10 +361,10 @@ $0
     fn test_copy_expand_with_lifetimes() {
         check_derive(
             r#"
+            macro Copy {}
             #[derive(Copy)]
             struct Foo<A, B, 'a, 'b>;
             "#,
-            known::Copy,
             // We currently just ignore lifetimes
             expect![["impl<T0:core::marker::Copy,T1:core::marker::Copy>core::marker::CopyforFoo<T0,T1>{}"]],
         );
@@ -361,10 +374,10 @@ $0
     fn test_clone_expand() {
         check_derive(
             r#"
+            macro Clone {}
             #[derive(Clone)]
             struct Foo<A, B>;
             "#,
-            known::Clone,
             expect![["impl<T0:core::clone::Clone,T1:core::clone::Clone>core::clone::CloneforFoo<T0,T1>{}"]],
         );
     }