diff --git a/crates/ide-assists/src/handlers/generate_enum_variant.rs b/crates/ide-assists/src/handlers/generate_enum_variant.rs index fa6242460b0f4..9b3fd8ed87cdd 100644 --- a/crates/ide-assists/src/handlers/generate_enum_variant.rs +++ b/crates/ide-assists/src/handlers/generate_enum_variant.rs @@ -136,6 +136,43 @@ enum Foo { fn main() { Foo::Baz } +", + ) + } + + #[test] + fn generate_basic_enum_variant_in_different_file() { + check_assist( + generate_enum_variant, + r" +//- /main.rs +mod foo; +use foo::Foo; + +fn main() { + Foo::Baz$0 +} + +//- /foo.rs +enum Foo { + Bar, +} +", + r" +//- /main.rs +mod foo; +use foo::Foo; + +fn main() { + Foo::Baz +} + +//- /foo.rs +enum Foo { + Bar, + Baz, +} + ", ) }