Updates books test

This commit is contained in:
Mouse Reeve 2021-09-27 12:26:56 -07:00
parent 36c00c49ab
commit dd3850a3ba
2 changed files with 4 additions and 2 deletions

View file

@ -70,6 +70,7 @@ def related_status(notification):
@register.simple_tag(takes_context=True) @register.simple_tag(takes_context=True)
def active_shelf(context, book): def active_shelf(context, book):
"""check what shelf a user has a book on, if any""" """check what shelf a user has a book on, if any"""
print(hasattr(book, "current_shelves"))
if hasattr(book, "current_shelves"): if hasattr(book, "current_shelves"):
return book.current_shelves[0] if len(book.current_shelves) else {"book": book} return book.current_shelves[0] if len(book.current_shelves) else {"book": book}

View file

@ -9,6 +9,7 @@ import responses
from django.contrib.auth.models import Group, Permission from django.contrib.auth.models import Group, Permission
from django.contrib.contenttypes.models import ContentType from django.contrib.contenttypes.models import ContentType
from django.core.files.uploadedfile import SimpleUploadedFile from django.core.files.uploadedfile import SimpleUploadedFile
from django.http import Http404
from django.template.response import TemplateResponse from django.template.response import TemplateResponse
from django.test import TestCase from django.test import TestCase
from django.test.client import RequestFactory from django.test.client import RequestFactory
@ -133,8 +134,8 @@ class BookViews(TestCase):
request.user = self.local_user request.user = self.local_user
with patch("bookwyrm.views.books.is_api_request") as is_api: with patch("bookwyrm.views.books.is_api_request") as is_api:
is_api.return_value = False is_api.return_value = False
result = view(request, 0) with self.assertRaises(Http404):
self.assertEqual(result.status_code, 404) view(request, 0)
def test_book_page_work_id(self): def test_book_page_work_id(self):
"""there are so many views, this just makes sure it LOADS""" """there are so many views, this just makes sure it LOADS"""