LightblueHttpClientTest.java 文件源码

@Test
public void testStreaming() throws Exception {
    when(httpTransport.executeRequestGetStream(any(LightblueRequest.class),anyString())).
        thenReturn(new StringBufferInputStream(streamingResponse.replaceAll("'","\"")));
    LightblueClientConfiguration c = new LightblueClientConfiguration();
    try (LightblueHttpClient httpClient = new LightblueHttpClient(c, httpTransport)) {
        DataFindRequest findRequest = new DataFindRequest("test");
        findRequest.where(Query.withValue("a = b"));
        findRequest.select(Projection.includeField("foo"));
        ResultStream response=httpClient.prepareFind(findRequest);
        final List<JsonNode> docs=new ArrayList<JsonNode>();
        response.run(new ResultStream.ForEachDoc() {
                @Override
                public boolean processDocument(ResultStream.StreamDoc doc) {
                    docs.add(doc.doc);
                    return true;
                }
            });
        Assert.assertEquals(9,docs.size());

    }
}