Added gsutil to depot tools

third_party/gsutil/boto/tests/integration/elastictranscoder/test_layer1.py

+# 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')