Test edge-cases

Author: GearsDatapacks

compiler-core/src/language_server/tests/action.rs

@@ -10372,3 +10372,68 @@ pub fn main() {
         find_position_of("case").select_until(find_position_of("}\n").under_char('\n'))
     );
 }
+
+#[test]
+fn extract_function_which_uses_constant() {
+    assert_code_action!(
+        EXTRACT_FUNCTION,
+        "
+const pi = 3.14
+
+pub fn main() {
+  let radius = 4.5
+
+  let circumference = radius *. pi *. 2.0
+
+  echo circumference
+}
+",
+        find_position_of("radius *.").select_until(find_position_of("2.0\n").under_char('\n'))
+    );
+}
+
+#[test]
+fn extract_function_which_uses_constant_in_guard() {
+    assert_code_action!(
+        EXTRACT_FUNCTION,
+        r#"
+const pi = 3.14
+
+pub fn main() {
+  let value = 3.15
+
+  let string = case value {
+    0.0 -> "Zero"
+    1.0 -> "One"
+    _ if value == pi -> "PI"
+    _ -> "Something else"
+  }
+
+  echo string
+}
+"#,
+        find_position_of("case").select_until(find_position_of("}\n").under_char('\n'))
+    );
+}
+
+#[test]
+fn no_code_action_to_extract_function_when_expression_is_not_fully_selected() {
+    assert_no_code_actions!(
+        EXTRACT_FUNCTION,
+        r#"
+fn print(text: String) { todo }
+
+pub fn main() {
+  let arguments = todo
+
+  case arguments {
+    ["help"] -> print("USAGE TEXT HERE")
+    _ -> panic as "Invalid args"
+  }
+}
+"#,
+        find_position_of("print")
+            .under_char('i')
+            .select_until(find_position_of("TEXT"))
+    );
+}

compiler-core/src/language_server/tests/snapshots/gleam_core__language_server__tests__action__extract_function_which_uses_constant.snap

@@ -0,0 +1,33 @@
+---
+source: compiler-core/src/language_server/tests/action.rs
+expression: "\nconst pi = 3.14\n\npub fn main() {\n  let radius = 4.5\n\n  let circumference = radius *. pi *. 2.0\n\n  echo circumference\n}\n"
+---
+----- BEFORE ACTION
+
+const pi = 3.14
+
+pub fn main() {
+  let radius = 4.5
+
+  let circumference = radius *. pi *. 2.0
+                      ▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+
+  echo circumference
+}
+
+
+----- AFTER ACTION
+
+const pi = 3.14
+
+pub fn main() {
+  let radius = 4.5
+
+  let circumference = function(radius)
+
+  echo circumference
+}
+
+fn function(radius: Float) -> Float {
+  radius *. pi *. 2.0
+}

compiler-core/src/language_server/tests/snapshots/gleam_core__language_server__tests__action__extract_function_which_uses_constant_in_guard.snap

@@ -0,0 +1,48 @@
+---
+source: compiler-core/src/language_server/tests/action.rs
+expression: "\nconst pi = 3.14\n\npub fn main() {\n  let value = 3.15\n\n  let string = case value {\n    0.0 -> \"Zero\"\n    1.0 -> \"One\"\n    _ if value == pi -> \"PI\"\n    _ -> \"Something else\"\n  }\n\n  echo string\n}\n"
+---
+----- BEFORE ACTION
+
+const pi = 3.14
+
+pub fn main() {
+  let value = 3.15
+
+  let string = case value {
+               ▔▔▔▔▔▔▔▔▔▔▔▔
+    0.0 -> "Zero"
+▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+    1.0 -> "One"
+▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+    _ if value == pi -> "PI"
+▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+    _ -> "Something else"
+▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+  }
+▔▔▔
+
+  echo string
+}
+
+
+----- AFTER ACTION
+
+const pi = 3.14
+
+pub fn main() {
+  let value = 3.15
+
+  let string = function(value)
+
+  echo string
+}
+
+fn function(value: Float) -> String {
+  case value {
+    0.0 -> "Zero"
+    1.0 -> "One"
+    _ if value == pi -> "PI"
+    _ -> "Something else"
+  }
+}