Fix koreader importing

This commit is contained in:
2024-01-26 23:44:52 -05:00
parent f22ef1a163
commit dfc1365fa3
3 changed files with 35 additions and 12 deletions

View File

@ -1,9 +1,13 @@
import hashlib
import pytest
import random
import pytest
from django.contrib.auth import get_user_model
from vrobbler.apps.books.koreader import KoReaderBookColumn
User = get_user_model()
ordinal = lambda n: "%d%s" % (
n,
"tsnrhtdd"[(n // 10 % 10 != 1) * (n % 10 < 4) * n % 10 :: 4],
@ -112,3 +116,8 @@ class KoReaderBookRows:
@pytest.fixture
def koreader_rows():
return KoReaderBookRows(book_count=1)
@pytest.fixture
def demo_user():
return User.objects.create(email="demo@example.com")