django-dav-events/dav_auth/tests/test_forms.py

# -*- coding: utf-8 -*-
from unittest import skip
from django.contrib.auth import get_user_model
from django.core import mail as django_mail
from django.utils.translation import ugettext

from dav_base.tests.generic import FormDataSet, FormsTestCase

from ..forms import LoginForm, SetPasswordForm, CreateAndSendPasswordForm

TEST_USERNAME = 'root@localhost'
TEST_PASSWORD = u'me||ön 2'
TEST_EMAIL = TEST_USERNAME
USERNAME_MAX_LENGTH = 254


class LoginFormTestCase(FormsTestCase):
    form_class = LoginForm

    def setUp(self):
        # Need a test user
        self.test_username = TEST_USERNAME
        self.test_password = TEST_PASSWORD
        model = get_user_model()
        self.user = model.objects.create_user(username=TEST_USERNAME, password=TEST_PASSWORD, email=TEST_EMAIL)

    def test_length(self):
        form = self.form_class()
        self.assertEqual(form.fields['username'].max_length, USERNAME_MAX_LENGTH)
        self.assertEqual(form.fields['password'].max_length, None)

    def test_labels(self):
        form = self.form_class()
        self.assertEqual(form.fields['username'].label, ugettext(u'E-Mail-Adresse'))
        self.assertEqual(form.fields['password'].label, ugettext(u'Password'))

    def test_required(self):
        form = self.form_class()
        self.assertTrue(form.fields['username'].required)
        self.assertTrue(form.fields['password'].required)

    def test_username_empty(self):
        data_sets = [
            FormDataSet({'username': '', 'password': self.test_password},
                        expected_errors=[('username', 'required')]),
        ]
        super(LoginFormTestCase, self).test_invalid_data(data_sets=data_sets)

    def test_password_empty(self):
        data_sets = [
            FormDataSet({'username': self.test_username, 'password': ''},
                        expected_errors=[('password', 'required')]),
        ]
        super(LoginFormTestCase, self).test_invalid_data(data_sets=data_sets)

    def test_username_too_long(self):
        data_sets = [
            FormDataSet({'username': 'u' * USERNAME_MAX_LENGTH + 'u', 'password': self.test_password},
                        expected_errors=[('username', 'max_length')]),
        ]
        super(LoginFormTestCase, self).test_invalid_data(data_sets=data_sets)

    def test_invalid_username(self):
        data_sets = [
            FormDataSet({'username': self.test_username[::-1], 'password': self.test_password},
                        expected_errors=[('__all__', 'invalid_login')]),
        ]
        super(LoginFormTestCase, self).test_invalid_data(data_sets=data_sets)

    def test_invalid_password(self):
        data_sets = [
            FormDataSet({'username': self.test_username, 'password': self.test_password[::-1]},
                        expected_errors=[('__all__', 'invalid_login')]),
        ]
        super(LoginFormTestCase, self).test_invalid_data(data_sets=data_sets)

    def test_inactive_user(self):
        self.user.is_active = False
        self.user.save()
        data_sets = [
            FormDataSet({'username': self.test_username, 'password': self.test_password},
                        expected_errors=[('__all__', 'invalid_login')]),
        ]
        super(LoginFormTestCase, self).test_invalid_data(data_sets=data_sets)

    def test_valid_credentials(self):
        data_sets = [
            FormDataSet({'username': self.test_username, 'password': self.test_password}),
        ]
        super(LoginFormTestCase, self).test_valid_data(data_sets=data_sets)


class SetPasswordFormTestCase(FormsTestCase):
    form_class = SetPasswordForm

    def setUp(self):
        # Need a test user
        self.test_username = TEST_USERNAME
        self.test_password = TEST_PASSWORD
        model = get_user_model()
        self.user = model.objects.create_user(username=TEST_USERNAME, password=TEST_PASSWORD, email=TEST_EMAIL)

    def test_required(self):
        form = self.form_class(user=self.user)
        self.assertTrue(form.fields['new_password'].required)
        self.assertTrue(form.fields['new_password_repeat'].required)
        self.assertFalse(form.fields['send_password_mail'].required)

    def test_mismatch(self):
        data_sets = [
            FormDataSet({'new_password': 'mellon12', 'new_password_repeat': 'mellon13'},
                        [('new_password_repeat', 'password_mismatch')]),
        ]
        super(SetPasswordFormTestCase, self).test_invalid_data(data_sets=data_sets, form_kwargs={'user': self.user})

    def test_empty(self):
        data_sets = [
            FormDataSet({'new_password': '', 'new_password_repeat': ''},
                        [('new_password', 'required')]),
        ]
        super(SetPasswordFormTestCase, self).test_invalid_data(data_sets=data_sets, form_kwargs={'user': self.user})

    def test_too_short(self):
        data_sets = [
            FormDataSet({'new_password': 'mellon', 'new_password_repeat': 'mellon'},
                        [('new_password', 'password_too_short')]),
        ]
        super(SetPasswordFormTestCase, self).test_invalid_data(data_sets=data_sets, form_kwargs={'user': self.user})

    def test_entirely_numeric(self):
        data_sets = [
            FormDataSet({'new_password': '1357924680', 'new_password_repeat': '1357924680'},
                        [('new_password', 'password_entirely_numeric')]),
        ]
        super(SetPasswordFormTestCase, self).test_invalid_data(data_sets=data_sets, form_kwargs={'user': self.user})

    def test_too_similar(self):
        data_sets = [
            FormDataSet({'new_password': self.test_username, 'new_password_repeat': self.test_username},
                        [('new_password', 'password_too_similar')]),
        ]
        super(SetPasswordFormTestCase, self).test_invalid_data(data_sets=data_sets, form_kwargs={'user': self.user})

    def test_too_common(self):
        data_sets = [
            FormDataSet({'new_password': 'password', 'new_password_repeat': 'password'},
                        [('new_password', 'password_too_common')]),
        ]
        super(SetPasswordFormTestCase, self).test_invalid_data(data_sets=data_sets, form_kwargs={'user': self.user})

    def test_valid(self):
        data_sets = [
            FormDataSet({'new_password': 'mellon12', 'new_password_repeat': 'mellon12'}),
            FormDataSet({'new_password': 'mellon12', 'new_password_repeat': 'mellon12', 'send_password_mail': True}),
            FormDataSet({'new_password': u'"ä§ Mellon12', 'new_password_repeat': u'"ä§ Mellon12'}),
            FormDataSet({'new_password': 'mellon12' * 128, 'new_password_repeat': 'mellon12' * 128}),
        ]
        super(SetPasswordFormTestCase, self).test_valid_data(data_sets=data_sets, form_kwargs={'user': self.user})

    def test_save(self):
        new_passwords = [
            u'"ä§ Mellon12'
            'mellon12' * 128,
        ]

        for new_password in new_passwords:
            data = {'new_password': new_password, 'new_password_repeat': new_password}
            form = SetPasswordForm(data=data, user=self.user)
            self.assertTrue(form.is_valid())
            form.save()
            self.assertEqual(len(django_mail.outbox), 0)
            self.assertTrue(self.client.login(username=self.test_username, password=new_password))

    @skip('Function is implemented in SetPasswordView instead of SetPasswordForm')
    def test_save_with_mail(self):  # pragma: no cover
        new_passwords = [
            u'"ä§ Mellon12'
            'mellon12' * 128,
        ]

        for new_password in new_passwords:
            data = {'new_password': new_password, 'new_password_repeat': new_password, 'send_password_mail': True}
            form = SetPasswordForm(data=data, user=self.user)
            self.assertTrue(form.is_valid())
            form.save()
            self.assertEqual(len(django_mail.outbox), 1)
            mail = django_mail.outbox[0]
            recipient = '%s <%s>' % (self.user.get_full_name(), self.user.email)
            recipients = mail.recipients()
            self.assertIn(recipient, recipients)
            self.assertEqual(len(recipients), 1)
            self.assertIn(new_password, mail.body)
            self.assertTrue(self.client.login(username=self.test_username, password=new_password))


class CreateAndSendPasswordFormTestCase(FormsTestCase):
    form_class = CreateAndSendPasswordForm

    valid_data_sets = (
        FormDataSet({'username': 'unittest@example.com'}),
    )
    invalid_data_sets = (
        FormDataSet({'username': ''}, expected_errors=[('username', 'required')]),
        FormDataSet({'username': 'u' * USERNAME_MAX_LENGTH + 'u'}, expected_errors=[('username', 'max_length')]),
    )

    def test_length(self):
        form = self.form_class()
        self.assertEqual(form.fields['username'].max_length, USERNAME_MAX_LENGTH)

    def test_labels(self):
        form = self.form_class()
        self.assertEqual(form.fields['username'].label, ugettext(u'E-Mail-Adresse'))

    def test_required(self):
        form = self.form_class()
        self.assertTrue(form.fields['username'].required)