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_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 setUp(self):
self.mail = Mandrill(app)
class FlaskMandrillTest(unittest.TestCase):
def setUp(self):
self.mail = Mandrill(app)
def test_get_api_key(self):
self.assertEqual(self.mail.api_key, app.config['MANDRILL_API_KEY'])
@patch('flask_mandrill.requests')
def test_base_email(self, mock_requests):
self.mail.send_email(
subject="subject",
text="text",
from_email='from',
to=[{'email': 'a@pop'}]
)
data = {
'async': False,
'ip_pool': '',
'key': app.config['MANDRILL_API_KEY'],
'message': {
'subject': 'subject',
'text': 'text',
'from_email': 'from',
'to': [{'email': 'a@pop'}]
}
}
mock_requests.post.assert_called_with(
self.mail.messages_endpoint,
data=json.dumps(data),
headers=headers
)
@patch('flask_mandrill.requests')
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_key(self):
mail = Mandrill()
self.assertRaises(ValueError, mail.send_email)
def test_fails_no_sender(self):
mail = Mandrill()
self.assertRaises(ValueError, mail.send_email, key='ABCDEFG')
@patch('flask_mandrill.requests')
def test_default_sender(self, mock_requests):
self.mail.send_email()
data = {
'async': False,
'ip_pool': '',
'key': app.config['MANDRILL_API_KEY'],
'message': {'from_email': 'from'}
}
mock_requests.post.assert_called_with(
self.mail.messages_endpoint,
data=json.dumps(data),
headers=headers
)
@patch('flask_mandrill.requests')
def test_sends_to_template(self, mock_requests):
self.mail.send_email(template_name="a_name")
data = {
'async': False,
'ip_pool': '',
'key': app.config['MANDRILL_API_KEY'],
'template_name': 'a_name',
'template_content': [],
'message': {'from_email': 'from'}
}
mock_requests.post.assert_called_with(
self.mail.templates_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)
class FlaskMandrillTest(unittest.TestCase):
def setUp(self):
self.mail = Mandrill(app)
def test_get_api_key(self):
self.assertEqual(self.mail.api_key, app.config['MANDRILL_API_KEY'])
@patch('flask_mandrill.requests')
def test_base_email(self, mock_requests):
self.mail.send_email(subject="subject",
text="text",
from_email='from',
to=[{
'email': 'a@pop'
}])
data = {
'async': False,
'ip_pool': '',
'key': app.config['MANDRILL_API_KEY'],
'message': {
'subject': 'subject',
'text': 'text',
'from_email': 'from',
'to': [{
'email': 'a@pop'
}]
}
}
mock_requests.post.assert_called_with(self.mail.messages_endpoint,
data=json.dumps(data),
headers=headers)
@patch('flask_mandrill.requests')
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_key(self):
mail = Mandrill()
self.assertRaises(ValueError, mail.send_email)
def test_fails_no_sender(self):
mail = Mandrill()
self.assertRaises(ValueError, mail.send_email, key='ABCDEFG')
@patch('flask_mandrill.requests')
def test_default_sender(self, mock_requests):
self.mail.send_email()
data = {
'async': False,
'ip_pool': '',
'key': app.config['MANDRILL_API_KEY'],
'message': {
'from_email': 'from'
}
}
mock_requests.post.assert_called_with(self.mail.messages_endpoint,
data=json.dumps(data),
headers=headers)
@patch('flask_mandrill.requests')
def test_sends_to_template(self, mock_requests):
self.mail.send_email(template_name="a_name")
data = {
'async': False,
'ip_pool': '',
'key': app.config['MANDRILL_API_KEY'],
'template_name': 'a_name',
'template_content': [],
'message': {
'from_email': 'from'
}
}
mock_requests.post.assert_called_with(self.mail.templates_endpoint,
data=json.dumps(data),
headers=headers)