diff --git a/crates/ide-assists/src/handlers/generate_delegate_trait.rs b/crates/ide-assists/src/handlers/generate_delegate_trait.rs index 92c10ec818a9..748acb46efbd 100644 --- a/crates/ide-assists/src/handlers/generate_delegate_trait.rs +++ b/crates/ide-assists/src/handlers/generate_delegate_trait.rs @@ -1716,4 +1716,65 @@ impl some_module::SomeTrait for B { }"#, ) } + + #[test] + fn test_fn_with_attrs() { + check_assist( + generate_delegate_trait, + r#" +struct A; + +trait T { + #[cfg(test)] + fn f(&self, a: u32); + #[cfg(not(test))] + fn f(&self, a: bool); +} + +impl T for A { + #[cfg(test)] + fn f(&self, a: u32) {} + #[cfg(not(test))] + fn f(&self, a: bool) {} +} + +struct B { + a$0: A, +} +"#, + r#" +struct A; + +trait T { + #[cfg(test)] + fn f(&self, a: u32); + #[cfg(not(test))] + fn f(&self, a: bool); +} + +impl T for A { + #[cfg(test)] + fn f(&self, a: u32) {} + #[cfg(not(test))] + fn f(&self, a: bool) {} +} + +struct B { + a: A, +} + +impl T for B { + #[cfg(test)] + fn f(&self, a: u32) { + ::f(&self.a, a) + } + + #[cfg(not(test))] + fn f(&self, a: bool) { + ::f(&self.a, a) + } +} +"#, + ); + } }