def test_runner_option_override_non_dataflow(self): # This is documenting the current behavior - when we add an EMR # service_type, this test should change. actual_options = [] beam_benchmark_helper.AddRunnerOptionMvnArgument( dpb_service.EMR, actual_options, None) self.assertListEqual([], actual_options)
def test_runner_option_override_empty_override(self): testOptionVal = "" actual_options = [] beam_benchmark_helper.AddRunnerOptionMvnArgument( dpb_service.DATAFLOW, actual_options, testOptionVal) self.assertListEqual([], actual_options)
def test_runner_option_override_dataflow(self): actual_options = [] beam_benchmark_helper.AddRunnerOptionMvnArgument( dpb_service.DATAFLOW, actual_options, None) self.assertListEqual(['"--runner=TestDataflowRunner"'], actual_options)