use of com.google.api.services.dataflow.model.SourceSplitShard in project beam by apache.
the class BatchDataflowWorkerTest method testIsSplitResponseTooLarge.
@Test
public void testIsSplitResponseTooLarge() throws IOException {
SourceSplitResponse splitResponse = new SourceSplitResponse();
splitResponse.setShards(ImmutableList.<SourceSplitShard>of(new SourceSplitShard(), new SourceSplitShard()));
assertThat(DataflowApiUtils.computeSerializedSizeBytes(splitResponse), greaterThan(0L));
}
Aggregations