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";
@@ -11166,5 +11167,58 @@ fn merge_case_branch_does_not_merge_branches_with_variables_with_same_name_and_d
   }
 }"#,
         find_position_of("Ok").select_until(find_position_of("Error"))
+}
+
+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("("));
+}
+
+#[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_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("("))
     );
 }

compiler-core/src/language_server/tests/snapshots/gleam_core__language_server__tests__action__add_multiple_annotations.snap.new

@@ -0,0 +1,29 @@
+---
+source: compiler-core/src/language_server/tests/action.rs
+assertion_line: 2928
+expression: "\npub const my_constant = 20\n\npub fn add_my_constant(value) {\n  let result = value + my_constant\n  result\n}\n"
+snapshot_kind: text
+---
+----- BEFORE ACTION
+
+pub const my_constant = 20
+▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+
+pub fn add_my_constant(value) {
+▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+  let result = value + my_constant
+▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔▔
+  result
+▔▔▔▔▔▔▔▔
+}
+↑
+
+
+----- AFTER ACTION
+
+pub const my_constant: Int = 20
+
+pub fn add_my_constant(value: Int) -> Int {
+  let result = value + my_constant
+  result
+}

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
+}