<filename>tests/test_backends.py
# -*- coding: utf8 -*-
from unittest import skipIf

import django
from django.contrib.auth import authenticate
from django.test import SimpleTestCase
from django.test.utils import mail, override_settings
from django.test import TestCase
from mock import MagicMock, patch

from nopassword.backends.base import NoPasswordBackend
from nopassword.backends.email import EmailBackend
from nopassword.backends.sms import TwilioBackend
from nopassword.models import LoginCode
from nopassword.utils import get_user_model


class AuthenticationBackendTests(TestCase):
    @override_settings(AUTH_USER_MODULE='tests.NoUsernameUser')
    def test_authenticate_with_custom_user_model(self):
        """When a custom user model is used that doesn't have a field
        called "username" return `None`
        """
        result = authenticate(username='username')
        self.assertIsNone(result)


@skipIf(django.VERSION < (1, 5), 'Custom user not supported')
@override_settings(AUTH_USER_MODEL='tests.PhoneNumberUser', NOPASSWORD_TWILIO_SID="aaaaaaaa",
                   NOPASSWORD_TWILIO_AUTH_TOKEN="<PASSWORD>", DEFAULT_FROM_NUMBER="+15555555")
class TwilioBackendTests(SimpleTestCase):
    def setUp(self):
        self.user = get_user_model().objects.create(username='twilio_user')
        self.code = LoginCode.create_code_for_user(self.user, next='/secrets/')
        self.assertEqual(len(self.code.code), 20)
        self.assertIsNotNone(authenticate(username=self.user.username, code=self.code.code))
        self.assertEqual(LoginCode.objects.filter(user=self.user, code=self.code.code).count(), 0)

    def tearDown(self):
        self.user.delete()

    @patch('nopassword.backends.sms.TwilioRestClient')
    def test_twilio_backend(self, mock_object):
        self.backend = TwilioBackend()
        self.backend.twilio_client.messages.create = MagicMock()
        self.backend.send_login_code(self.code)
        self.assertTrue(mock_object.called)
        self.assertTrue(self.backend.twilio_client.messages.create.called)
        _, kwargs = self.backend.twilio_client.messages.create.call_args
        self.assertIn(self.code.login_url(secure=False), kwargs.get('body'))

        authenticate(username=self.user.username)
        self.assertEqual(LoginCode.objects.filter(user=self.user).count(), 1)

    @patch('nopassword.backends.sms.TwilioRestClient')
    def test_twilio_backend_with_https(self, mock_object):
        self.backend = TwilioBackend()
        self.backend.twilio_client.messages.create = MagicMock()
        self.backend.send_login_code(self.code, secure=True, host='secure.example.com')
        _, kwargs = self.backend.twilio_client.messages.create.call_args
        login_url = self.code.login_url(secure=True, host='secure.example.com')
        self.assertIn(login_url, kwargs.get('body'))


@skipIf(django.VERSION < (1, 5), 'Custom user not supported')
class EmailBackendTests(SimpleTestCase):
    def setUp(self):
        self.user = get_user_model().objects.create(
            username='email_user',
            email='<EMAIL>',
        )
        self.code = LoginCode.create_code_for_user(self.user, next='/secrets/')
        self.backend = EmailBackend()

    def tearDown(self):
        self.user.delete()

    def test_email_backend(self):
        "Send email via EmailBackend with default options"
        mail.outbox = []
        self.backend.send_login_code(self.code)
        self.assertEqual(1, len(mail.outbox))
        message = mail.outbox[0]
        http_url = self.code.login_url()
        self.assertIn(http_url, message.body)
        self.assertTrue(http_url.startswith('http:'))
        self.assertEqual([self.user.email], message.to)

    def test_email_backend_with_https(self):
        "Send email via EmailBackend with secure=True"
        mail.outbox = []
        self.backend.send_login_code(self.code, secure=True, host='secure.example.com')
        self.assertEqual(1, len(mail.outbox))
        message = mail.outbox[0]
        https_url = self.code.login_url(secure=True, host='secure.example.com')
        self.assertTrue(https_url.startswith('https:'))
        self.assertIn(https_url, message.body)


class TestBackendUtils(TestCase):
    def setUp(self):
        self.user = get_user_model().objects.create(username='test_user')
        self.inactive_user = get_user_model().objects.create(username='inactive', is_active=False)
        self.backend = NoPasswordBackend()

    def tearDown(self):
        self.user.delete()
        self.inactive_user.delete()

    def test_verify_user(self):
        self.assertTrue(self.backend.verify_user(self.user))
        self.assertFalse(self.backend.verify_user(self.inactive_user))

    def test_send_login_code(self):
        self.assertRaises(NotImplementedError,
                          self.backend.send_login_code,
                          code=None,
                          secure=False)

<filename>tests/test_backends.py
# -*- coding: utf8 -*-
from unittest import skipIf

import django
from django.contrib.auth import authenticate
from django.test import SimpleTestCase
from django.test.utils import mail, override_settings
from django.test import TestCase
from mock import MagicMock, patch

from nopassword.backends.base import NoPasswordBackend
from nopassword.backends.email import EmailBackend
from nopassword.backends.sms import TwilioBackend
from nopassword.models import LoginCode
from nopassword.utils import get_user_model


class AuthenticationBackendTests(TestCase):
    @override_settings(AUTH_USER_MODULE='tests.NoUsernameUser')
    def test_authenticate_with_custom_user_model(self):
        """When a custom user model is used that doesn't have a field
        called "username" return `None`
        """
        result = authenticate(username='username')
        self.assertIsNone(result)


@skipIf(django.VERSION < (1, 5), 'Custom user not supported')
@override_settings(AUTH_USER_MODEL='tests.PhoneNumberUser', NOPASSWORD_TWILIO_SID="aaaaaaaa",
                   NOPASSWORD_TWILIO_AUTH_TOKEN="<PASSWORD>", DEFAULT_FROM_NUMBER="+15555555")
class TwilioBackendTests(SimpleTestCase):
    def setUp(self):
        self.user = get_user_model().objects.create(username='twilio_user')
        self.code = LoginCode.create_code_for_user(self.user, next='/secrets/')
        self.assertEqual(len(self.code.code), 20)
        self.assertIsNotNone(authenticate(username=self.user.username, code=self.code.code))
        self.assertEqual(LoginCode.objects.filter(user=self.user, code=self.code.code).count(), 0)

    def tearDown(self):
        self.user.delete()

    @patch('nopassword.backends.sms.TwilioRestClient')
    def test_twilio_backend(self, mock_object):
        self.backend = TwilioBackend()
        self.backend.twilio_client.messages.create = MagicMock()
        self.backend.send_login_code(self.code)
        self.assertTrue(mock_object.called)
        self.assertTrue(self.backend.twilio_client.messages.create.called)
        _, kwargs = self.backend.twilio_client.messages.create.call_args
        self.assertIn(self.code.login_url(secure=False), kwargs.get('body'))

        authenticate(username=self.user.username)
        self.assertEqual(LoginCode.objects.filter(user=self.user).count(), 1)

    @patch('nopassword.backends.sms.TwilioRestClient')
    def test_twilio_backend_with_https(self, mock_object):
        self.backend = TwilioBackend()
        self.backend.twilio_client.messages.create = MagicMock()
        self.backend.send_login_code(self.code, secure=True, host='secure.example.com')
        _, kwargs = self.backend.twilio_client.messages.create.call_args
        login_url = self.code.login_url(secure=True, host='secure.example.com')
        self.assertIn(login_url, kwargs.get('body'))


@skipIf(django.VERSION < (1, 5), 'Custom user not supported')
class EmailBackendTests(SimpleTestCase):
    def setUp(self):
        self.user = get_user_model().objects.create(
            username='email_user',
            email='<EMAIL>',
        )
        self.code = LoginCode.create_code_for_user(self.user, next='/secrets/')
        self.backend = EmailBackend()

    def tearDown(self):
        self.user.delete()

    def test_email_backend(self):
        "Send email via EmailBackend with default options"
        mail.outbox = []
        self.backend.send_login_code(self.code)
        self.assertEqual(1, len(mail.outbox))
        message = mail.outbox[0]
        http_url = self.code.login_url()
        self.assertIn(http_url, message.body)
        self.assertTrue(http_url.startswith('http:'))
        self.assertEqual([self.user.email], message.to)

    def test_email_backend_with_https(self):
        "Send email via EmailBackend with secure=True"
        mail.outbox = []
        self.backend.send_login_code(self.code, secure=True, host='secure.example.com')
        self.assertEqual(1, len(mail.outbox))
        message = mail.outbox[0]
        https_url = self.code.login_url(secure=True, host='secure.example.com')
        self.assertTrue(https_url.startswith('https:'))
        self.assertIn(https_url, message.body)


class TestBackendUtils(TestCase):
    def setUp(self):
        self.user = get_user_model().objects.create(username='test_user')
        self.inactive_user = get_user_model().objects.create(username='inactive', is_active=False)
        self.backend = NoPasswordBackend()

    def tearDown(self):
        self.user.delete()
        self.inactive_user.delete()

    def test_verify_user(self):
        self.assertTrue(self.backend.verify_user(self.user))
        self.assertFalse(self.backend.verify_user(self.inactive_user))

    def test_send_login_code(self):
        self.assertRaises(NotImplementedError,
                          self.backend.send_login_code,
                          code=None,
                          secure=False)

# -*- coding: utf8 -*- When a custom user model is used that doesn't have a field         called "username" return `None`