def test_runner_profile_override_non_dataflow(self):
# This is documenting the current behavior - when we add an EMR
# service_type, this test should change.
actual_mvn_command = []
beam_benchmark_helper.AddRunnerProfileMvnArgument(
dpb_service.EMR, actual_mvn_command, None)
self.assertListEqual([], actual_mvn_command)
def test_runner_profile_override_empty_override(self):
testOptionVal = ""
actual_mvn_command = []
beam_benchmark_helper.AddRunnerProfileMvnArgument(
dpb_service.DATAFLOW, actual_mvn_command, testOptionVal)
self.assertListEqual([], actual_mvn_command)
def test_runner_profile_override_dataflow(self):
actual_mvn_command = []
beam_benchmark_helper.AddRunnerProfileMvnArgument(
dpb_service.DATAFLOW, actual_mvn_command, None)
self.assertListEqual(['-Pdataflow-runner'], actual_mvn_command)