def setUp(self): self.session = FakeSession({'config_file': 'myconfigfile'}) self.subscribe = CloudTrailSubscribe(self.session) self.subscribe.iam = Mock() self.subscribe.iam.GetUser = Mock( return_value={'User': {'Arn': '::::123:456'}}) self.subscribe.s3 = Mock() self.subscribe.s3.endpoint = Mock() self.subscribe.s3.endpoint.region_name = 'us-east-1' policy_template = io.StringIO(initial_value=u'{"Statement": []}') self.subscribe.s3.GetObject = Mock( return_value={'Body': policy_template}) self.subscribe.s3.ListBuckets = Mock( return_value={'Buckets': [{'Name': 'test2'}]}) self.subscribe.sns = Mock() self.subscribe.sns.endpoint = Mock() self.subscribe.sns.endpoint.region_name = 'us-east-1' self.subscribe.sns.ListTopics = Mock( return_value={'Topics': [{'TopicArn': ':test2'}]}) self.subscribe.sns.CreateTopic = Mock( return_value={'TopicArn': 'foo'}) self.subscribe.sns.GetTopicAttributes = Mock( return_value={'Attributes': {'Policy': '{"Statement": []}'}})
def setUp(self): self.session = FakeSession({'config_file': 'myconfigfile'}) self.subscribe = CloudTrailSubscribe(self.session) self.subscribe.region_name = 'us-east-1' self.subscribe.iam = Mock() self.subscribe.iam.get_user = Mock( return_value={'User': { 'Arn': '::::123:456' }}) self.subscribe.s3 = Mock() self.subscribe.s3.meta.region_name = 'us-east-1' policy_template = six.BytesIO(six.b(u'{"Statement": []}')) self.subscribe.s3.get_object = Mock( return_value={'Body': policy_template}) self.subscribe.s3.head_bucket.return_value = {} self.subscribe.sns = Mock() self.subscribe.sns.meta.region_name = 'us-east-1' self.subscribe.sns.list_topics = Mock( return_value={'Topics': [{ 'TopicArn': ':test2' }]}) self.subscribe.sns.create_topic = Mock( return_value={'TopicArn': 'foo'}) self.subscribe.sns.get_topic_attributes = Mock( return_value={'Attributes': { 'Policy': '{"Statement": []}' }})
def test_excludes_global_synopsis_when_command_table_not_empty(self): help_command = self.create_help_command() arg_table = self.create_arg_table() help_command.arg_table = arg_table fake_command = FakeCommand(FakeSession()) fake_command.NAME = 'command' help_command.command_table = {'command': fake_command} rendered = self.generate_global_synopsis_docs(help_command) self.assert_global_args_not_documented(arg_table, rendered)
def setUp(self): self.session = FakeSession() self.emitter = mock.Mock(wraps=HierarchicalEmitter()) self.session.emitter = self.emitter
def setUp(self): self.application_name = 'MyApp' self.description = 'MyApp revision' self.source = '/tmp' self.appspec = 'appspec.yml' self.appspec_path = '{0}/{1}'.format(self.source, self.appspec) self.bucket = 'foo' self.key = 'bar/baz.zip' self.s3_location = 's3://' + self.bucket + '/' + self.key self.eTag = '"1a2b3cd45e"' self.version_id = '12341234-1234-1234-1234-123412341234' self.upload_id = 'upload_id' self.region = 'us-east-1' self.endpoint_url = 'https://codedeploy-%s.amazonaws.com' % self.region self.args = argparse.Namespace() self.args.application_name = self.application_name self.args.s3_location = self.s3_location self.args.ignore_hidden_files = False self.args.no_ignore_hidden_files = False self.args.description = self.description self.args.source = self.source self.globals = argparse.Namespace() self.globals.region = self.region self.globals.endpoint_url = self.endpoint_url self.globals.verify_ssl = None self.endpoint_args = { 'region_name': self.region, 'endpoint_url': self.endpoint_url, 'verify': None } self.upload_response = { 'ETag': self.eTag, 'VersionId': self.version_id } self.revision = { 'revisionType': 'S3', 's3Location': { 'bucket': self.bucket, 'key': self.key, 'bundleType': 'zip', 'eTag': self.eTag, 'version': self.version_id } } self.bundle_mock = MagicMock() self.bundle_mock.tell.return_value = (5 << 20) self.bundle_mock.read.return_value = b'a' * (5 << 20) self.bundle_mock.__enter__.return_value = self.bundle_mock self.bundle_mock.__exit__.return_value = None self.zipfile_mock = MagicMock() self.zipfile_mock.write.return_value = None self.zipfile_mock.close.return_value = None self.zipfile_mock.__enter__.return_value = self.zipfile_mock self.zipfile_mock.__exit__.return_value = None self.session = FakeSession({'config_file': 'myconfigfile'})