diff --git a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/retrieval/fanout/FanOutRecordsPublisherTest.java b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/retrieval/fanout/FanOutRecordsPublisherTest.java index f82329a1..ebdd5f40 100644 --- a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/retrieval/fanout/FanOutRecordsPublisherTest.java +++ b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/retrieval/fanout/FanOutRecordsPublisherTest.java @@ -5,7 +5,6 @@ import static org.junit.Assert.assertThat; import static org.junit.Assert.fail; import static org.mockito.Matchers.any; import static org.mockito.Mockito.doNothing; -import static org.mockito.Mockito.mock; import static org.mockito.Mockito.never; import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; @@ -56,6 +55,8 @@ public class FanOutRecordsPublisherTest { private SdkPublisher publisher; @Mock private Subscription subscription; + @Mock + private Subscriber subscriber; private SubscribeToShardEvent batchEvent; @@ -201,8 +202,6 @@ public class FanOutRecordsPublisherTest { .forClass(FanOutRecordsPublisher.RecordFlow.class); ArgumentCaptor inputCaptor = ArgumentCaptor.forClass(ProcessRecordsInput.class); - Subscriber subscriber = mock(Subscriber.class); - source.subscribe(subscriber); verify(kinesisClient).subscribeToShard(any(SubscribeToShardRequest.class), flowCaptor.capture());