From d327f3036c2db0cffb91807c301d48f8e37165e2 Mon Sep 17 00:00:00 2001
From: roife <roifewu@gmail.com>
Date: Wed, 10 Jan 2024 15:00:58 +0800
Subject: [PATCH] Add test 'comments_in_block_expr' in in 'extract_function'

---
 .../src/handlers/extract_function.rs          | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/crates/ide-assists/src/handlers/extract_function.rs b/crates/ide-assists/src/handlers/extract_function.rs
index 3d91cb6a0f..1eb28626f7 100644
--- a/crates/ide-assists/src/handlers/extract_function.rs
+++ b/crates/ide-assists/src/handlers/extract_function.rs
@@ -5976,6 +5976,37 @@ fn $0fun_name() -> ControlFlow<()> {
         );
     }
 
+    #[test]
+    fn comments_in_block_expr() {
+        check_assist(
+            extract_function,
+            r#"
+fn f() {
+    let c = $0{
+        // comment 1
+        let a = 2 + 3;
+        // comment 2
+        let b = 5;
+        a + b
+    }$0;
+}
+"#,
+            r#"
+fn f() {
+    let c = fun_name();
+}
+
+fn $0fun_name() -> i32 {
+    // comment 1
+    let a = 2 + 3;
+    // comment 2
+    let b = 5;
+    a + b
+}
+"#,
+        );
+    }
+
     #[test]
     fn in_left_curly_is_not_applicable() {
         cov_mark::check!(extract_function_in_braces_is_not_applicable);