def register_extensions(flask_app): """ Register Flask extensions. """ Babel(flask_app) # Babel is currently only used with Flask-Admin. CSRFProtect(flask_app) # Enable CSRF protection globally. mandrill = Mandrill() mandrill.init_app(flask_app) flask_app.extensions[ 'mandrill'] = mandrill # Provide easy access to the mandrill extension in Blueprints. login_manager.init_app(flask_app) init_social(flask_app, db_session)
def test_specify_key(self, mock_requests): mail = Mandrill() mail.send_email(key="ABCDEFG", from_email='from') data = { 'async': False, 'ip_pool': '', 'key': "ABCDEFG", 'message': { 'from_email': 'from' } } mock_requests.post.assert_called_with(self.mail.messages_endpoint, data=json.dumps(data), headers=headers)
def test_fails_no_sender(self): mail = Mandrill() self.assertRaises(ValueError, mail.send_email, key='ABCDEFG')
def test_fails_no_key(self): mail = Mandrill() self.assertRaises(ValueError, mail.send_email)
def setUp(self): self.mail = Mandrill(app)