mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-06-06 15:29:34 +00:00
9d502f5ee2
Pylint's `bad-classmethod-argument` is disabled for each definition to avoid rewriting the method bodies just to rename `self` → `cls`. This can be done gradually, as the setUpTestData methods are modified along the way.
67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
""" test for app action functionality """
|
|
from collections import namedtuple
|
|
from unittest.mock import patch
|
|
from django.template.response import TemplateResponse
|
|
from django.test import TestCase
|
|
from django.test.client import RequestFactory
|
|
from bookwyrm.tests.validate_html import validate_html
|
|
|
|
from bookwyrm import models, views
|
|
|
|
|
|
class ImportTroubleshootViews(TestCase):
|
|
"""goodreads import views"""
|
|
|
|
@classmethod
|
|
def setUpTestData(self): # pylint: disable=bad-classmethod-argument
|
|
"""we need basic test data and mocks"""
|
|
with patch("bookwyrm.suggested_users.rerank_suggestions_task.delay"), patch(
|
|
"bookwyrm.activitystreams.populate_stream_task.delay"
|
|
), patch("bookwyrm.lists_stream.populate_lists_task.delay"):
|
|
self.local_user = models.User.objects.create_user(
|
|
"mouse@local.com",
|
|
"mouse@mouse.mouse",
|
|
"password",
|
|
local=True,
|
|
localname="mouse",
|
|
)
|
|
models.SiteSettings.objects.create()
|
|
|
|
def setUp(self):
|
|
"""individual test setup"""
|
|
self.factory = RequestFactory()
|
|
|
|
def test_import_troubleshoot_get(self):
|
|
"""there are so many views, this just makes sure it LOADS"""
|
|
view = views.ImportTroubleshoot.as_view()
|
|
import_job = models.ImportJob.objects.create(user=self.local_user, mappings={})
|
|
request = self.factory.get("")
|
|
request.user = self.local_user
|
|
with patch("bookwyrm.tasks.app.AsyncResult") as async_result:
|
|
async_result.return_value = []
|
|
result = view(request, import_job.id)
|
|
self.assertIsInstance(result, TemplateResponse)
|
|
validate_html(result.render())
|
|
self.assertEqual(result.status_code, 200)
|
|
|
|
def test_retry_import(self):
|
|
"""retry failed items"""
|
|
view = views.ImportTroubleshoot.as_view()
|
|
import_job = models.ImportJob.objects.create(
|
|
user=self.local_user, privacy="unlisted", mappings={}
|
|
)
|
|
request = self.factory.post("")
|
|
request.user = self.local_user
|
|
|
|
MockTask = namedtuple("Task", ("id"))
|
|
with patch("bookwyrm.models.import_job.start_import_task.delay") as mock:
|
|
mock.return_value = MockTask(123)
|
|
view(request, import_job.id)
|
|
|
|
self.assertEqual(models.ImportJob.objects.count(), 2)
|
|
retry_job = models.ImportJob.objects.last()
|
|
|
|
self.assertTrue(retry_job.retry)
|
|
self.assertEqual(retry_job.user, self.local_user)
|
|
self.assertEqual(retry_job.privacy, "unlisted")
|