mirrors/bookwyrm
1
0
Fork 1
mirror of https://github.com/bookwyrm-social/bookwyrm.git synced 2025-02-16 19:15:16 +00:00
Code Issues Projects Releases Wiki Activity

Adds test for query that creates bookwyrm export query

Browse source
This commit is contained in:
Mouse Reeve 2024-06-08 20:10:08 -07:00
parent b84ca3d03f
commit 9984e853e5
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
bookwyrm/tests/models/test_bookwyrm_export_job.py
View file

@ -5,13 +5,16 @@ import pathlib
from unittest.mock import patch
from django.db import connection
from django.utils import timezone
from django.test import TestCase
from bookwyrm import models
from bookwyrm.tests.query_logger import QueryLogger, raise_long_query_runtime
from bookwyrm.utils.tar import BookwyrmTarFile
# pylint: disable=invalid-name
class BookwyrmExportJob(TestCase):
"""testing user export functions"""
@ -159,6 +162,14 @@ class BookwyrmExportJob(TestCase):
models.bookwyrm_export_job.create_export_json_task(job_id=self.job.id)
self.job.refresh_from_db()
def test_create_export_job_query_time(self):
"""test the creation of the job"""
query_logger = QueryLogger()
with connection.execute_wrapper(query_logger):
with patch("bookwyrm.models.bookwyrm_export_job.create_archive_task.delay"):
models.bookwyrm_export_job.create_export_json_task(job_id=self.job.id)
raise_long_query_runtime(query_logger.queries)
def test_add_book_to_user_export_job(self):
"""does AddBookToUserExportJob ...add the book to the export?"""
self.assertIsNotNone(self.job.export_json["books"])