test: add test for transformation indentation

Author: HerringtonDarkholme

crates/config/src/transform/mod.rs

@@ -118,4 +118,30 @@ mod test {
     let ret = Transform::deserialize(&trans, &env);
     assert!(ret.is_ok());
   }
+
+  #[test]
+  fn test_transform_indentation() {
+    let src = "
+if (true) {
+  let a = {
+    b: 123
+  }
+}
+";
+    let expected = "{
+  b: 123
+}";
+    let mut trans = HashMap::new();
+    let tr = from_str("{ substring: { source: $A } }").expect("should work");
+    trans.insert("TR".into(), tr);
+    let grep = TypeScript::Tsx.ast_grep(src);
+    let root = grep.root();
+    let mut nm = root.find("let a = $A").expect("should find");
+    let env = DeserializeEnv::new(TypeScript::Tsx);
+    let trans = Transform::deserialize(&trans, &env).expect("should deserialize");
+    trans.apply_transform(nm.get_env_mut(), &Default::default(), &Default::default());
+    let actual = nm.get_env().get_transformed("TR").expect("should have TR");
+    let actual = std::str::from_utf8(actual).expect("should work");
+    assert_eq!(actual, expected);
+  }
 }

crates/config/src/transform/transformation.rs

@@ -435,5 +435,25 @@ mod test {
     assert_eq!(actual, "camelcase_not");
     Ok(())
   }
+
+  #[test]
+  fn test_transform_indentation_with_insertion() -> R {
+    let src = "
+if (true) {
+  let a = {
+    b: 123
+  }
+}
+";
+    // note the indentation
+    let expected = "{
+    b: 123
+  }";
+    let tr = parse("{ substring: { source: $A } }")?;
+    let actual = get_transformed(src, "let a = $A", &tr).ok_or(())?;
+    assert_eq!(actual, expected);
+    Ok(())
+  }
+
   // TODO: add a symbolic test for Rewrite
 }