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

Example 86 with StreamInput

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

the class BasePipelineAggregationTestCase method testSerialization.

/**
     * Test serialization and deserialization of the test AggregatorFactory.
     */
public void testSerialization() throws IOException {
    AF testAgg = createTestAggregatorFactory();
    try (BytesStreamOutput output = new BytesStreamOutput()) {
        output.writeNamedWriteable(testAgg);
        try (StreamInput in = new NamedWriteableAwareStreamInput(output.bytes().streamInput(), namedWriteableRegistry)) {
            PipelineAggregationBuilder deserializedQuery = in.readNamedWriteable(PipelineAggregationBuilder.class);
            assertEquals(deserializedQuery, testAgg);
            assertEquals(deserializedQuery.hashCode(), testAgg.hashCode());
            assertNotSame(deserializedQuery, testAgg);
        }
    }
}

Example 87 with StreamInput

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

the class SearchRequestTests method testSerialization.

public void testSerialization() throws Exception {
    SearchRequest searchRequest = createSearchRequest();
    try (BytesStreamOutput output = new BytesStreamOutput()) {
        searchRequest.writeTo(output);
        try (StreamInput in = new NamedWriteableAwareStreamInput(output.bytes().streamInput(), namedWriteableRegistry)) {
            SearchRequest deserializedRequest = new SearchRequest();
            deserializedRequest.readFrom(in);
            assertEquals(deserializedRequest, searchRequest);
            assertEquals(deserializedRequest.hashCode(), searchRequest.hashCode());
            assertNotSame(deserializedRequest, searchRequest);
        }
    }
}

Example 88 with StreamInput

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

the class ScriptExceptionTests method testRoundTrip.

/** ensure we can round trip in serialization */
public void testRoundTrip() throws IOException {
    ScriptException e = new ScriptException("messageData", new Exception("causeData"), Arrays.asList("stack1", "stack2"), "sourceData", "langData");
    ByteArrayOutputStream bytes = new ByteArrayOutputStream();
    StreamOutput output = new DataOutputStreamOutput(new DataOutputStream(bytes));
    e.writeTo(output);
    output.close();
    StreamInput input = new InputStreamStreamInput(new ByteArrayInputStream(bytes.toByteArray()));
    ScriptException e2 = new ScriptException(input);
    input.close();
    assertEquals(e.getMessage(), e2.getMessage());
    assertEquals(e.getScriptStack(), e2.getScriptStack());
    assertEquals(e.getScript(), e2.getScript());
    assertEquals(e.getLang(), e2.getLang());
}

Example 89 with StreamInput

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

the class SearchSortValuesTests method testSerialization.

public void testSerialization() throws IOException {
    SearchSortValues sortValues = createTestItem();
    try (BytesStreamOutput output = new BytesStreamOutput()) {
        sortValues.writeTo(output);
        try (StreamInput in = output.bytes().streamInput()) {
            SearchSortValues deserializedCopy = new SearchSortValues(in);
            assertEquals(sortValues, deserializedCopy);
            assertEquals(sortValues.hashCode(), deserializedCopy.hashCode());
            assertNotSame(sortValues, deserializedCopy);
        }
    }
}

Example 90 with StreamInput

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

the class ExtendedBoundsTests method testTransportRoundTrip.

public void testTransportRoundTrip() throws IOException {
    ExtendedBounds orig = randomExtendedBounds();
    BytesReference origBytes;
    try (BytesStreamOutput out = new BytesStreamOutput()) {
        orig.writeTo(out);
        origBytes = out.bytes();
    }
    ExtendedBounds read;
    try (StreamInput in = origBytes.streamInput()) {
        read = new ExtendedBounds(in);
        assertEquals("read fully", 0, in.available());
    }
    assertEquals(orig, read);
    BytesReference readBytes;
    try (BytesStreamOutput out = new BytesStreamOutput()) {
        read.writeTo(out);
        readBytes = out.bytes();
    }
    assertEquals(origBytes, readBytes);
}