Examples with InputStreamStreamInput org.elasticsearch.common.io.stream.InputStreamStreamInput used on opensource projects

Example 16 with InputStreamStreamInput

use of org.elasticsearch.common.io.stream.InputStreamStreamInput in project elasticsearch by elastic.

the class SignificanceHeuristicTests method testStreamResponse.

// test that stream output can actually be read - does not replace bwc test
public void testStreamResponse() throws Exception {
    Version version = randomVersion(random());
    InternalMappedSignificantTerms<?, ?> sigTerms = getRandomSignificantTerms(getRandomSignificanceheuristic());
    // write
    ByteArrayOutputStream outBuffer = new ByteArrayOutputStream();
    OutputStreamStreamOutput out = new OutputStreamStreamOutput(outBuffer);
    out.setVersion(version);
    out.writeNamedWriteable(sigTerms);
    // read
    ByteArrayInputStream inBuffer = new ByteArrayInputStream(outBuffer.toByteArray());
    StreamInput in = new InputStreamStreamInput(inBuffer);
    // populates the registry through side effects
    SearchModule searchModule = new SearchModule(Settings.EMPTY, false, emptyList());
    NamedWriteableRegistry registry = new NamedWriteableRegistry(searchModule.getNamedWriteables());
    in = new NamedWriteableAwareStreamInput(in, registry);
    in.setVersion(version);
    InternalMappedSignificantTerms<?, ?> read = (InternalMappedSignificantTerms<?, ?>) in.readNamedWriteable(InternalAggregation.class);
    assertEquals(sigTerms.significanceHeuristic, read.significanceHeuristic);
    SignificantTerms.Bucket originalBucket = sigTerms.getBuckets().get(0);
    SignificantTerms.Bucket streamedBucket = read.getBuckets().get(0);
    assertThat(originalBucket.getKeyAsString(), equalTo(streamedBucket.getKeyAsString()));
    assertThat(originalBucket.getSupersetDf(), equalTo(streamedBucket.getSupersetDf()));
    assertThat(originalBucket.getSubsetDf(), equalTo(streamedBucket.getSubsetDf()));
    assertThat(streamedBucket.getSubsetSize(), equalTo(10L));
    assertThat(streamedBucket.getSupersetSize(), equalTo(20L));
}

Example 17 with InputStreamStreamInput

use of org.elasticsearch.common.io.stream.InputStreamStreamInput in project elasticsearch by elastic.

the class TermVectorsUnitTests method testStreamResponse.

public void testStreamResponse() throws Exception {
    TermVectorsResponse outResponse = new TermVectorsResponse("a", "b", "c");
    outResponse.setExists(true);
    writeStandardTermVector(outResponse);
    // write
    ByteArrayOutputStream outBuffer = new ByteArrayOutputStream();
    OutputStreamStreamOutput out = new OutputStreamStreamOutput(outBuffer);
    outResponse.writeTo(out);
    // read
    ByteArrayInputStream esInBuffer = new ByteArrayInputStream(outBuffer.toByteArray());
    InputStreamStreamInput esBuffer = new InputStreamStreamInput(esInBuffer);
    TermVectorsResponse inResponse = new TermVectorsResponse("a", "b", "c");
    inResponse.readFrom(esBuffer);
    // see if correct
    checkIfStandardTermVector(inResponse);
    outResponse = new TermVectorsResponse("a", "b", "c");
    writeEmptyTermVector(outResponse);
    // write
    outBuffer = new ByteArrayOutputStream();
    out = new OutputStreamStreamOutput(outBuffer);
    outResponse.writeTo(out);
    // read
    esInBuffer = new ByteArrayInputStream(outBuffer.toByteArray());
    esBuffer = new InputStreamStreamInput(esInBuffer);
    inResponse = new TermVectorsResponse("a", "b", "c");
    inResponse.readFrom(esBuffer);
    assertTrue(inResponse.isExists());
}