luciano/amazon-kinesis-client
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Removing mock and introducing @Mock

Browse source
This commit is contained in:
Sahil Palvia 2018-08-10 14:27:15 -07:00
parent cd7dc1f9b1
commit e780421036
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
amazon-kinesis-client/src/test/java/software/amazon/kinesis/retrieval/fanout/FanOutRecordsPublisherTest.java
View file

@ -5,7 +5,6 @@ import static org.junit.Assert.assertThat;
import static org.junit.Assert.fail; import static org.junit.Assert.fail;
import static org.mockito.Matchers.any; import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doNothing; import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never; import static org.mockito.Mockito.never;
import static org.mockito.Mockito.times; import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verify;
@ -56,6 +55,8 @@ public class FanOutRecordsPublisherTest {
private SdkPublisher<SubscribeToShardEventStream> publisher; private SdkPublisher<SubscribeToShardEventStream> publisher;
@Mock @Mock
private Subscription subscription; private Subscription subscription;
@Mock
private Subscriber<ProcessRecordsInput> subscriber;
private SubscribeToShardEvent batchEvent; private SubscribeToShardEvent batchEvent;
@ -201,8 +202,6 @@ public class FanOutRecordsPublisherTest {
.forClass(FanOutRecordsPublisher.RecordFlow.class); .forClass(FanOutRecordsPublisher.RecordFlow.class);
ArgumentCaptor<ProcessRecordsInput> inputCaptor = ArgumentCaptor.forClass(ProcessRecordsInput.class); ArgumentCaptor<ProcessRecordsInput> inputCaptor = ArgumentCaptor.forClass(ProcessRecordsInput.class);
Subscriber<ProcessRecordsInput> subscriber = mock(Subscriber.class);
source.subscribe(subscriber); source.subscribe(subscriber);
verify(kinesisClient).subscribeToShard(any(SubscribeToShardRequest.class), flowCaptor.capture()); verify(kinesisClient).subscribeToShard(any(SubscribeToShardRequest.class), flowCaptor.capture());