diff options
author | Zac Sweers <zac.sweers@gmail.com> | 2019-08-10 22:37:45 -0400 |
---|---|---|
committer | Zac Sweers <zac.sweers@gmail.com> | 2019-08-10 22:37:45 -0400 |
commit | 4e8f72f6a989ffbf99f0df4bbf0ab408ba412887 (patch) | |
tree | 9b0c197946732e7ade23170e8402fa61d01cdbcb | |
parent | d70a3035e5ed6d03108b26c557ff26aca34467dc (diff) | |
download | javapoet-4e8f72f6a989ffbf99f0df4bbf0ab408ba412887.tar.gz |
Add CodeBlock.Builder#clear() method
Analogous to the change added in KotlinPoet
-rw-r--r-- | src/main/java/com/squareup/javapoet/CodeBlock.java | 6 | ||||
-rw-r--r-- | src/test/java/com/squareup/javapoet/CodeBlockTest.java | 9 |
2 files changed, 15 insertions, 0 deletions
diff --git a/src/main/java/com/squareup/javapoet/CodeBlock.java b/src/main/java/com/squareup/javapoet/CodeBlock.java index 47c6ff7..02542f5 100644 --- a/src/main/java/com/squareup/javapoet/CodeBlock.java +++ b/src/main/java/com/squareup/javapoet/CodeBlock.java @@ -424,6 +424,12 @@ public final class CodeBlock { 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 index 2862809..11b75fa 100644 --- 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(); + } } |