@@ -266,6 +266,7 @@ func ppPostProcessing(ast *ASTNode, path []*ASTNode, ppString string) string {
// Add initial indent only if we are inside a block statement
if stringutil.IndexOf(parent.Name, []string{
+ NodeIN,
NodeASSIGN,
NodePRESET,
NodeKVP,
@@ -630,4 +630,20 @@ sink SomeSink
return
}
+ input = `
+
+ for [a,b,c] in foo {
+a := 1
+a := 2
+}
+`
+ if err := UnitTestPrettyPrinting(input, "",
+ `for [a, b, c] in foo {
+ a := 1
+ a := 2
+}`); err != nil {
+ t.Error(err)
+ return
+ }