Repository: beam Updated Branches: refs/heads/release-2.2.0 6db1db7b4 -> c6b6668a5
Unit test for label pipeline option Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/6f07e3f9 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/6f07e3f9 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/6f07e3f9 Branch: refs/heads/release-2.2.0 Commit: 6f07e3f9e735768e57ba7b902d8a80b3285e9a93 Parents: 48ae7d1 Author: Ahmet Altay <[email protected]> Authored: Fri Oct 13 15:53:15 2017 -0700 Committer: Ahmet Altay <[email protected]> Committed: Mon Oct 30 14:42:12 2017 -0700 ---------------------------------------------------------------------- .../runners/dataflow/internal/apiclient_test.py | 28 ++++++++++++++++++++ 1 file changed, 28 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/6f07e3f9/sdks/python/apache_beam/runners/dataflow/internal/apiclient_test.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/runners/dataflow/internal/apiclient_test.py b/sdks/python/apache_beam/runners/dataflow/internal/apiclient_test.py index ecd6003..79cbd1c 100644 --- a/sdks/python/apache_beam/runners/dataflow/internal/apiclient_test.py +++ b/sdks/python/apache_beam/runners/dataflow/internal/apiclient_test.py @@ -195,6 +195,34 @@ class UtilTest(unittest.TestCase): for experiment in env.proto.experiments: self.assertNotIn('runner_harness_container_image=', experiment) + def test_labels(self): + pipeline_options = PipelineOptions( + ['--project', 'test_project', '--job_name', 'test_job_name', + '--temp_location', 'gs://test-location/temp']) + job = apiclient.Job(pipeline_options) + self.assertIsNone(job.proto.labels) + + pipeline_options = PipelineOptions( + ['--project', 'test_project', '--job_name', 'test_job_name', + '--temp_location', 'gs://test-location/temp', + '--label', 'key1=value1', + '--label', 'key2', + '--label', 'key3=value3', + '--labels', 'key4=value4', + '--labels', 'key5']) + job = apiclient.Job(pipeline_options) + self.assertEqual(5, len(job.proto.labels.additionalProperties)) + self.assertEqual('key1', job.proto.labels.additionalProperties[0].key) + self.assertEqual('value1', job.proto.labels.additionalProperties[0].value) + self.assertEqual('key2', job.proto.labels.additionalProperties[1].key) + self.assertEqual('', job.proto.labels.additionalProperties[1].value) + self.assertEqual('key3', job.proto.labels.additionalProperties[2].key) + self.assertEqual('value3', job.proto.labels.additionalProperties[2].value) + self.assertEqual('key4', job.proto.labels.additionalProperties[3].key) + self.assertEqual('value4', job.proto.labels.additionalProperties[3].value) + self.assertEqual('key5', job.proto.labels.additionalProperties[4].key) + self.assertEqual('', job.proto.labels.additionalProperties[4].value) + if __name__ == '__main__': unittest.main()
