Search in sources :

Example 1 with SourceSplitShard

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));
}
Also used : SourceSplitResponse(com.google.api.services.dataflow.model.SourceSplitResponse) SourceSplitShard(com.google.api.services.dataflow.model.SourceSplitShard) Test(org.junit.Test)

Aggregations

SourceSplitResponse (com.google.api.services.dataflow.model.SourceSplitResponse)1 SourceSplitShard (com.google.api.services.dataflow.model.SourceSplitShard)1 Test (org.junit.Test)1