Misago/misago/users/tests/test_usercp_views.py

from django.contrib.auth import get_user_model
from django.core import mail
from django.core.urlresolvers import reverse

from misago.admin.testutils import AdminTestCase

from misago.core import threadstore


class ChangeForumOptionsTests(AdminTestCase):
    def setUp(self):
        super(ChangeForumOptionsTests, self).setUp()
        self.view_link = reverse('misago:usercp_change_forum_options')

    def test_change_forum_options_get(self):
        """GET to usercp change options view returns 200"""
        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Change forum options', response.content)

    def test_change_forum_options_post(self):
        """POST to usercp change options view returns 302"""
        response = self.client.post(self.view_link, data={
                'timezone': 'Asia/Qatar',
                'is_hiding_presence': '1',
                'subscribe_to_started_threads': '0',
                'subscribe_to_replied_threads': '1',
            })

        self.assertEqual(response.status_code, 302)

        test_user = get_user_model().objects.get(pk=self.test_admin.pk)
        self.assertEqual(test_user.timezone, 'Asia/Qatar')
        self.assertEqual(test_user.is_hiding_presence, 1)
        self.assertEqual(test_user.subscribe_to_started_threads, 0)
        self.assertEqual(test_user.subscribe_to_replied_threads, 1)


class ChangeAvatarTests(AdminTestCase):
    def setUp(self):
        super(ChangeAvatarTests, self).setUp()
        self.view_link = reverse('misago:usercp_change_avatar')

    def test_avatar_get(self):
        """GET to change avatar returns 200"""
        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)

        self.test_admin.is_avatar_banned = True
        self.test_admin.avatar_ban_user_message = 'Your avatar is banned.'
        self.test_admin.save()

        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Your avatar is banned', response.content)

    def test_set_gravatar(self):
        """view sets user gravatar"""
        self.test_admin.set_email('kontakt@rpiton.com')
        self.test_admin.save()

        response = self.client.post(self.view_link, data={'dl-gravatar': '1'})
        self.assertEqual(response.status_code, 302)

        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Gravatar was downloaded', response.content)

        self.test_admin.set_email('test@test.com')
        self.test_admin.save()

        self.client.post(self.view_link, data={'dl-gravatar': '1'})
        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('No Gravatar is associated', response.content)

    def test_set_dynamic(self):
        """view sets user dynamic avatar"""
        response = self.client.post(self.view_link, data={'set-dynamic': '1'})
        self.assertEqual(response.status_code, 302)

        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('New avatar based', response.content)

class EditSignatureTests(AdminTestCase):
    def setUp(self):
        super(EditSignatureTests, self).setUp()
        self.view_link = reverse('misago:usercp_edit_signature')

    def test_signature_no_permission(self):
        """edit signature view with no ACL returns 404"""
        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 404)

    def test_signature_banned(self):
        """GET to usercp change options view returns 200"""
        role = self.test_admin.roles.all()[0]
        permissions = role.permissions
        account_permissions = permissions['misago.users.permissions.account']
        account_permissions['can_have_signature'] = 1
        role.permissions = permissions

        self.test_admin.is_signature_banned = True
        self.test_admin.signature_ban_user_message = 'Your siggy is banned.'
        self.test_admin.save()

        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Your siggy is banned', response.content)

    def test_signature_change(self):
        """GET to usercp change options view returns 200"""
        role = self.test_admin.roles.all()[0]
        permissions = role.permissions
        account_permissions = permissions['misago.users.permissions.account']
        account_permissions['can_have_signature'] = 1
        role.permissions = permissions

        self.test_admin.is_signature_banned = False
        self.test_admin.save()

        response = self.client.post(self.view_link,
            data={'signature': 'Hello siggy!'})
        self.assertEqual(response.status_code, 302)

        response = self.client.get(self.view_link)
        self.assertIn('<p>Hello siggy!</p>', response.content)


class ChangeUsernameTests(AdminTestCase):
    def setUp(self):
        super(ChangeUsernameTests, self).setUp()
        self.view_link = reverse('misago:usercp_change_username')

    def test_change_username_get(self):
        """GET to usercp change username view returns 200"""
        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Change username', response.content)

    def test_change_username_post(self):
        """POST to usercp change username view returns 302"""
        response = self.client.post(self.view_link,
                                    data={'new_username': 'Boberson'})
        self.assertEqual(response.status_code, 302)

        test_user = get_user_model().objects.get(pk=self.test_admin.pk)
        self.assertEqual(test_user.username, 'Boberson')

        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn(test_user.username, response.content)


class ChangeEmailPasswordTests(AdminTestCase):
    def setUp(self):
        super(ChangeEmailPasswordTests, self).setUp()
        self.view_link = reverse('misago:usercp_change_email_password')

        threadstore.clear()

    def _link_from_mail(self, mail_body):
        for line in mail.outbox[0].body.splitlines():
            if line.strip().startswith('http://testserver/'):
                return line.strip()[len('http://testserver'):]
        raise ValueError("mail body didn't contain link with token")

    def test_change_email_password_get(self):
        """GET to usercp change email/pass view returns 200"""
        response = self.client.get(self.view_link)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Change email or password', response.content)

    def test_change_email(self):
        """POST to usercp change email view returns 302"""
        response = self.client.post(self.view_link,
                                    data={'new_email': 'newmail@test.com',
                                          'current_password': 'Pass.123'})
        self.assertEqual(response.status_code, 302)

        self.assertIn('Confirm changes to', mail.outbox[0].subject)
        confirmation_link = self._link_from_mail(mail.outbox[0].body)

        response = self.client.get(confirmation_link)
        self.assertEqual(response.status_code, 302)

        User = get_user_model()
        test_user = User.objects.get(email='newmail@test.com')

    def test_change_password(self):
        """POST to usercp change password view returns 302"""
        response = self.client.post(self.view_link,
                                    data={'new_password': 'newpass123',
                                          'current_password': 'Pass.123'})
        self.assertEqual(response.status_code, 302)

        self.assertIn('Confirm changes to', mail.outbox[0].subject)
        confirmation_link = self._link_from_mail(mail.outbox[0].body)

        response = self.client.get(confirmation_link)
        self.assertEqual(response.status_code, 302)

        User = get_user_model()
        test_user = User.objects.get(pk=self.test_admin.pk)
        self.assertFalse(test_user.check_password('Pass.123'))
        self.assertTrue(test_user.check_password('newpass123'))