Updates tests

This commit is contained in:
Mouse Reeve
2021-01-02 08:38:27 -08:00
parent d828b0ead9
commit 86060f795d
5 changed files with 29 additions and 15 deletions

View File

@ -26,7 +26,7 @@ def get_edition(book_id):
''' look up a book in the db and return an edition '''
book = models.Book.objects.select_subclasses().get(id=book_id)
if isinstance(book, models.Work):
book = book.default_edition
book = book.get_default_edition()
return book
def get_user_from_username(username):