Index: third_party/gsutil/boto/tests/integration/elastictranscoder/test_layer1.py |
diff --git a/third_party/gsutil/boto/tests/integration/elastictranscoder/test_layer1.py b/third_party/gsutil/boto/tests/integration/elastictranscoder/test_layer1.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..fa2f840c45ef75c0d31b93d80dc14e4f8a2128d2 |
--- /dev/null |
+++ b/third_party/gsutil/boto/tests/integration/elastictranscoder/test_layer1.py |
@@ -0,0 +1,115 @@ |
+# Copyright (c) 2013 Amazon.com, Inc. or its affiliates. All Rights Reserved |
+# |
+# Permission is hereby granted, free of charge, to any person obtaining a |
+# copy of this software and associated documentation files (the |
+# "Software"), to deal in the Software without restriction, including |
+# without limitation the rights to use, copy, modify, merge, publish, dis- |
+# tribute, sublicense, and/or sell copies of the Software, and to permit |
+# persons to whom the Software is furnished to do so, subject to the fol- |
+# lowing conditions: |
+# |
+# The above copyright notice and this permission notice shall be included |
+# in all copies or substantial portions of the Software. |
+# |
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS |
+# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- |
+# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT |
+# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, |
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
+# IN THE SOFTWARE. |
+# |
+import unittest |
+import time |
+ |
+from boto.elastictranscoder.layer1 import ElasticTranscoderConnection |
+from boto.elastictranscoder.exceptions import ValidationException |
+import boto.s3 |
+import boto.sns |
+import boto.iam |
+import boto.sns |
+ |
+ |
+class TestETSLayer1PipelineManagement(unittest.TestCase): |
+ def setUp(self): |
+ self.api = ElasticTranscoderConnection() |
+ self.s3 = boto.connect_s3() |
+ self.sns = boto.connect_sns() |
+ self.iam = boto.connect_iam() |
+ self.sns = boto.connect_sns() |
+ self.timestamp = str(int(time.time())) |
+ self.input_bucket = 'boto-pipeline-%s' % self.timestamp |
+ self.output_bucket = 'boto-pipeline-out-%s' % self.timestamp |
+ self.role_name = 'boto-ets-role-%s' % self.timestamp |
+ self.pipeline_name = 'boto-pipeline-%s' % self.timestamp |
+ self.s3.create_bucket(self.input_bucket) |
+ self.s3.create_bucket(self.output_bucket) |
+ self.addCleanup(self.s3.delete_bucket, self.input_bucket) |
+ self.addCleanup(self.s3.delete_bucket, self.output_bucket) |
+ self.role = self.iam.create_role(self.role_name) |
+ self.role_arn = self.role['create_role_response']['create_role_result']\ |
+ ['role']['arn'] |
+ self.addCleanup(self.iam.delete_role, self.role_name) |
+ |
+ def create_pipeline(self): |
+ pipeline = self.api.create_pipeline( |
+ self.pipeline_name, self.input_bucket, |
+ self.output_bucket, self.role_arn, |
+ {'Progressing': '', 'Completed': '', 'Warning': '', 'Error': ''}) |
+ pipeline_id = pipeline['Pipeline']['Id'] |
+ |
+ self.addCleanup(self.api.delete_pipeline, pipeline_id) |
+ return pipeline_id |
+ |
+ def test_create_delete_pipeline(self): |
+ pipeline = self.api.create_pipeline( |
+ self.pipeline_name, self.input_bucket, |
+ self.output_bucket, self.role_arn, |
+ {'Progressing': '', 'Completed': '', 'Warning': '', 'Error': ''}) |
+ pipeline_id = pipeline['Pipeline']['Id'] |
+ |
+ self.api.delete_pipeline(pipeline_id) |
+ |
+ def test_can_retrieve_pipeline_information(self): |
+ pipeline_id = self.create_pipeline() |
+ |
+ # The pipeline shows up in list_pipelines |
+ pipelines = self.api.list_pipelines()['Pipelines'] |
+ pipeline_names = [p['Name'] for p in pipelines] |
+ self.assertIn(self.pipeline_name, pipeline_names) |
+ |
+ # The pipeline shows up in read_pipeline |
+ response = self.api.read_pipeline(pipeline_id) |
+ self.assertEqual(response['Pipeline']['Id'], pipeline_id) |
+ |
+ def test_update_pipeline(self): |
+ pipeline_id = self.create_pipeline() |
+ self.api.update_pipeline_status(pipeline_id, 'Paused') |
+ |
+ response = self.api.read_pipeline(pipeline_id) |
+ self.assertEqual(response['Pipeline']['Status'], 'Paused') |
+ |
+ def test_update_pipeline_notification(self): |
+ pipeline_id = self.create_pipeline() |
+ response = self.sns.create_topic('pipeline-errors') |
+ topic_arn = response['CreateTopicResponse']['CreateTopicResult']\ |
+ ['TopicArn'] |
+ self.addCleanup(self.sns.delete_topic, topic_arn) |
+ |
+ self.api.update_pipeline_notifications( |
+ pipeline_id, |
+ {'Progressing': '', 'Completed': '', |
+ 'Warning': '', 'Error': topic_arn}) |
+ |
+ response = self.api.read_pipeline(pipeline_id) |
+ self.assertEqual(response['Pipeline']['Notifications']['Error'], |
+ topic_arn) |
+ |
+ def test_list_jobs_by_pipeline(self): |
+ pipeline_id = self.create_pipeline() |
+ response = self.api.list_jobs_by_pipeline(pipeline_id) |
+ self.assertEqual(response['Jobs'], []) |
+ |
+ def test_proper_error_when_pipeline_does_not_exist(self): |
+ with self.assertRaises(ValidationException): |
+ self.api.read_pipeline('badpipelineid') |