@Test
public void loops() throws Exception {

  String program =      "void main() { for (a; b; c) while (d) do e; while (f); }";
  String programAfter = "void main() { for (a; b; c) { while (d) { do { e; } while (f); } } }";
  TranslationUnit tu = ParseHelper.parse(program);
  TranslationUnit transformed = AddBraces.transform(tu);
  assertNotEquals(tu, transformed);
  assertEquals(PrettyPrinterVisitor.prettyPrintAsString(
      ParseHelper.parse(programAfter)),
      PrettyPrinterVisitor.prettyPrintAsString(transformed));
}