chore: write tests + snapshots

Author: ankddev

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

@@ -112,6 +112,7 @@ const ASSIGN_UNUSED_RESULT: &str = "Assign unused Result value to `_`";
 const ADD_MISSING_PATTERNS: &str = "Add missing patterns";
 const ADD_ANNOTATION: &str = "Add type annotation";
 const ADD_ANNOTATIONS: &str = "Add type annotations";
+const ANNOTATE_TOP_LEVEL_TYPE_DEFINITIONS: &str = "Annotate all top level type definitions";
 const CONVERT_FROM_USE: &str = "Convert from `use`";
 const CONVERT_TO_USE: &str = "Convert to `use`";
 const EXTRACT_VARIABLE: &str = "Extract variable";
@@ -10554,3 +10555,58 @@ pub fn main() {
         find_position_of("let c").select_until(find_position_of("* d"))
     );
 }
+
+#[test]
+fn annotate_all_top_level_definitions_function() {
+    assert_code_action!(
+        ANNOTATE_TOP_LEVEL_TYPE_DEFINITIONS,
+        r#"
+pub fn add_two(thing) {
+  thing + 2
+}
+
+pub fn add_one(thing) {
+  thing + 1
+}
+"#,
+        find_position_of("fn").select_until(find_position_of("("))
+    );
+}
+
+#[test]
+fn annotate_all_top_level_definitions_constant() {
+    assert_code_action!(
+        ANNOTATE_TOP_LEVEL_TYPE_DEFINITIONS,
+        r#"
+pub const answer = 42
+
+pub fn add_two(thing) {
+  thing + 2
+}
+
+pub fn add_one(thing) {
+  thing + 1
+}
+"#,
+        find_position_of("const").select_until(find_position_of("="))
+    );
+}
+
+#[test]
+fn annotate_all_top_level_definitions_dont_affect_local_vars() {
+    assert_code_action!(
+        ANNOTATE_TOP_LEVEL_TYPE_DEFINITIONS,
+        r#"
+pub const answer = 42
+
+pub fn add_two(thing) {
+  thing + 2
+
+pub fn add_one(thing) {
+  let result = thing + 1
+  result
+}
+"#,
+        find_position_of("fn").select_until(find_position_of("("))
+    );
+}

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

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

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

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

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

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