1 files changed, 123 insertions, 0 deletions
diff --git a/cli/tests/integration_tests_lsp.rs b/cli/tests/integration_tests_lsp.rs
index 6b0f6b792..fa88ad17d 100644
--- a/cli/tests/integration_tests_lsp.rs
+++ b/cli/tests/integration_tests_lsp.rs
@@ -1659,3 +1659,126 @@ fn lsp_performance() {
   }
   shutdown(&mut client);
 }
+
+#[test]
+fn lsp_format_json() {
+  let mut client = init("initialize_params.json");
+  client
+    .write_notification(
+      "textDocument/didOpen",
+      json!({
+        "textDocument": {
+          "uri": "file:///a/file.json",
+          "languageId": "json",
+          "version": 1,
+          "text": "{\"key\":\"value\"}"
+        }
+      }),
+    )
+    .unwrap();
+
+  let (maybe_res, maybe_err) = client
+    .write_request::<_, _, Value>(
+      "textDocument/formatting",
+      json!({
+          "textDocument": {
+            "uri": "file:///a/file.json"
+          },
+          "options": {
+            "tabSize": 2,
+            "insertSpaces": true
+          }
+      }),
+    )
+    .unwrap();
+
+  assert!(maybe_err.is_none());
+  assert_eq!(
+    maybe_res,
+    Some(json!([
+      {
+        "range": {
+          "start": {
+            "line": 0,
+            "character": 1
+          },
+          "end": {
+            "line": 0,
+            "character": 1
+          }
+        },
+        "newText": " "
+      },
+      {
+        "range": {
+          "start": { "line": 0, "character": 7 },
+          "end": { "line": 0, "character": 7 }
+        },
+        "newText": " "
+      },
+      {
+        "range": {
+          "start": { "line": 0, "character": 14 },
+          "end": { "line": 0, "character": 15 }
+        },
+        "newText": " }\n"
+      }
+    ]))
+  );
+  shutdown(&mut client);
+}
+
+#[test]
+fn lsp_format_markdown() {
+  let mut client = init("initialize_params.json");
+  client
+    .write_notification(
+      "textDocument/didOpen",
+      json!({
+        "textDocument": {
+          "uri": "file:///a/file.md",
+          "languageId": "markdown",
+          "version": 1,
+          "text": "#   Hello World"
+        }
+      }),
+    )
+    .unwrap();
+
+  let (maybe_res, maybe_err) = client
+    .write_request::<_, _, Value>(
+      "textDocument/formatting",
+      json!({
+        "textDocument": {
+          "uri": "file:///a/file.md"
+        },
+        "options": {
+          "tabSize": 2,
+          "insertSpaces": true
+        }
+      }),
+    )
+    .unwrap();
+
+  assert!(maybe_err.is_none());
+  assert_eq!(
+    maybe_res,
+    Some(json!([
+      {
+        "range": {
+          "start": { "line": 0, "character": 1 },
+          "end": { "line": 0, "character": 3 }
+        },
+        "newText": ""
+      },
+      {
+        "range": {
+          "start": { "line": 0, "character": 15 },
+          "end": { "line": 0, "character": 15 }
+        },
+        "newText": "\n"
+      }
+    ]))
+  );
+  shutdown(&mut client);
+}