feat(fmt): support YAML (#24717)

1 files changed, 13 insertions, 2 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);
 }