feat(fmt): support formatting code blocks in Jupyter notebooks (#21310)

1 files changed, 20 insertions, 3 deletions

diff --git a/cli/tests/integration/fmt_tests.rs b/cli/tests/integration/fmt_tests.rs
index b8c77c529..5d555abf5 100644
--- a/cli/tests/integration/fmt_tests.rs
+++ b/cli/tests/integration/fmt_tests.rs
@@ -27,18 +27,24 @@ fn fmt_test() {
     testdata_fmt_dir.join("badly_formatted.json");
   let badly_formatted_json = t.path().join("badly_formatted.json");
   badly_formatted_original_json.copy(&badly_formatted_json);
-  // First, check formatting by ignoring the badly formatted file.
 
+  let fixed_ipynb = testdata_fmt_dir.join("badly_formatted_fixed.ipynb");
+  let badly_formatted_original_ipynb =
+    testdata_fmt_dir.join("badly_formatted.ipynb");
+  let badly_formatted_ipynb = t.path().join("badly_formatted.ipynb");
+  badly_formatted_original_ipynb.copy(&badly_formatted_ipynb);
+
+  // First, check formatting by ignoring the badly formatted file.
   let output = context
     .new_command()
     .current_dir(&testdata_fmt_dir)
     .args_vec(vec![
       "fmt".to_string(),
       format!(
-        "--ignore={badly_formatted_js},{badly_formatted_md},{badly_formatted_json}",
+        "--ignore={badly_formatted_js},{badly_formatted_md},{badly_formatted_json},{badly_formatted_ipynb}",
       ),
       format!(
-        "--check {badly_formatted_js} {badly_formatted_md} {badly_formatted_json}",
+        "--check {badly_formatted_js} {badly_formatted_md} {badly_formatted_json} {badly_formatted_ipynb}",
       ),
     ])
     .run();
@@ -57,6 +63,7 @@ fn fmt_test() {
       badly_formatted_js.to_string(),
       badly_formatted_md.to_string(),
       badly_formatted_json.to_string(),
+      badly_formatted_ipynb.to_string(),
     ])
     .run();
 
@@ -72,6 +79,7 @@ fn fmt_test() {
       badly_formatted_js.to_string(),
       badly_formatted_md.to_string(),
       badly_formatted_json.to_string(),
+      badly_formatted_ipynb.to_string(),
     ])
     .run();
 
@@ -81,12 +89,15 @@ 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_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_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_ipynb, actual_ipynb);
 }
 
 #[test]
@@ -198,6 +209,12 @@ itest!(fmt_stdin_json {
   output_str: Some("{ \"key\": \"value\" }\n"),
 });
 
+itest!(fmt_stdin_ipynb {
+  args: "fmt --ext=ipynb -",
+  input: Some(include_str!("../testdata/fmt/badly_formatted.ipynb")),
+  output_str: Some(include_str!("../testdata/fmt/badly_formatted_fixed.ipynb")),
+});
+
 itest!(fmt_stdin_check_formatted {
   args: "fmt --check -",
   input: Some("const a = 1;\n"),