moviewyrm/bookwyrm/tests/views/test_readthrough.py

106 lines
3.7 KiB
Python
Raw Normal View History

2021-03-08 16:49:10 +00:00
""" tests updating reading progress """
2021-02-08 17:38:28 +00:00
from datetime import datetime
from unittest.mock import patch
from django.test import TestCase, Client
from django.utils import timezone
2021-01-17 21:09:49 +00:00
from bookwyrm import models
2021-03-08 16:49:10 +00:00
2021-08-03 23:21:29 +00:00
@patch("bookwyrm.suggested_users.rerank_suggestions_task.delay")
2021-09-06 21:50:33 +00:00
@patch("bookwyrm.activitystreams.populate_stream_task.delay")
2021-11-12 17:17:00 +00:00
@patch("bookwyrm.models.activitypub_mixin.broadcast_task.apply_async")
2021-09-07 01:39:14 +00:00
@patch("bookwyrm.activitystreams.add_book_statuses_task.delay")
@patch("bookwyrm.activitystreams.remove_book_statuses_task.delay")
class ReadThrough(TestCase):
2021-04-26 16:15:42 +00:00
"""readthrough tests"""
2021-03-08 16:49:10 +00:00
def setUp(self):
2021-04-26 16:15:42 +00:00
"""basic user and book data"""
self.client = Client()
2021-08-02 23:05:40 +00:00
self.work = models.Work.objects.create(title="Example Work")
2021-08-02 23:05:40 +00:00
self.edition = models.Edition.objects.create(
title="Example Edition", parent_work=self.work
)
with patch("bookwyrm.suggested_users.rerank_suggestions_task.delay"), patch(
"bookwyrm.activitystreams.populate_stream_task.delay"
2021-12-09 23:03:01 +00:00
), patch("bookwyrm.lists_stream.populate_lists_task.delay"):
2021-08-03 17:25:53 +00:00
self.user = models.User.objects.create_user(
"cinco", "cinco@example.com", "seissiete", local=True, localname="cinco"
)
2021-11-12 17:17:00 +00:00
with patch("bookwyrm.models.activitypub_mixin.broadcast_task.apply_async"):
2021-02-08 17:38:28 +00:00
self.client.force_login(self.user)
2021-09-07 01:55:48 +00:00
@patch("bookwyrm.activitystreams.remove_user_statuses_task.delay")
2021-09-07 01:58:45 +00:00
def test_create_basic_readthrough(self, *_):
"""A basic readthrough doesn't create a progress update"""
self.assertEqual(self.edition.readthrough_set.count(), 0)
2021-03-08 16:49:10 +00:00
self.client.post(
2021-06-09 18:16:52 +00:00
"/reading-status/start/{}".format(self.edition.id),
2021-03-08 16:49:10 +00:00
{
"start_date": "2020-11-27",
},
)
readthroughs = self.edition.readthrough_set.all()
self.assertEqual(len(readthroughs), 1)
self.assertEqual(readthroughs[0].progressupdate_set.count(), 0)
2021-02-08 17:38:28 +00:00
self.assertEqual(
2021-03-08 16:49:10 +00:00
readthroughs[0].start_date, datetime(2020, 11, 27, tzinfo=timezone.utc)
)
self.assertEqual(readthroughs[0].progress, None)
self.assertEqual(readthroughs[0].finish_date, None)
2021-09-07 01:55:48 +00:00
@patch("bookwyrm.activitystreams.remove_user_statuses_task.delay")
2021-09-07 01:58:45 +00:00
def test_create_progress_readthrough(self, *_):
2021-04-26 16:15:42 +00:00
"""a readthrough with progress"""
self.assertEqual(self.edition.readthrough_set.count(), 0)
2021-03-08 16:49:10 +00:00
self.client.post(
2021-06-09 18:16:52 +00:00
"/reading-status/start/{}".format(self.edition.id),
2021-03-08 16:49:10 +00:00
{
"start_date": "2020-11-27",
},
)
readthroughs = self.edition.readthrough_set.all()
self.assertEqual(len(readthroughs), 1)
2021-02-08 17:38:28 +00:00
self.assertEqual(
2021-03-08 16:49:10 +00:00
readthroughs[0].start_date, datetime(2020, 11, 27, tzinfo=timezone.utc)
)
self.assertEqual(readthroughs[0].finish_date, None)
# Update progress
2021-03-08 16:49:10 +00:00
self.client.post(
"/edit-readthrough",
{
"id": readthroughs[0].id,
"progress": 100,
},
)
2021-03-08 16:49:10 +00:00
progress_updates = (
readthroughs[0].progressupdate_set.order_by("updated_date").all()
)
self.assertEqual(len(progress_updates), 1)
self.assertEqual(progress_updates[0].mode, models.ProgressMode.PAGE)
self.assertEqual(progress_updates[0].progress, 100)
2021-02-08 17:38:28 +00:00
# Edit doesn't publish anything
2021-03-08 16:49:10 +00:00
self.client.post(
"/delete-readthrough",
{
"id": readthroughs[0].id,
},
)
readthroughs = self.edition.readthrough_set.all()
updates = self.user.progressupdate_set.all()
self.assertEqual(len(readthroughs), 0)
self.assertEqual(len(updates), 0)