@Override public Stubber doReturn(Object toBeReturned) { return stubber().doReturn(toBeReturned); }
@Override public Stubber doReturn(Object toBeReturned, Object... nextToBeReturned) { return stubber().doReturn(toBeReturned, nextToBeReturned); }
public BDDStubber willReturn(Object toBeReturned) { return new BDDStubberImpl(mockitoStubber.doReturn(toBeReturned)); }
public BDDStubber willReturn(Object toBeReturned) { return new BDDStubberImpl(mockitoStubber.doReturn(toBeReturned)); }
private void prepareMultipleRowsForExecutor() throws KettleException { doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value12" } ) .doReturn( new Object[] { "value12" } ) .doReturn( new Object[] { "value12" } ) .doReturn( null ) .when( executor ).getRow(); } }
private void prepareMultipleRowsForExecutor() throws KettleException { doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value1" } ) .doReturn( new Object[] { "value12" } ) .doReturn( new Object[] { "value12" } ) .doReturn( new Object[] { "value12" } ) .doReturn( null ) .when( executor ).getRow(); }
return MOCKITO_CORE.stubber().doReturn(toBeReturned);
@CheckReturnValue public static Stubber doReturn(Object toBeReturned, Object... toBeReturnedNext) { return MOCKITO_CORE.stubber().doReturn(toBeReturned, toBeReturnedNext);
private void prepareOneRowForExecutor() throws Exception { doReturn( new Object[] { "row" } ).doReturn( null ).when( executor ).getRow(); }
@Test public void testBasicReadObject_SocketExceptionReThrown() throws IOException, ClassNotFoundException { DataInput in = mock(DataInput.class); doReturn(DSCODE.DS_NO_FIXED_ID.toByte()).doReturn(DSCODE.CLASS.toByte()) .doReturn(DSCODE.STRING.toByte()).when(in).readByte(); doReturn( "org.apache.geode.internal.InternalDataSerializerJUnitTest$SocketExceptionThrowingDataSerializable") .when(in).readUTF(); assertThatThrownBy(() -> InternalDataSerializer.basicReadObject(in)) .isInstanceOf(SocketException.class); }
private List<StoreFileInfo> mockStoreFiles(Path regionStoreDir, int howMany, long timestamp) throws IOException { List<StoreFileInfo> infos = Lists.newArrayList(); int i = 0; while (i < howMany) { StoreFileInfo storeFileInfo = mock(StoreFileInfo.class); doReturn(timestamp).doReturn(timestamp).when(storeFileInfo).getModificationTime(); doReturn(new Path(regionStoreDir, RandomStringUtils.randomAlphabetic(10))).when(storeFileInfo) .getPath(); infos.add(storeFileInfo); i++; } return infos; }
@Before public void setUp() throws Exception { current = Locale.getDefault(); Locale.setDefault( Locale.UK ); helper = StepMockUtil.getStepMockHelper( SelectValuesMeta.class, "SelectValues_LocaleHandling_Test" ); when( helper.stepMeta.isDoingErrorHandling() ).thenReturn( true ); step = new SelectValues( helper.stepMeta, helper.stepDataInterface, 1, helper.transMeta, helper.trans ); step = spy( step ); // Dec 28, 2015 Calendar calendar = Calendar.getInstance(); calendar.set( 2015, Calendar.DECEMBER, 28, 0, 0, 0 ); doReturn( new Object[] { calendar.getTime() } ).doReturn( null ) .when( step ).getRow(); }
/** * Do Cache full exception * @throws IOException * @throws InterruptedException */ @Test public void testCacheFullException() throws IOException, InterruptedException { this.bc.cacheBlock(this.plainKey, plainCacheable); RAMQueueEntry rqe = q.remove(); RAMQueueEntry spiedRqe = Mockito.spy(rqe); final CacheFullException cfe = new CacheFullException(0, 0); BucketEntry mockedBucketEntry = Mockito.mock(BucketEntry.class); Mockito.doThrow(cfe). doReturn(mockedBucketEntry). when(spiedRqe).writeToCache(Mockito.any(), Mockito.any(), Mockito.any()); this.q.add(spiedRqe); doDrainOfOneEntry(bc, wt, q); }
@Test public void interceptConnect_retry() throws IOException { doThrow(RetryException.class).doReturn(connected).when(chain).processConnect(); interceptor.interceptConnect(chain); verify(chain, times(2)).processConnect(); verify(chain).resetConnectForRetry(); verify(cache, never()).catchException(any(IOException.class)); verify(outputStream, never()).catchBlockConnectException(chain.getBlockIndex()); }
@Test public void interceptConnect_failedReleaseConnection() throws IOException { final DownloadConnection connection = mock(DownloadConnection.class); when(chain.getConnection()).thenReturn(connection); doThrow(IOException.class).doReturn(connected).when(chain).processConnect(); thrown.expect(IOException.class); interceptor.interceptConnect(chain); verify(cache).catchException(any(IOException.class)); verify(outputStream).catchBlockConnectException(anyInt()); }
@Test public void canReturnNull() throws Exception { doReturn(beginResponse()) .doReturn(ApiFutures.immediateFailedFuture(new Exception())) .doReturn(beginResponse(ByteString.copyFromUtf8("foo2"))) .doReturn(commitResponse(0, 0)) .when(firestoreMock) .sendRequest(requestCapture.capture(), Matchers.<UnaryCallable<Message, Message>>any()); ApiFuture<String> transaction = firestoreMock.runTransaction( new Transaction.Function<String>() { @Override public String updateCallback(Transaction transaction) { return null; } }, options); assertEquals(null, transaction.get()); }
@Test public void delete_withoutTrashFolderExisting_shouldThrow() throws Exception { ImapFolder folder = createFolder("Folder"); prepareImapFolderForOpen(OPEN_MODE_RW); ImapFolder trashFolder = createFolder("Trash"); when(imapStore.getFolder("Trash")).thenReturn(trashFolder); List<ImapMessage> messages = singletonList(createImapMessage("2")); List<ImapResponse> copyResponses = singletonList( createImapResponse("x OK [COPYUID 23 2 102] Success") ); when(imapConnection.executeSimpleCommand("UID COPY 2 \"Trash\"")).thenReturn(copyResponses); folder.open(OPEN_MODE_RW); doThrow(NegativeImapResponseException.class).doReturn(Collections.emptyList()) .when(imapConnection).executeSimpleCommand("STATUS \"Trash\" (RECENT)"); try { folder.delete(messages, "Trash"); fail("Expected exception"); } catch (FolderNotFoundException e) { assertEquals("Trash", e.getFolderServerId()); } }
private void executeAndAssertResults( List<Object[]> expected ) throws Exception { FieldSplitterMeta meta = new FieldSplitterMeta(); meta.allocate( 3 ); meta.setFieldName( new String[] { "s1", "s2", "s3" } ); meta.setFieldType( new int[] { ValueMetaInterface.TYPE_STRING, ValueMetaInterface.TYPE_STRING, ValueMetaInterface.TYPE_STRING } ); meta.setSplitField( "string" ); meta.setDelimiter( "," ); FieldSplitterData data = new FieldSplitterData(); FieldSplitter step = createAndInitStep( meta, data ); RowMeta input = new RowMeta(); input.addValueMeta( new ValueMetaString( "string" ) ); step.setInputRowMeta( input ); step = spy( step ); doReturn( new String[] { "a, ,a" } ) .doReturn( new String[] { "b,,b" } ) .doReturn( new String[] { null } ) .when( step ).getRow(); List<Object[]> actual = TransTestingUtil.execute( step, meta, data, 3, false ); TransTestingUtil.assertResult( expected, actual ); }
private void executeAndAssertResults( List<Object[]> expected ) throws Exception { RegexEvalMeta meta = new RegexEvalMeta(); meta.allocate( 2 ); meta.getFieldName()[ 0 ] = "string"; meta.getFieldName()[ 1 ] = "matcher"; meta.setFieldType( new int[] { ValueMetaInterface.TYPE_STRING, ValueMetaInterface.TYPE_STRING } ); meta.setResultFieldName( "string" ); meta.setReplacefields( true ); meta.setMatcher( "matcher" ); meta.setAllowCaptureGroupsFlag( true ); RegexEvalData data = new RegexEvalData(); RegexEval step = createAndInitStep( meta, data ); RowMeta input = new RowMeta(); input.addValueMeta( new ValueMetaString( "string" ) ); input.addValueMeta( new ValueMetaString( "matcher" ) ); step.setInputRowMeta( input ); step = spy( step ); doReturn( new String[] { " ", " " } ) .doReturn( new String[] { "", "" } ) .doReturn( new String[] { null, null } ) .when( step ).getRow(); // dummy pattern, just to contain two groups // needed to activate a branch with conversion from string data.pattern = Pattern.compile( "(a)(a)" ); List<Object[]> actual = TransTestingUtil.execute( step, meta, data, 3, false ); TransTestingUtil.assertResult( expected, actual ); }