hclwrite: heredoc tokens are in line.lead, not line.assign (#95)

hclwrite/format.go

@@ -79,13 +79,13 @@ func formatIndent(lines []formatLine) {
 		netBrackets := 0
 		for _, token := range line.lead {
 			netBrackets += tokenBracketChange(token)
-		}
-		for _, token := range line.assign {
-			netBrackets += tokenBracketChange(token)
 			if token.Type == hclsyntax.TokenOHeredoc {
 				inHeredoc = true
 			}
 		}
+		for _, token := range line.assign {
+			netBrackets += tokenBracketChange(token)
+		}
 
 		switch {
 		case netBrackets > 0:

hclwrite/format_test.go

@@ -520,6 +520,30 @@ foo {
   ${a}${b}${c} ${d}
 EOT
 }
+`,
+		},
+		{
+			`
+foo {
+bar = <<EOT
+Foo bar baz
+EOT
+}
+
+baz {
+default="string"
+}
+`,
+			`
+foo {
+  bar = <<EOT
+Foo bar baz
+EOT
+}
+
+baz {
+  default = "string"
+}
 `,
 		},
 	}