def test_emitter_build_session_from_empty_config(self, Session):
e = notify_sns.SNSNotificationEmitter({'aws_region': 'test'})
e.build_session()
assert e.session is Session.return_value
Session.assert_called_once_with(
region_name='test',
aws_access_key_id=None,
aws_secret_access_key=None,
profile_name=None,
)
def test_send_message_for_Event(self, get_topic):
topic = get_topic.return_value
manager = Mock()
manager.config = {'tasks': {}}
task = Mock(wraps=Task(manager, 'fake'))
task.options.test = False
event = Mock()
task.accepted = [event]
e = notify_sns.SNSNotificationEmitter({'aws_region': 'test', 'sns_topic_arn': 'arn'})
e.send_notifications(task)
topic.publish.assert_called_once_with(Message=event.render.return_value)
def test_emitter_uses_config_credentials(self, Session):
e = notify_sns.SNSNotificationEmitter({
'aws_region': None,
'aws_access_key_id': 'DUMMY',
'aws_secret_access_key': 'DUMMYKEY',
'profile_name': 'profile-name',
})
e.build_session()
Session.assert_called_once_with(
region_name=None,
aws_access_key_id='DUMMY',
aws_secret_access_key='DUMMYKEY',
profile_name='profile-name',
)
def test_dry_run_does_not_send_message(self, get_topic):
topic = get_topic.return_value
manager = Mock()
manager.config = {'tasks': {}}
task = Mock(wraps=Task(manager, 'fake'))
task.options.test = True
event = Mock()
task.accepted = [event]
e = notify_sns.SNSNotificationEmitter({'aws_region': 'test', 'sns_topic_arn': 'arn'})
e.send_notifications(task)
event.render.assert_called_once_with(notify_sns.DEFAULT_TEMPLATE_VALUE)
assert not topic.publish.called