Examples with StreamRequest org.apache.cassandra.streaming.StreamRequest used on opensource projects

Example 1 with StreamRequest

use of org.apache.cassandra.streaming.StreamRequest in project cassandra by apache.

the class StreamRequestTest method serializationRoundTrip.

@Test
public void serializationRoundTrip() throws Throwable {
    StreamRequest orig = new StreamRequest(ks, atEndpoint(Arrays.asList(range(1, 2), range(3, 4), range(5, 6)), Collections.emptyList()), atEndpoint(Collections.emptyList(), Arrays.asList(range(5, 6), range(7, 8))), Arrays.asList("a", "b", "c"));
    int expectedSize = (int) StreamRequest.serializer.serializedSize(orig, version);
    try (DataOutputBuffer out = new DataOutputBuffer(expectedSize)) {
        StreamRequest.serializer.serialize(orig, out, version);
        Assert.assertEquals(expectedSize, out.buffer().limit());
        try (DataInputBuffer in = new DataInputBuffer(out.buffer(), false)) {
            StreamRequest decoded = StreamRequest.serializer.deserialize(in, version);
            Assert.assertEquals(orig.keyspace, decoded.keyspace);
            Util.assertRCEquals(orig.full, decoded.full);
            Util.assertRCEquals(orig.transientReplicas, decoded.transientReplicas);
            Assert.assertEquals(orig.columnFamilies, decoded.columnFamilies);
        }
    }
}