Fix edge case for ImportGranularity guessing

1 files changed, 15 insertions, 0 deletions

diff --git a/crates/ide_db/src/helpers/insert_use/tests.rs b/crates/ide_db/src/helpers/insert_use/tests.rs
index 78a2a87b3..70b11bf81 100644
--- a/crates/ide_db/src/helpers/insert_use/tests.rs
+++ b/crates/ide_db/src/helpers/insert_use/tests.rs
@@ -663,6 +663,13 @@ use foo::bar::qux;
 ",
         ImportGranularityGuess::Item,
     );
+    check_guess(
+        r"
+use foo::bar::Bar;
+use foo::baz;
+",
+        ImportGranularityGuess::Item,
+    );
 }
 
 #[test]
@@ -682,6 +689,14 @@ use foo::{baz::{qux, quux}, bar};
 ",
         ImportGranularityGuess::Module,
     );
+    check_guess(
+        r"
+use foo::bar::Bar;
+use foo::baz::Baz;
+use foo::{Foo, Qux};
+",
+        ImportGranularityGuess::Module,
+    );
 }
 
 #[test]