''' 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 from bookwyrm.activitypub import ActivitypubResponse class ListViews(TestCase): ''' tag views''' 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', 'mouseword', local=True, localname='mouse', remote_id='https://example.com/users/mouse', ) self.rat = models.User.objects.create_user( 'rat@local.com', 'rat@rat.com', 'ratword', local=True, localname='rat', remote_id='https://example.com/users/rat', ) work = models.Work.objects.create(title='Work') self.book = models.Edition.objects.create( title='Example Edition', remote_id='https://example.com/book/1', parent_work=work, ) with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): self.list = models.List.objects.create( name='Test List', user=self.local_user) self.anonymous_user = AnonymousUser self.anonymous_user.is_authenticated = False models.SiteSettings.objects.create() def test_lists_page(self): ''' there are so many views, this just makes sure it LOADS ''' view = views.Lists.as_view() with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): models.List.objects.create(name='Public list', user=self.local_user) models.List.objects.create( name='Private list', privacy='private', user=self.local_user) request = self.factory.get('') request.user = self.local_user result = view(request) self.assertIsInstance(result, TemplateResponse) result.render() self.assertEqual(result.status_code, 200) request.user = self.anonymous_user result = view(request) self.assertIsInstance(result, TemplateResponse) result.render() self.assertEqual(result.status_code, 200) def test_lists_create(self): ''' create list view ''' real_broadcast = models.List.broadcast def mock_broadcast(_, activity, user, **kwargs): ''' ok ''' self.assertEqual(user.remote_id, self.local_user.remote_id) self.assertEqual(activity['type'], 'Create') self.assertEqual(activity['actor'], self.local_user.remote_id) models.List.broadcast = mock_broadcast view = views.Lists.as_view() request = self.factory.post('', { 'name': 'A list', 'description': 'wow', 'privacy': 'unlisted', 'curation': 'open', 'user': self.local_user.id, }) request.user = self.local_user result = view(request) self.assertEqual(result.status_code, 302) new_list = models.List.objects.filter(name='A list').get() self.assertEqual(new_list.description, 'wow') self.assertEqual(new_list.privacy, 'unlisted') self.assertEqual(new_list.curation, 'open') models.List.broadcast = real_broadcast def test_list_page(self): ''' there are so many views, this just makes sure it LOADS ''' view = views.List.as_view() request = self.factory.get('') request.user = self.local_user with patch('bookwyrm.views.list.is_api_request') as is_api: is_api.return_value = False result = view(request, self.list.id) self.assertIsInstance(result, TemplateResponse) result.render() self.assertEqual(result.status_code, 200) request.user = self.anonymous_user with patch('bookwyrm.views.list.is_api_request') as is_api: is_api.return_value = False result = view(request, self.list.id) self.assertIsInstance(result, TemplateResponse) result.render() self.assertEqual(result.status_code, 200) with patch('bookwyrm.views.list.is_api_request') as is_api: is_api.return_value = True result = view(request, self.list.id) self.assertIsInstance(result, ActivitypubResponse) self.assertEqual(result.status_code, 200) request = self.factory.get('/?page=1') request.user = self.local_user with patch('bookwyrm.views.list.is_api_request') as is_api: is_api.return_value = True result = view(request, self.list.id) self.assertIsInstance(result, ActivitypubResponse) self.assertEqual(result.status_code, 200) def test_list_edit(self): ''' edit a list ''' real_broadcast = models.List.broadcast def mock_broadcast(_, activity, user, **kwargs): ''' ok ''' self.assertEqual(user.remote_id, self.local_user.remote_id) self.assertEqual(activity['type'], 'Update') self.assertEqual(activity['actor'], self.local_user.remote_id) self.assertEqual(activity['object']['id'], self.list.remote_id) models.List.broadcast = mock_broadcast view = views.List.as_view() request = self.factory.post('', { 'name': 'New Name', 'description': 'wow', 'privacy': 'direct', 'curation': 'curated', 'user': self.local_user.id, }) request.user = self.local_user result = view(request, self.list.id) self.assertEqual(result.status_code, 302) self.list.refresh_from_db() self.assertEqual(self.list.name, 'New Name') self.assertEqual(self.list.description, 'wow') self.assertEqual(self.list.privacy, 'direct') self.assertEqual(self.list.curation, 'curated') models.List.broadcast = real_broadcast def test_curate_page(self): ''' there are so many views, this just makes sure it LOADS ''' view = views.Curate.as_view() with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): models.List.objects.create(name='Public list', user=self.local_user) models.List.objects.create( name='Private list', privacy='private', user=self.local_user) request = self.factory.get('') request.user = self.local_user result = view(request, self.list.id) self.assertIsInstance(result, TemplateResponse) result.render() self.assertEqual(result.status_code, 200) request.user = self.anonymous_user result = view(request, self.list.id) self.assertEqual(result.status_code, 302) def test_curate_approve(self): ''' approve a pending item ''' real_broadcast = models.List.broadcast def mock_broadcast(_, activity, user, **kwargs): ''' ok ''' self.assertEqual(user.remote_id, self.local_user.remote_id) self.assertEqual(activity['type'], 'Add') self.assertEqual(activity['actor'], self.local_user.remote_id) self.assertEqual(activity['target'], self.list.remote_id) models.ListItem.broadcast = mock_broadcast view = views.Curate.as_view() with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): pending = models.ListItem.objects.create( book_list=self.list, user=self.local_user, book=self.book, approved=False ) request = self.factory.post('', { 'item': pending.id, 'approved': 'true', }) request.user = self.local_user view(request, self.list.id) pending.refresh_from_db() self.assertEqual(self.list.books.count(), 1) self.assertEqual(self.list.listitem_set.first(), pending) self.assertTrue(pending.approved) models.ListItem.broadcast = real_broadcast def test_curate_reject(self): ''' approve a pending item ''' view = views.Curate.as_view() with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): pending = models.ListItem.objects.create( book_list=self.list, user=self.local_user, book=self.book, approved=False ) request = self.factory.post('', { 'item': pending.id, 'approved': 'false', }) request.user = self.local_user with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): view(request, self.list.id) self.assertFalse(self.list.books.exists()) self.assertFalse(models.ListItem.objects.exists()) def test_add_book(self): ''' put a book on a list ''' real_broadcast = models.List.broadcast def mock_broadcast(_, activity, user): ''' ok ''' self.assertEqual(user.remote_id, self.local_user.remote_id) self.assertEqual(activity['type'], 'Add') self.assertEqual(activity['actor'], self.local_user.remote_id) self.assertEqual(activity['target'], self.list.remote_id) models.ListItem.broadcast = mock_broadcast request = self.factory.post('', { 'book': self.book.id, }) request.user = self.local_user views.list.add_book(request, self.list.id) item = self.list.listitem_set.get() self.assertEqual(item.book, self.book) self.assertEqual(item.user, self.local_user) self.assertTrue(item.approved) models.ListItem.broadcast = real_broadcast def test_add_book_outsider(self): ''' put a book on a list ''' real_broadcast = models.List.broadcast def mock_broadcast(_, activity, user): ''' ok ''' self.assertEqual(user.remote_id, self.rat.remote_id) self.assertEqual(activity['type'], 'Add') self.assertEqual(activity['actor'], self.rat.remote_id) self.assertEqual(activity['target'], self.list.remote_id) models.ListItem.broadcast = mock_broadcast self.list.curation = 'open' self.list.save(broadcast=False) request = self.factory.post('', { 'book': self.book.id, }) request.user = self.rat views.list.add_book(request, self.list.id) item = self.list.listitem_set.get() self.assertEqual(item.book, self.book) self.assertEqual(item.user, self.rat) self.assertTrue(item.approved) models.ListItem.broadcast = real_broadcast def test_add_book_pending(self): ''' put a book on a list awaiting approval ''' real_broadcast = models.List.broadcast def mock_broadcast(_, activity, user): ''' ok ''' self.assertEqual(user.remote_id, self.rat.remote_id) self.assertEqual(activity['type'], 'Add') self.assertEqual(activity['actor'], self.rat.remote_id) self.assertEqual(activity['target'], self.list.remote_id) self.assertEqual(activity['object']['id'], self.book.remote_id) models.ListItem.broadcast = mock_broadcast self.list.curation = 'curated' self.list.save(broadcast=False) request = self.factory.post('', { 'book': self.book.id, }) request.user = self.rat views.list.add_book(request, self.list.id) item = self.list.listitem_set.get() self.assertEqual(item.book, self.book) self.assertEqual(item.user, self.rat) self.assertFalse(item.approved) models.ListItem.broadcast = real_broadcast def test_add_book_self_curated(self): ''' put a book on a list automatically approved ''' real_broadcast = models.ListItem.broadcast def mock_broadcast(_, activity, user): ''' ok ''' self.assertEqual(user.remote_id, self.local_user.remote_id) self.assertEqual(activity['type'], 'Add') self.assertEqual(activity['actor'], self.local_user.remote_id) self.assertEqual(activity['target'], self.list.remote_id) models.ListItem.broadcast = mock_broadcast self.list.curation = 'curated' self.list.save(broadcast=False) request = self.factory.post('', { 'book': self.book.id, }) request.user = self.local_user views.list.add_book(request, self.list.id) item = self.list.listitem_set.get() self.assertEqual(item.book, self.book) self.assertEqual(item.user, self.local_user) self.assertTrue(item.approved) models.ListItem.broadcast = real_broadcast def test_remove_book(self): ''' take an item off a list ''' real_broadcast = models.ListItem.broadcast with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): item = models.ListItem.objects.create( book_list=self.list, user=self.local_user, book=self.book, ) self.assertTrue(self.list.listitem_set.exists()) def mock_broadcast(_, activity, user): ''' ok ''' self.assertEqual(user.remote_id, self.local_user.remote_id) self.assertEqual(activity['type'], 'Remove') self.assertEqual(activity['actor'], self.local_user.remote_id) self.assertEqual(activity['target'], self.list.remote_id) models.ListItem.broadcast = mock_broadcast request = self.factory.post('', { 'item': item.id, }) request.user = self.local_user views.list.remove_book(request, self.list.id) self.assertFalse(self.list.listitem_set.exists()) models.ListItem.broadcast = real_broadcast def test_remove_book_unauthorized(self): ''' take an item off a list ''' with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'): item = models.ListItem.objects.create( book_list=self.list, user=self.local_user, book=self.book, ) self.assertTrue(self.list.listitem_set.exists()) request = self.factory.post('', { 'item': item.id, }) request.user = self.rat views.list.remove_book(request, self.list.id) self.assertTrue(self.list.listitem_set.exists())