Examples with Mockito.when org.mockito.Mockito.when used on opensource projects

Example 26 with Mockito.when

use of org.mockito.Mockito.when in project pentaho-kettle by pentaho.

the class TextFileInputTest method testFolderFromPreviousStep.

/**
 * This test handles the case where a folder is given for the target of the previous step. Note, that it uses
 * 2 ram files and sets the mock to return "ram://." to trigger the directory logic.
 * @throws Exception - Test exception
 */
@Test
public void testFolderFromPreviousStep() throws Exception {
    final String virtualFile = createVirtualFile("test-file1.txt", "1,\n");
    final String virtualFile2 = createVirtualFile("test-file2.txt", "1,\n");
    BaseFileField col2 = field("col2");
    col2.setIfNullValue("DEFAULT");
    TextFileInputMeta meta = createMetaObject(field("col1"), col2);
    VariableSpace space = new Variables();
    space.initializeVariablesFrom(null);
    meta.inputFiles.passingThruFields = true;
    meta.inputFiles.acceptingFilenames = true;
    TextFileInputData data = createDataObject(virtualFile, ",", "col1", "col2");
    data.files.addFile(KettleVFS.getFileObject(virtualFile2));
    TextFileInput input = Mockito.spy(StepMockUtil.getStep(TextFileInput.class, TextFileInputMeta.class, "test"));
    RowSet rowset = Mockito.mock(RowSet.class);
    RowMetaInterface rwi = Mockito.mock(RowMetaInterface.class);
    Object[] obj1 = new Object[2];
    Object[] obj2 = new Object[2];
    Mockito.doReturn(rowset).when(input).findInputRowSet(null);
    Mockito.doReturn(null).when(input).getRowFrom(rowset);
    Mockito.when(input.getRowFrom(rowset)).thenReturn(obj1, null);
    Mockito.when(input.getTransMeta().listVariables()).thenReturn(space.listVariables());
    Mockito.when(input.getTransMeta().getVariable(anyString())).thenAnswer((Answer<String>) invocation -> space.getVariable((String) invocation.getArguments()[0]));
    Mockito.doReturn(rwi).when(rowset).getRowMeta();
    Mockito.when(rwi.getString(obj2, 0)).thenReturn("ram:///.");
    List<Object[]> output = TransTestingUtil.execute(input, meta, data, 0, false);
    List<String> passThroughKeys = new ArrayList<>(data.passThruFields.keySet());
    Assert.assertNotNull(passThroughKeys);
    // set order is not guaranteed - order alphabetically
    passThroughKeys.sort(String.CASE_INSENSITIVE_ORDER);
    assertEquals(2, passThroughKeys.size());
    Assert.assertNotNull(passThroughKeys.get(0));
    Assert.assertTrue(passThroughKeys.get(0).startsWith("0_ram"));
    Assert.assertTrue(passThroughKeys.get(0).endsWith("test-file1.txt"));
    Assert.assertNotNull(passThroughKeys.get(1));
    Assert.assertTrue(passThroughKeys.get(1).startsWith("1_ram"));
    Assert.assertTrue(passThroughKeys.get(1).endsWith("test-file2.txt"));
    deleteVfsFile(virtualFile);
    deleteVfsFile(virtualFile2);
}