RiverSourceTests.java 文件源码

@Test
@Parameters({"sql2", "n"})
public void testSimpleStarQuery(String sql, @Optional Integer n) throws Exception {
    List<Object> params = new LinkedList<Object>();
    RiverMouth output = new MockRiverMouth() {
        @Override
        public void index(IndexableObject object, boolean create) throws IOException {
            logger.debug("object={}", object);
        }
    };
    PreparedStatement statement = source.prepareQuery(sql);
    source.bind(statement, params);
    ResultSet results = source.executeQuery(statement);
    KeyValueStreamListener listener = new StringKeyValueStreamListener()
            .output(output);
    long rows = 0L;
    source.beforeRows(results, listener);
    while (source.nextRow(results, listener)) {
        rows++;
    }
    source.afterRows(results, listener);
    assertEquals(rows, n == null ? 5 : n);
    source.close(results);
    source.close(statement);
}