From e780421036151d15230b0795be4b369aa2a86745 Mon Sep 17 00:00:00 2001
From: Sahil Palvia <spalvia@amazon.com>
Date: Fri, 10 Aug 2018 14:27:15 -0700
Subject: [PATCH] Removing mock and introducing @Mock

---
 .../kinesis/retrieval/fanout/FanOutRecordsPublisherTest.java | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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<SubscribeToShardEventStream> publisher;
     @Mock
     private Subscription subscription;
+    @Mock
+    private Subscriber<ProcessRecordsInput> subscriber;
 
     private SubscribeToShardEvent batchEvent;
 
@@ -201,8 +202,6 @@ public class FanOutRecordsPublisherTest {
                 .forClass(FanOutRecordsPublisher.RecordFlow.class);
         ArgumentCaptor<ProcessRecordsInput> inputCaptor = ArgumentCaptor.forClass(ProcessRecordsInput.class);
 
-        Subscriber<ProcessRecordsInput> subscriber = mock(Subscriber.class);
-
         source.subscribe(subscriber);
 
         verify(kinesisClient).subscribeToShard(any(SubscribeToShardRequest.class), flowCaptor.capture());