diff --git a/src/main/java/teetime/stage/io/File2SeqOfWords.java b/src/main/java/teetime/stage/io/File2SeqOfWords.java index 389fd1610365b34b346b0cc24fb7ad86fb9a2ff9..76062e4a833ff6ed31f4243c08dd7916b16fbe68 100644 --- a/src/main/java/teetime/stage/io/File2SeqOfWords.java +++ b/src/main/java/teetime/stage/io/File2SeqOfWords.java @@ -47,6 +47,10 @@ public final class File2SeqOfWords extends AbstractConsumerStage<File> { this("UTF-8", 1024); } + public File2SeqOfWords(final int bufferCapacity) { + this("UTF-8", bufferCapacity); + } + public File2SeqOfWords(final String charset, final int bufferCapacity) { super(); this.charset = charset; diff --git a/src/test/java/teetime/stage/io/File2SeqOfWordsTest.java b/src/test/java/teetime/stage/io/File2SeqOfWordsTest.java new file mode 100644 index 0000000000000000000000000000000000000000..66cdd8837505468be5d54b6e347903b6420c31cb --- /dev/null +++ b/src/test/java/teetime/stage/io/File2SeqOfWordsTest.java @@ -0,0 +1,25 @@ +package teetime.stage.io; + +import static org.junit.Assert.assertEquals; + +import java.io.File; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; + +import org.junit.Test; + +import teetime.framework.test.StageTester; + +public class File2SeqOfWordsTest { + + @Test + public void testExecute() throws Exception { + File2SeqOfWords stage = new File2SeqOfWords(14); + List<String> outputList = new ArrayList<String>(); + StageTester.test(stage).send(Arrays.asList(new File("./src/test/resources/data/input.txt"))).to(stage.getInputPort()).and().receive(outputList) + .from(stage.getOutputPort()).start(); + assertEquals(outputList.get(0), "Lorem ipsum"); + } + +}