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)