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()