Fixed test.

nwaldispuehl · Dec 28, 2013 · ac5c39c · ac5c39c
1 parent 53c8c3d
commit ac5c39c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/...thesizor.builder/src/main/java/ch/retorte/textsynthesizor/builder/MarkovChainBuilder.java b/...thesizor.builder/src/main/java/ch/retorte/textsynthesizor/builder/MarkovChainBuilder.java
@@ -122,7 +122,8 @@ String generateStringOfLength(int tokens) {
     return on(tokenizer.getDelimiter()).join(result);
   }
 
-  private boolean isChainEmpty() {
+  @VisibleForTesting
+  boolean isChainEmpty() {
     return chain.isEmpty();
   }
 

diff --git a/...izor.builder/src/test/java/ch/retorte/textsynthesizor/builder/MarkovChainBuilderTest.java b/...izor.builder/src/test/java/ch/retorte/textsynthesizor/builder/MarkovChainBuilderTest.java
@@ -87,6 +87,7 @@ public void shouldGenerateStringOfLength() {
     builder.initializeBufferWithSize(1);
     Token createdToken = createTokenWith("abc");
     when(builder.extractNextToken()).thenReturn(createdToken);
+    when(builder.isChainEmpty()).thenReturn(false);
 
     // when
     builder.generateStringOfLength(1);