Removing mock and introducing @Mock
This commit is contained in:
parent
cd7dc1f9b1
commit
e780421036
1 changed files with 2 additions and 3 deletions
|
|
@ -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());
|
||||
|
|
|
|||
Loading…
Reference in a new issue