def test_import_config_diff_number(self):
# Arrange
mailbox = Mailbox(phone_number='+15555555555', carrier='Foo Wireless')
db.session.add(mailbox)
db.session.commit()
# Act
with patch('app.setup.views.Mailbox.import_config_image') as mock:
from werkzeug.exceptions import Forbidden
with self.assertRaises(Forbidden):
_import_config('+17777777777', 'http://i.imgur.com/VMSuO1N.gif')
# Assert
self.assertFalse(mock.called)
def test_import_config_no_mailbox(self):
# Act
with patch('app.setup.views.Mailbox.import_config_image', return_value='Imported!') as mock:
result = _import_config('+15555555555', 'http://i.imgur.com/VMSuO1N.gif')
# Assert
self.assertEqual(result, 'Imported!')
mock.assert_called_once_with('http://i.imgur.com/VMSuO1N.gif')
def test_import_config_same_number(self):
# Arrange
mailbox = Mailbox(phone_number='+15555555555', carrier='Foo Wireless')
db.session.add(mailbox)
db.session.commit()
# Act
with patch('app.setup.views.Mailbox.import_config_image', return_value='Imported!') as mock:
result = _import_config('+15555555555', 'http://i.imgur.com/VMSuO1N.gif')
# Assert
self.assertEqual(result, 'Imported!')
mock.assert_called_once_with('http://i.imgur.com/VMSuO1N.gif')