def __init__(self, config, session, logger):
self.config = config
self.logger = logger
self.session = session
api_key = decrypt(self.config, self.logger, self.session,
'sendgrid_api_key')
self.sendgrid_client = sendgrid.SendGridAPIClient(api_key)
def __init__(self, config, session, logger):
smtp_server = config['smtp_server']
smtp_port = int(config.get('smtp_port', 25))
smtp_ssl = bool(config.get('smtp_ssl', True))
smtp_username = config.get('smtp_username')
smtp_password = utils.decrypt(config, logger, session, 'smtp_password')
smtp_connection = smtplib.SMTP(smtp_server, smtp_port)
if smtp_ssl:
smtp_connection.starttls()
smtp_connection.ehlo()
if smtp_username or smtp_password:
smtp_connection.login(smtp_username, smtp_password)
self._smtp_connection = smtp_connection
def test_decrypt_none(self):
self.assertEqual(utils.decrypt({'queue_url': 'aws'}, Mock(), Mock(), 'test'), None)
self.assertEqual(utils.decrypt({'queue_url': 'asq://'}, Mock(), Mock(), 'test'), None)
def test_azure_decrypt(self, azure_decrypt_mock):
utils.decrypt({'queue_url': 'asq://', 'test': 'test'}, Mock(), Mock(), 'test')
azure_decrypt_mock.assert_called_once()
def test_kms_decrypt(self, kms_decrypt_mock):
utils.decrypt({'queue_url': 'aws', 'test': 'test'}, Mock(), Mock(), 'test')
kms_decrypt_mock.assert_called_once()
