def test_success(self, mfa_settings, twilio_client):
mfa_settings.TWILIO_ACCOUNT_SID = "sid"
mfa_settings.TWILIO_AUTH_TOKEN = "authtoken"
mfa_settings.TWILIO_SMS_POOL_SID = "0987654321"
twilio_client_instance = Mock()
twilio_client.return_value = twilio_client_instance
send_mfa_code_text_message(mfa_instance=self.mfa, mfa_code=self.code)
twilio_client_instance.messages.create.assert_called_once_with(
body="Two Factor Authentication Code: 123456",
to="1234567890",
from_="0987654321")
def test_failed_sms_error(self, mfa_settings, twilio_client):
mfa_settings.TWILIO_ACCOUNT_SID = "sid"
mfa_settings.TWILIO_AUTH_TOKEN = "authtoken"
mfa_settings.TWILIO_SMS_POOL_SID = "0987654321"
twilio_client_instance = Mock()
twilio_client_instance.messages.create.side_effect = (
TwilioRestException(400, "abc"))
twilio_client.return_value = twilio_client_instance
with self.assertRaises(TwilioMessageError):
send_mfa_code_text_message(mfa_instance=self.mfa,
mfa_code=self.code)
def test_no_twilio_credentials(self, twilio_client):
twilio_client_instance = Mock()
twilio_client.return_value = twilio_client_instance
send_mfa_code_text_message(mfa_instance=self.mfa, mfa_code=self.code)
twilio_client_instance.messages.create.assert_not_called()
