new shelf type: fix unit test

This commit is contained in:
Your Name 2024-01-20 22:55:19 -05:00 committed by Henri Dickson
parent fbc23a1e7f
commit 8982934b6a

View file

@ -47,7 +47,7 @@ class ShelfTest(TestCase):
def test_shelf(self): def test_shelf(self):
user = User.register(email="a@b.com", username="user") user = User.register(email="a@b.com", username="user")
shelf_manager = user.identity.shelf_manager shelf_manager = user.identity.shelf_manager
self.assertEqual(len(shelf_manager.shelf_list.items()), 3) self.assertEqual(len(shelf_manager.shelf_list.items()), 4)
book1 = Edition.objects.create(title="Hyperion") book1 = Edition.objects.create(title="Hyperion")
book2 = Edition.objects.create(title="Andymion") book2 = Edition.objects.create(title="Andymion")
q1 = shelf_manager.get_shelf(ShelfType.WISHLIST) q1 = shelf_manager.get_shelf(ShelfType.WISHLIST)