/** Constructs a runner from the provided options. */ public static TestDataflowRunner fromOptions(PipelineOptions options) { TestDataflowPipelineOptions dataflowOptions = options.as(TestDataflowPipelineOptions.class); String tempLocation = Joiner.on("/") .join(dataflowOptions.getTempRoot(), dataflowOptions.getJobName(), "output", "results"); dataflowOptions.setTempLocation(tempLocation); return new TestDataflowRunner( dataflowOptions, DataflowClient.create(options.as(DataflowPipelineOptions.class))); }
@Before public void setup() { MockitoAnnotations.initMocks(this); when(mockWorkflowClient.projects()).thenReturn(mockProjects); when(mockProjects.locations()).thenReturn(mockLocations); when(mockLocations.jobs()).thenReturn(mockJobs); options = PipelineOptionsFactory.as(TestDataflowPipelineOptions.class); options.setDataflowClient(mockWorkflowClient); options.setProject(PROJECT_ID); options.setRunner(DataflowRunner.class); options.setTempLocation("gs://fakebucket/temp"); options.setPathValidatorClass(NoopPathValidator.class); options.setGcpCredential(new TestCredential()); }
@Before public void setup() { MockitoAnnotations.initMocks(this); when(mockWorkflowClient.projects()).thenReturn(mockProjects); when(mockProjects.locations()).thenReturn(mockLocations); when(mockLocations.jobs()).thenReturn(mockJobs); options = PipelineOptionsFactory.as(TestDataflowPipelineOptions.class); options.setDataflowClient(mockWorkflowClient); options.setProject(PROJECT_ID); options.setRegion(REGION_ID); options.setRunner(DataflowRunner.class); options.setTempLocation("gs://fakebucket/temp"); options.setPathValidatorClass(NoopPathValidator.class); options.setGcpCredential(new TestCredential()); }
@Before public void setUp() throws Exception { MockitoAnnotations.initMocks(this); options = PipelineOptionsFactory.as(TestDataflowPipelineOptions.class); options.setAppName("TestAppName"); options.setProject("test-project"); options.setTempLocation("gs://test/temp/location"); options.setTempRoot("gs://test"); options.setGcpCredential(new TestCredential()); options.setRunner(TestDataflowRunner.class); options.setPathValidatorClass(NoopPathValidator.class); }