Search in sources :

Example 6 with DataflowPipelineWorkerPoolOptions

use of org.apache.beam.runners.dataflow.options.DataflowPipelineWorkerPoolOptions in project beam by apache.

the class DataflowRunnerTest method testZoneAndWorkerRegionMutuallyExclusive.

@Test
public void testZoneAndWorkerRegionMutuallyExclusive() {
    DataflowPipelineWorkerPoolOptions options = PipelineOptionsFactory.as(DataflowPipelineWorkerPoolOptions.class);
    options.setZone("us-east1-b");
    options.setWorkerRegion("us-east1");
    assertThrows(IllegalArgumentException.class, () -> DataflowRunner.validateWorkerSettings(options));
}
Also used : DataflowPipelineWorkerPoolOptions(org.apache.beam.runners.dataflow.options.DataflowPipelineWorkerPoolOptions) Test(org.junit.Test) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest)

Example 7 with DataflowPipelineWorkerPoolOptions

use of org.apache.beam.runners.dataflow.options.DataflowPipelineWorkerPoolOptions in project beam by apache.

the class DataflowRunnerTest method testZoneAndWorkerZoneMutuallyExclusive.

@Test
public void testZoneAndWorkerZoneMutuallyExclusive() {
    DataflowPipelineWorkerPoolOptions options = PipelineOptionsFactory.as(DataflowPipelineWorkerPoolOptions.class);
    options.setZone("us-east1-b");
    options.setWorkerZone("us-east1-c");
    assertThrows(IllegalArgumentException.class, () -> DataflowRunner.validateWorkerSettings(options));
}
Also used : DataflowPipelineWorkerPoolOptions(org.apache.beam.runners.dataflow.options.DataflowPipelineWorkerPoolOptions) Test(org.junit.Test) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest)

Example 8 with DataflowPipelineWorkerPoolOptions

use of org.apache.beam.runners.dataflow.options.DataflowPipelineWorkerPoolOptions in project beam by apache.

the class DataflowRunnerTest method testZoneAliasWorkerZone.

@Test
public void testZoneAliasWorkerZone() {
    DataflowPipelineWorkerPoolOptions options = PipelineOptionsFactory.as(DataflowPipelineWorkerPoolOptions.class);
    options.setZone("us-east1-b");
    DataflowRunner.validateWorkerSettings(options);
    assertNull(options.getZone());
    assertEquals("us-east1-b", options.getWorkerZone());
}
Also used : DataflowPipelineWorkerPoolOptions(org.apache.beam.runners.dataflow.options.DataflowPipelineWorkerPoolOptions) Test(org.junit.Test) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest)

Aggregations

DataflowPipelineWorkerPoolOptions (org.apache.beam.runners.dataflow.options.DataflowPipelineWorkerPoolOptions)8 Test (org.junit.Test)8 PrepareForTest (org.powermock.core.classloader.annotations.PrepareForTest)8 DataflowPipelineOptions (org.apache.beam.runners.dataflow.options.DataflowPipelineOptions)2 Matchers.containsString (org.hamcrest.Matchers.containsString)2