nextInputStream(); inputRowSet = currentInputStream(); row = inputRowSet.getRowImmediate();
public static List<Object[]> execute( BaseStep step, StepMetaInterface meta, StepDataInterface data, int expectedRowsAmount, boolean checkIsDone ) throws Exception { RowSet output = new BlockingRowSet( Math.max( 1, expectedRowsAmount ) ); step.setOutputRowSets( Collections.singletonList( output ) ); int i = 0; List<Object[]> result = new ArrayList<>( expectedRowsAmount ); while ( step.processRow( meta, data ) && i < expectedRowsAmount ) { Object[] row = output.getRowImmediate(); assertNotNull( Integer.toString( i ), row ); result.add( row ); i++; } assertEquals( "The amount of executions should be equal to expected", expectedRowsAmount, i ); if ( checkIsDone ) { assertTrue( output.isDone() ); } return result; }
public void doTest( String content, String enclosure ) throws Exception { RowSet output = new QueueRowSet(); File tmp = createTestFile( "utf-8", content ); try { CsvInputMeta meta = createMeta( tmp, createInputFileFields( "f1", "f2" ), enclosure ); CsvInputData data = new CsvInputData(); csvInput.init( meta, data ); csvInput.addRowSetToOutputRowSets( output ); try { csvInput.processRow( meta, data ); } finally { csvInput.dispose( meta, data ); } } finally { tmp.delete(); } Object[] row = output.getRowImmediate(); assertNotNull( row ); assertEquals( "value1", row[ 0 ] ); assertEquals( "value2", row[ 1 ] ); assertNull( output.getRowImmediate() ); }
public void doTest( String content, String[] expected ) throws Exception { RowSet output = new QueueRowSet(); File tmp = createTestFile( encoding, content ); try { CsvInputMeta meta = createMeta( tmp, createInputFileFields( "f1", "f2", "f3" ) ); CsvInputData data = new CsvInputData(); csvInput.init( meta, data ); csvInput.addRowSetToOutputRowSets( output ); try { csvInput.processRow( meta, data ); } finally { csvInput.dispose( meta, data ); } } finally { tmp.delete(); } Object[] row = output.getRowImmediate(); assertNotNull( row ); assertEquals( expected[0], row[0] ); assertEquals( expected[1], row[1] ); assertEquals( expected[2], row[2] ); assertNull( output.getRowImmediate() ); }
private void doTest( String content ) throws Exception { RowSet output = new QueueRowSet(); File tmp = createTestFile( "utf-8", content ); try { CsvInputMeta meta = createMeta( tmp, createInputFileFields( "f1", "f2", "f3" ) ); CsvInputData data = new CsvInputData(); csvInput.init( meta, data ); csvInput.addRowSetToOutputRowSets( output ); try { csvInput.processRow( meta, data ); } finally { csvInput.dispose( meta, data ); } } finally { tmp.delete(); } Object[] row = output.getRowImmediate(); assertNotNull( row ); assertEquals( "value1", row[ 0 ] ); assertEquals( "value2", row[ 1 ] ); assertEquals( "value3", row[ 2 ] ); assertNull( output.getRowImmediate() ); }
@Test public void collectsResultsFromInternalTransformation() throws Exception { prepareOneRowForExecutor(); RowMetaAndData expectedResult = new RowMetaAndData( new RowMeta(), "fake result" ); internalResult.getRows().add( expectedResult ); RowSet rowSet = new QueueRowSet(); // any value except null StepMeta stepMeta = mockStepAndMapItToRowSet( "stepMetaMock", rowSet ); meta.setOutputRowsSourceStepMeta( stepMeta ); executor.init( meta, data ); executor.setInputRowMeta( new RowMeta() ); assertTrue( "Passing one line at first time", executor.processRow( meta, data ) ); assertFalse( "Executing the internal trans during the second round", executor.processRow( meta, data ) ); Object[] resultsRow = rowSet.getRowImmediate(); assertNotNull( resultsRow ); assertArrayEquals( expectedResult.getData(), resultsRow ); assertNull( "Only one row is expected", rowSet.getRowImmediate() ); }
assertFalse( "Executing the internal trans during the second round", executor.processRow( meta, data ) ); Object[] resultsRow = rowSet.getRowImmediate(); assertNotNull( resultsRow ); assertNull( "Only one row is expected", rowSet.getRowImmediate() );