feat(fmt): support YAML (#24717)

5 files changed, 36 insertions, 3 deletions

diff --git a/tests/integration/fmt_tests.rs b/tests/integration/fmt_tests.rs
index 417454888..17adef6f8 100644
--- a/tests/integration/fmt_tests.rs
+++ b/tests/integration/fmt_tests.rs
@@ -37,6 +37,12 @@ fn fmt_test() {
   let badly_formatted_ipynb = t.path().join("badly_formatted.ipynb");
   badly_formatted_original_ipynb.copy(&badly_formatted_ipynb);
 
+  let fixed_yaml = testdata_fmt_dir.join("badly_formatted_fixed.yaml");
+  let badly_formatted_original_yaml =
+    testdata_fmt_dir.join("badly_formatted.yaml");
+  let badly_formatted_yaml = t.path().join("badly_formatted.yaml");
+  badly_formatted_original_yaml.copy(&badly_formatted_yaml);
+
   // First, check formatting by ignoring the badly formatted file.
   let output = context
     .new_command()
@@ -44,10 +50,10 @@ fn fmt_test() {
     .args_vec(vec![
       "fmt".to_string(),
       format!(
-        "--ignore={badly_formatted_js},{badly_formatted_md},{badly_formatted_json},{badly_formatted_ipynb}",
+        "--ignore={badly_formatted_js},{badly_formatted_md},{badly_formatted_json},{badly_formatted_yaml},{badly_formatted_ipynb}",
       ),
       format!(
-        "--check {badly_formatted_js} {badly_formatted_md} {badly_formatted_json} {badly_formatted_ipynb}",
+        "--check {badly_formatted_js} {badly_formatted_md} {badly_formatted_json} {badly_formatted_yaml} {badly_formatted_ipynb}",
       ),
     ])
     .run();
@@ -66,6 +72,7 @@ fn fmt_test() {
       badly_formatted_js.to_string(),
       badly_formatted_md.to_string(),
       badly_formatted_json.to_string(),
+      badly_formatted_yaml.to_string(),
       badly_formatted_ipynb.to_string(),
     ])
     .run();
@@ -82,6 +89,7 @@ fn fmt_test() {
       badly_formatted_js.to_string(),
       badly_formatted_md.to_string(),
       badly_formatted_json.to_string(),
+      badly_formatted_yaml.to_string(),
       badly_formatted_ipynb.to_string(),
     ])
     .run();
@@ -92,14 +100,17 @@ fn fmt_test() {
   let expected_js = fixed_js.read_to_string();
   let expected_md = fixed_md.read_to_string();
   let expected_json = fixed_json.read_to_string();
+  let expected_yaml = fixed_yaml.read_to_string();
   let expected_ipynb = fixed_ipynb.read_to_string();
   let actual_js = badly_formatted_js.read_to_string();
   let actual_md = badly_formatted_md.read_to_string();
   let actual_json = badly_formatted_json.read_to_string();
+  let actual_yaml = badly_formatted_yaml.read_to_string();
   let actual_ipynb = badly_formatted_ipynb.read_to_string();
   assert_eq!(expected_js, actual_js);
   assert_eq!(expected_md, actual_md);
   assert_eq!(expected_json, actual_json);
+  assert_eq!(expected_yaml, actual_yaml);
   assert_eq!(expected_ipynb, actual_ipynb);
 }
 
diff --git a/tests/testdata/fmt/badly_formatted.md b/tests/testdata/fmt/badly_formatted.md
index 26afe483b..05a4b2f97 100644
--- a/tests/testdata/fmt/badly_formatted.md
+++ b/tests/testdata/fmt/badly_formatted.md
@@ -43,4 +43,9 @@ function foo(): number {
   "numbers":
   ["1",      "2"]
 }
-```
\ No newline at end of file
+```
+
+```yaml
+-  item1
+-  item2
+```
diff --git a/tests/testdata/fmt/badly_formatted.yaml b/tests/testdata/fmt/badly_formatted.yaml
new file mode 100644
index 000000000..69c89f377
--- /dev/null
+++ b/tests/testdata/fmt/badly_formatted.yaml
@@ -0,0 +1,6 @@
+key1   :    value1
+key2 : value2
+key3:
+  - item1
+  -  item2
+  -    item3
diff --git a/tests/testdata/fmt/badly_formatted_fixed.md b/tests/testdata/fmt/badly_formatted_fixed.md
index 8ba74cac3..7a7d1913b 100644
--- a/tests/testdata/fmt/badly_formatted_fixed.md
+++ b/tests/testdata/fmt/badly_formatted_fixed.md
@@ -35,3 +35,8 @@ function foo(): number {
   "numbers": ["1", "2"]
 }
 ```
+
+```yaml
+- item1
+- item2
+```
diff --git a/tests/testdata/fmt/badly_formatted_fixed.yaml b/tests/testdata/fmt/badly_formatted_fixed.yaml
new file mode 100644
index 000000000..e167d34c2
--- /dev/null
+++ b/tests/testdata/fmt/badly_formatted_fixed.yaml
@@ -0,0 +1,6 @@
+key1: value1
+key2: value2
+key3:
+  - item1
+  - item2
+  - item3