moviewyrm/bookwyrm/tests/views/test_authentication.py

194 lines
7.1 KiB
Python
Raw Normal View History

2021-03-08 16:49:10 +00:00
""" test for app action functionality """
2021-01-12 17:09:05 +00:00
from unittest.mock import patch
from django.contrib.auth.models import AnonymousUser
from django.core.exceptions import PermissionDenied
from django.http.response import Http404
from django.template.response import TemplateResponse
from django.test import TestCase
2021-01-12 17:53:04 +00:00
from django.test.client import RequestFactory
2021-01-12 17:09:05 +00:00
from bookwyrm import models, views
from bookwyrm.settings import DOMAIN
# pylint: disable=too-many-public-methods
class AuthenticationViews(TestCase):
2021-04-26 16:15:42 +00:00
"""login and password management"""
2021-03-08 16:49:10 +00:00
2021-01-12 17:53:04 +00:00
def setUp(self):
2021-04-26 16:15:42 +00:00
"""we need basic test data and mocks"""
2021-01-12 17:53:04 +00:00
self.factory = RequestFactory()
2021-05-26 21:57:29 +00:00
with patch("bookwyrm.preview_images.generate_user_preview_image_task.delay"):
self.local_user = models.User.objects.create_user(
"mouse@local.com",
"mouse@mouse.com",
"password",
local=True,
localname="mouse",
)
2021-01-12 17:53:04 +00:00
self.anonymous_user = AnonymousUser
self.anonymous_user.is_authenticated = False
2021-05-26 21:57:29 +00:00
with patch("bookwyrm.preview_images.generate_site_preview_image_task.delay"):
self.settings = models.SiteSettings.objects.create(id=1)
2021-01-12 17:53:04 +00:00
def test_login_get(self):
2021-04-26 16:15:42 +00:00
"""there are so many views, this just makes sure it LOADS"""
2021-01-12 17:53:04 +00:00
login = views.Login.as_view()
2021-03-08 16:49:10 +00:00
request = self.factory.get("")
2021-01-12 17:53:04 +00:00
request.user = self.anonymous_user
result = login(request)
2021-01-12 17:09:05 +00:00
self.assertIsInstance(result, TemplateResponse)
result.render()
2021-01-12 17:09:05 +00:00
self.assertEqual(result.status_code, 200)
request.user = self.local_user
2021-01-12 17:53:04 +00:00
result = login(request)
2021-03-08 16:49:10 +00:00
self.assertEqual(result.url, "/")
2021-01-12 17:09:05 +00:00
self.assertEqual(result.status_code, 302)
def test_register(self):
2021-04-26 16:15:42 +00:00
"""create a user"""
2021-01-12 17:53:04 +00:00
view = views.Register.as_view()
self.assertEqual(models.User.objects.count(), 1)
2021-01-12 17:09:05 +00:00
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
2021-01-12 17:09:05 +00:00
{
2021-03-08 16:49:10 +00:00
"localname": "nutria-user.user_nutria",
"password": "mouseword",
"email": "aa@bb.cccc",
},
)
2021-05-26 21:57:29 +00:00
with patch("bookwyrm.preview_images.generate_user_preview_image_task.delay"):
with patch("bookwyrm.views.authentication.login"):
response = view(request)
2021-01-12 17:53:04 +00:00
self.assertEqual(models.User.objects.count(), 2)
2021-01-12 17:09:05 +00:00
self.assertEqual(response.status_code, 302)
nutria = models.User.objects.last()
2021-03-08 16:49:10 +00:00
self.assertEqual(nutria.username, "nutria-user.user_nutria@%s" % DOMAIN)
self.assertEqual(nutria.localname, "nutria-user.user_nutria")
2021-01-12 17:09:05 +00:00
self.assertEqual(nutria.local, True)
def test_register_trailing_space(self):
2021-04-26 16:15:42 +00:00
"""django handles this so weirdly"""
2021-01-12 17:53:04 +00:00
view = views.Register.as_view()
2021-01-12 17:09:05 +00:00
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
{"localname": "nutria ", "password": "mouseword", "email": "aa@bb.ccc"},
)
2021-05-26 21:57:29 +00:00
with patch("bookwyrm.preview_images.generate_user_preview_image_task.delay"):
with patch("bookwyrm.views.authentication.login"):
response = view(request)
2021-01-12 17:53:04 +00:00
self.assertEqual(models.User.objects.count(), 2)
2021-01-12 17:09:05 +00:00
self.assertEqual(response.status_code, 302)
nutria = models.User.objects.last()
2021-03-08 16:49:10 +00:00
self.assertEqual(nutria.username, "nutria@%s" % DOMAIN)
self.assertEqual(nutria.localname, "nutria")
2021-01-12 17:09:05 +00:00
self.assertEqual(nutria.local, True)
def test_register_invalid_email(self):
2021-04-26 16:15:42 +00:00
"""gotta have an email"""
2021-01-12 17:53:04 +00:00
view = views.Register.as_view()
self.assertEqual(models.User.objects.count(), 1)
2021-01-12 17:09:05 +00:00
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/", {"localname": "nutria", "password": "mouseword", "email": "aa"}
)
2021-01-12 17:53:04 +00:00
response = view(request)
self.assertEqual(models.User.objects.count(), 1)
response.render()
2021-01-12 17:09:05 +00:00
def test_register_invalid_username(self):
2021-04-26 16:15:42 +00:00
"""gotta have an email"""
2021-01-12 17:53:04 +00:00
view = views.Register.as_view()
self.assertEqual(models.User.objects.count(), 1)
2021-01-12 17:09:05 +00:00
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
{"localname": "nut@ria", "password": "mouseword", "email": "aa@bb.ccc"},
)
2021-01-12 17:53:04 +00:00
response = view(request)
self.assertEqual(models.User.objects.count(), 1)
response.render()
2021-01-12 17:09:05 +00:00
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
{"localname": "nutr ia", "password": "mouseword", "email": "aa@bb.ccc"},
)
2021-01-12 17:53:04 +00:00
response = view(request)
self.assertEqual(models.User.objects.count(), 1)
response.render()
2021-01-12 17:09:05 +00:00
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
{"localname": "nut@ria", "password": "mouseword", "email": "aa@bb.ccc"},
)
2021-01-12 17:53:04 +00:00
response = view(request)
self.assertEqual(models.User.objects.count(), 1)
response.render()
2021-01-12 17:09:05 +00:00
def test_register_closed_instance(self):
2021-04-26 16:15:42 +00:00
"""you can't just register"""
2021-01-12 17:53:04 +00:00
view = views.Register.as_view()
2021-01-12 17:09:05 +00:00
self.settings.allow_registration = False
self.settings.save()
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
{"localname": "nutria ", "password": "mouseword", "email": "aa@bb.ccc"},
)
2021-01-12 17:09:05 +00:00
with self.assertRaises(PermissionDenied):
2021-01-12 17:53:04 +00:00
view(request)
2021-01-12 17:09:05 +00:00
def test_register_invite(self):
2021-04-26 16:15:42 +00:00
"""you can't just register"""
2021-01-12 17:53:04 +00:00
view = views.Register.as_view()
2021-01-12 17:09:05 +00:00
self.settings.allow_registration = False
self.settings.save()
models.SiteInvite.objects.create(
2021-03-08 16:49:10 +00:00
code="testcode", user=self.local_user, use_limit=1
)
2021-01-12 17:09:05 +00:00
self.assertEqual(models.SiteInvite.objects.get().times_used, 0)
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
2021-01-12 17:09:05 +00:00
{
2021-03-08 16:49:10 +00:00
"localname": "nutria",
"password": "mouseword",
"email": "aa@bb.ccc",
"invite_code": "testcode",
},
)
2021-05-26 21:57:29 +00:00
with patch("bookwyrm.preview_images.generate_user_preview_image_task.delay"):
with patch("bookwyrm.views.authentication.login"):
response = view(request)
2021-01-12 17:53:04 +00:00
self.assertEqual(models.User.objects.count(), 2)
2021-01-12 17:09:05 +00:00
self.assertEqual(response.status_code, 302)
self.assertEqual(models.SiteInvite.objects.get().times_used, 1)
# invite already used to max capacity
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
2021-01-12 17:09:05 +00:00
{
2021-03-08 16:49:10 +00:00
"localname": "nutria2",
"password": "mouseword",
"email": "aa@bb.ccc",
"invite_code": "testcode",
},
)
2021-01-12 17:09:05 +00:00
with self.assertRaises(PermissionDenied):
2021-01-12 17:53:04 +00:00
response = view(request)
self.assertEqual(models.User.objects.count(), 2)
2021-01-12 17:09:05 +00:00
# bad invite code
request = self.factory.post(
2021-03-08 16:49:10 +00:00
"register/",
2021-01-12 17:09:05 +00:00
{
2021-03-08 16:49:10 +00:00
"localname": "nutria3",
"password": "mouseword",
"email": "aa@bb.ccc",
"invite_code": "dkfkdjgdfkjgkdfj",
},
)
2021-01-12 17:09:05 +00:00
with self.assertRaises(Http404):
2021-01-12 17:53:04 +00:00
response = view(request)
self.assertEqual(models.User.objects.count(), 2)