Add CodeBlock.Builder#clear() method

Analogous to the change added in KotlinPoet
square · Aug 11, 2019 · 4e8f72f · 4e8f72f
1 parent d70a303
commit 4e8f72f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/main/java/com/squareup/javapoet/CodeBlock.java b/src/main/java/com/squareup/javapoet/CodeBlock.java
@@ -424,6 +424,12 @@ public Builder unindent() {
       return this;
     }
 
+    public Builder clear() {
+      formatParts.clear();
+      args.clear();
+      return this;
+    }
+
     public CodeBlock build() {
       return new CodeBlock(this);
     }

diff --git a/src/test/java/com/squareup/javapoet/CodeBlockTest.java b/src/test/java/com/squareup/javapoet/CodeBlockTest.java
@@ -339,4 +339,13 @@ public final class CodeBlockTest {
     CodeBlock joined = codeBlocks.stream().collect(CodeBlock.joining(" || ", "start {", "} end"));
     assertThat(joined.toString()).isEqualTo("start {\"hello\" || world.World || need tacos} end");
   }
+
+  @Test public void clear() {
+    CodeBlock block = CodeBlock.builder()
+        .addStatement("$S", "Test string")
+        .clear()
+        .build();
+
+    assertThat(block.toString()).isEmpty();
+  }
 }