test: write more tests for code action

Author: ankddev

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

@@ -10692,3 +10692,68 @@ pub fn add_one(thing) {
         find_position_of("fn").select_until(find_position_of("("))
     );
 }
+
+fn annotate_all_top_level_definitions_already_annotated() {
+    assert_no_code_actions!(
+        ANNOTATE_TOP_LEVEL_DEFINITIONS,
+        r#"
+pub const answer: Int = 42
+
+pub fn add_two(thing: Int) -> {
+  thing + 2
+}
+
+pub fn add_one(thing: Int) -> Int {
+  thing + 1
+}
+"#,
+        find_position_of("fn").select_until(find_position_of("("))
+    );
+}
+
+#[test]
+fn annotate_all_top_level_definitions_inside_body() {
+    assert_no_code_actions!(
+        ANNOTATE_TOP_LEVEL_DEFINITIONS,
+        r#"
+pub fn add_one(thing) {
+  thing + 1
+}
+"#,
+        find_position_of("fn").select_until(find_position_of("("))
+    );
+}
+
+#[test]
+fn annotate_all_top_level_definitions_partially_annotated() {
+    assert_code_action!(
+        ANNOTATE_TOP_LEVEL_DEFINITIONS,
+        r#"
+pub const answer: Int = 42
+pub const another_answer = 43
+
+pub fn add_two(thing) -> Int {
+  thing + 2
+}
+
+pub fn add_one(thing: Int) {
+  thing + 1
+}
+"#,
+        find_position_of("fn").select_until(find_position_of("("))
+    );
+}
+
+// FIXME: make test working
+// #[test]
+// fn annotate_all_top_level_definitions_with_two_generic_functions() {
+//     assert_code_action!(
+//         ANNOTATE_TOP_LEVEL_DEFINITIONS,
+//         r#"
+// fn wibble(one) { todo }
+
+// fn wobble(other) { todo }
+// "#,
+//         find_position_of("wobble").to_selection()
+//     );
+// }

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

@@ -0,0 +1,31 @@
+---
+source: compiler-core/src/language_server/tests/action.rs
+expression: "\npub const answer: Int = 42\npub const another_answer = 43\n\npub fn add_two(thing) -> Int {\n  thing + 2\n}\n\npub fn add_one(thing: Int) {\n  thing + 1\n}\n"
+---
+----- BEFORE ACTION
+
+pub const answer: Int = 42
+pub const another_answer = 43
+
+pub fn add_two(thing) -> Int {
+    ▔▔▔▔▔▔▔▔▔▔↑               
+  thing + 2
+}
+
+pub fn add_one(thing: Int) {
+  thing + 1
+}
+
+
+----- AFTER ACTION
+
+pub const answer: Int = 42
+pub const another_answer: Int = 43
+
+pub fn add_two(thing: Int) -> Int {
+  thing + 2
+}
+
+pub fn add_one(thing: Int) -> Int {
+  thing + 1
+}