output.put((RestApiCommand) cmds.get(0), jsonStr); return output;
output.put((JdbcCommand) cmds.get(0), resultSet); return output;
output.put((JdbcCommand) cmds.get(0), resultSet); return output;
@BeforeClass public void setup() { output = new JdbcCommandOutput(); try { output.put(new JdbcCommand(), buildMockResultSet()); } catch (Exception e) { // hack for test failure assertEquals("PostgresqlExtractorTest: error initializing mock result set", "false"); } state = new WorkUnitState(); state.setId("id"); postgresqlExtractor = new PostgresqlExtractor((WorkUnitState) state); }
@Test public void testGetData() throws Exception { CommandOutput<JdbcCommand, ResultSet> output = new JdbcCommandOutput(); output.put(new JdbcCommand(), buildMockResultSet()); State state = new WorkUnitState(); state.setId("id"); JdbcExtractor jdbcExtractor = new MysqlExtractor((WorkUnitState) state); List<String> columnNames = Lists.newArrayListWithCapacity(COLUMNS.size()); for (MockJdbcColumn mockJdbcColumn:COLUMNS) { columnNames.add(mockJdbcColumn.getColumnName()); } jdbcExtractor.setHeaderRecord(columnNames); Iterator<JsonElement> itr = jdbcExtractor.getData(output); // Make sure there is an element in the iterator assertTrue(itr.hasNext()); JsonObject obj = itr.next().getAsJsonObject(); // Verify the columns for (MockJdbcColumn column : COLUMNS) { assertEquals(obj.get(column.getColumnName()).getAsString(), column.getValue()); } }
@BeforeClass public void setup() { output = new JdbcCommandOutput(); try { output.put(new JdbcCommand(), buildMockResultSet()); } catch (Exception e) { // hack for test failure assertEquals("OracleExtractorTest: error initializing mock result set", "false"); } state = new WorkUnitState(); state.setId("id"); oracleExtractor = new OracleExtractor((WorkUnitState) state); }
output.put((RestApiCommand) cmds.get(0), jsonStr); return output;
output.put((JdbcCommand) cmds.get(0), resultSet); return output;
output.put((JdbcCommand) cmds.get(0), resultSet); return output;