forked from mirrors/bookwyrm
136 lines
5.1 KiB
Python
136 lines
5.1 KiB
Python
''' test for app action functionality '''
|
|
from unittest.mock import patch
|
|
|
|
from django.contrib.auth.models import AnonymousUser
|
|
from django.template.response import TemplateResponse
|
|
from django.test import TestCase
|
|
from django.test.client import RequestFactory
|
|
|
|
from bookwyrm import models, views
|
|
|
|
|
|
class PasswordViews(TestCase):
|
|
''' view user and edit profile '''
|
|
def setUp(self):
|
|
''' we need basic test data and mocks '''
|
|
self.factory = RequestFactory()
|
|
self.local_user = models.User.objects.create_user(
|
|
'mouse@local.com', 'mouse@mouse.com', 'password',
|
|
local=True, localname='mouse')
|
|
self.anonymous_user = AnonymousUser
|
|
self.anonymous_user.is_authenticated = False
|
|
|
|
|
|
def test_password_reset_request(self):
|
|
''' there are so many views, this just makes sure it LOADS '''
|
|
view = views.PasswordResetRequest.as_view()
|
|
request = self.factory.get('')
|
|
request.user = self.local_user
|
|
|
|
result = view(request)
|
|
self.assertIsInstance(result, TemplateResponse)
|
|
self.assertEqual(result.template_name, 'password_reset_request.html')
|
|
self.assertEqual(result.status_code, 200)
|
|
|
|
|
|
def test_password_reset_request_post(self):
|
|
''' send 'em an email '''
|
|
request = self.factory.post('', {'email': 'aa@bb.ccc'})
|
|
view = views.PasswordResetRequest.as_view()
|
|
resp = view(request)
|
|
self.assertEqual(resp.status_code, 302)
|
|
|
|
request = self.factory.post('', {'email': 'mouse@mouse.com'})
|
|
with patch('bookwyrm.emailing.send_email.delay'):
|
|
resp = view(request)
|
|
self.assertEqual(resp.template_name, 'password_reset_request.html')
|
|
|
|
self.assertEqual(
|
|
models.PasswordReset.objects.get().user, self.local_user)
|
|
|
|
def test_password_reset(self):
|
|
''' there are so many views, this just makes sure it LOADS '''
|
|
view = views.PasswordReset.as_view()
|
|
code = models.PasswordReset.objects.create(user=self.local_user)
|
|
request = self.factory.get('')
|
|
request.user = self.anonymous_user
|
|
result = view(request, code.code)
|
|
self.assertIsInstance(result, TemplateResponse)
|
|
self.assertEqual(result.template_name, 'password_reset.html')
|
|
self.assertEqual(result.status_code, 200)
|
|
|
|
|
|
def test_password_reset_post(self):
|
|
''' reset from code '''
|
|
view = views.PasswordReset.as_view()
|
|
code = models.PasswordReset.objects.create(user=self.local_user)
|
|
request = self.factory.post('', {
|
|
'password': 'hi',
|
|
'confirm-password': 'hi'
|
|
})
|
|
with patch('bookwyrm.views.password.login'):
|
|
resp = view(request, code.code)
|
|
self.assertEqual(resp.status_code, 302)
|
|
self.assertFalse(models.PasswordReset.objects.exists())
|
|
|
|
def test_password_reset_wrong_code(self):
|
|
''' reset from code '''
|
|
view = views.PasswordReset.as_view()
|
|
models.PasswordReset.objects.create(user=self.local_user)
|
|
request = self.factory.post('', {
|
|
'password': 'hi',
|
|
'confirm-password': 'hi'
|
|
})
|
|
resp = view(request, 'jhgdkfjgdf')
|
|
self.assertEqual(resp.template_name, 'password_reset.html')
|
|
self.assertTrue(models.PasswordReset.objects.exists())
|
|
|
|
def test_password_reset_mismatch(self):
|
|
''' reset from code '''
|
|
view = views.PasswordReset.as_view()
|
|
code = models.PasswordReset.objects.create(user=self.local_user)
|
|
request = self.factory.post('', {
|
|
'password': 'hi',
|
|
'confirm-password': 'hihi'
|
|
})
|
|
resp = view(request, code.code)
|
|
self.assertEqual(resp.template_name, 'password_reset.html')
|
|
self.assertTrue(models.PasswordReset.objects.exists())
|
|
|
|
|
|
def test_password_change_get(self):
|
|
''' there are so many views, this just makes sure it LOADS '''
|
|
view = views.ChangePassword.as_view()
|
|
request = self.factory.get('')
|
|
request.user = self.local_user
|
|
|
|
result = view(request)
|
|
self.assertIsInstance(result, TemplateResponse)
|
|
self.assertEqual(result.template_name, 'change_password.html')
|
|
self.assertEqual(result.status_code, 200)
|
|
|
|
|
|
def test_password_change(self):
|
|
''' change password '''
|
|
view = views.ChangePassword.as_view()
|
|
password_hash = self.local_user.password
|
|
request = self.factory.post('', {
|
|
'password': 'hi',
|
|
'confirm-password': 'hi'
|
|
})
|
|
request.user = self.local_user
|
|
with patch('bookwyrm.views.password.login'):
|
|
view(request)
|
|
self.assertNotEqual(self.local_user.password, password_hash)
|
|
|
|
def test_password_change_mismatch(self):
|
|
''' change password '''
|
|
view = views.ChangePassword.as_view()
|
|
password_hash = self.local_user.password
|
|
request = self.factory.post('', {
|
|
'password': 'hi',
|
|
'confirm-password': 'hihi'
|
|
})
|
|
request.user = self.local_user
|
|
view(request)
|
|
self.assertEqual(self.local_user.password, password_hash)
|