From 601108c71908a8e0bfe344c2b06306858a7b16f4 Mon Sep 17 00:00:00 2001 From: "Pfifer, Justin" Date: Wed, 27 Sep 2017 10:09:20 -0700 Subject: [PATCH] Fix unit test for behavior change of TerminalResult Got distracted, and forgot to run the unit tests. --- .../lib/worker/KinesisDataFetcherTest.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisDataFetcherTest.java b/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisDataFetcherTest.java index bbf72734..dbac3a54 100644 --- a/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisDataFetcherTest.java +++ b/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisDataFetcherTest.java @@ -18,6 +18,7 @@ import static org.hamcrest.CoreMatchers.equalTo; import static org.hamcrest.CoreMatchers.notNullValue; import static org.hamcrest.CoreMatchers.nullValue; import static org.hamcrest.MatcherAssert.assertThat; +import static org.hamcrest.collection.IsEmptyCollection.empty; import static org.mockito.Matchers.anyInt; import static org.mockito.Matchers.anyString; import static org.mockito.Matchers.eq; @@ -34,6 +35,9 @@ import java.util.ArrayList; import java.util.Date; import java.util.List; +import org.hamcrest.CoreMatchers; +import org.hamcrest.Matcher; +import org.hamcrest.Matchers; import org.junit.Assert; import org.junit.BeforeClass; import org.junit.Test; @@ -269,7 +273,11 @@ public class KinesisDataFetcherTest { DataFetcherResult terminal = dataFetcher.getRecords(100); assertThat(terminal.isShardEnd(), equalTo(true)); - assertThat(terminal.getResult(), nullValue()); + assertThat(terminal.getResult(), notNullValue()); + GetRecordsResult terminalResult = terminal.getResult(); + assertThat(terminalResult.getRecords(), notNullValue()); + assertThat(terminalResult.getRecords(), empty()); + assertThat(terminalResult.getNextShardIterator(), nullValue()); assertThat(terminal, equalTo(dataFetcher.TERMINAL_RESULT)); verify(kinesisProxy, never()).get(anyString(), anyInt());