diff --git a/crates/ide_assists/src/handlers/reorder_impl.rs b/crates/ide_assists/src/handlers/reorder_impl.rs index 72d8892481c..54a9a468ecc 100644 --- a/crates/ide_assists/src/handlers/reorder_impl.rs +++ b/crates/ide_assists/src/handlers/reorder_impl.rs @@ -79,9 +79,12 @@ pub(crate) fn reorder_impl(acc: &mut Assists, ctx: &AssistContext) -> Option<()> "Sort methods", target, |builder| { - methods.into_iter().zip(sorted).for_each(|(old, new)| { - ted::replace(builder.make_ast_mut(old).syntax(), new.clone_for_update().syntax()) - }); + let methods = + methods.into_iter().map(|fn_| builder.make_ast_mut(fn_)).collect::>(); + methods + .into_iter() + .zip(sorted) + .for_each(|(old, new)| ted::replace(old.syntax(), new.clone_for_update().syntax())); }, ) } @@ -160,7 +163,7 @@ $0impl Bar for Foo {} } #[test] - fn reorder_impl_trait_methods() { + fn reorder_impl_trait_functions() { check_assist( reorder_impl, r#" @@ -197,4 +200,33 @@ impl Bar for Foo { "#, ) } + + #[test] + fn reorder_impl_trait_methods_uneven_ident_lengths() { + check_assist( + reorder_impl, + r#" +trait Bar { + fn foo(&mut self) {} + fn fooo(&mut self) {} +} + +struct Foo; +impl Bar for Foo { + fn fooo(&mut self) {} + fn foo(&mut self) {$0} +}"#, + r#" +trait Bar { + fn foo(&mut self) {} + fn fooo(&mut self) {} +} + +struct Foo; +impl Bar for Foo { + fn foo(&mut self) {} + fn fooo(&mut self) {} +}"#, + ) + } }