lib.itmens/journal/tests.py

154 lines
6.8 KiB
Python
Raw Normal View History

2022-12-11 23:20:28 +00:00
from django.test import TestCase
from .models import *
from catalog.models import *
from users.models import User
class CollectionTest(TestCase):
def setUp(self):
self.book1 = Edition.objects.create(title="Hyperion")
self.book2 = Edition.objects.create(title="Andymion")
self.user = User.objects.create()
pass
def test_collection(self):
collection = Collection.objects.create(title="test", owner=self.user)
2022-12-12 16:46:37 +00:00
collection = Collection.objects.filter(title="test", owner=self.user).first()
self.assertEqual(collection.catalog_item.title, "test")
2022-12-29 14:30:31 -05:00
member1 = collection.append_item(self.book1)
member1.note = "my notes"
member1.save()
2022-12-11 23:20:28 +00:00
collection.append_item(self.book2)
self.assertEqual(list(collection.ordered_items), [self.book1, self.book2])
collection.move_up_item(self.book1)
self.assertEqual(list(collection.ordered_items), [self.book1, self.book2])
collection.move_up_item(self.book2)
self.assertEqual(list(collection.ordered_items), [self.book2, self.book1])
2022-12-29 14:30:31 -05:00
member1 = collection.get_member_for_item(self.book1)
self.assertEqual(member1.note, "my notes")
2022-12-11 23:20:28 +00:00
2022-12-13 18:12:43 +00:00
class ShelfTest(TestCase):
2022-12-11 23:20:28 +00:00
def setUp(self):
pass
2022-12-13 18:12:43 +00:00
def test_shelf(self):
2022-12-11 23:20:28 +00:00
user = User.objects.create(mastodon_site="site", username="name")
2022-12-13 18:12:43 +00:00
shelf_manager = ShelfManager(user=user)
2022-12-29 23:49:28 -05:00
self.assertEqual(user.shelf_set.all().count(), 3)
2022-12-11 23:20:28 +00:00
book1 = Edition.objects.create(title="Hyperion")
book2 = Edition.objects.create(title="Andymion")
2022-12-29 23:49:28 -05:00
q1 = shelf_manager.get_shelf(ShelfType.WISHLIST)
q2 = shelf_manager.get_shelf(ShelfType.PROGRESS)
2022-12-11 23:20:28 +00:00
self.assertIsNotNone(q1)
self.assertIsNotNone(q2)
self.assertEqual(q1.members.all().count(), 0)
self.assertEqual(q2.members.all().count(), 0)
2022-12-17 16:18:16 -05:00
shelf_manager.move_item(book1, ShelfType.WISHLIST)
shelf_manager.move_item(book2, ShelfType.WISHLIST)
2022-12-11 23:20:28 +00:00
self.assertEqual(q1.members.all().count(), 2)
2022-12-17 16:18:16 -05:00
shelf_manager.move_item(book1, ShelfType.PROGRESS)
2022-12-11 23:20:28 +00:00
self.assertEqual(q1.members.all().count(), 1)
self.assertEqual(q2.members.all().count(), 1)
2022-12-29 14:30:31 -05:00
shelf_manager.move_item(book1, ShelfType.PROGRESS, metadata={"progress": 1})
2022-12-11 23:20:28 +00:00
self.assertEqual(q1.members.all().count(), 1)
self.assertEqual(q2.members.all().count(), 1)
2022-12-13 18:12:43 +00:00
log = shelf_manager.get_log_for_item(book1)
2022-12-11 23:20:28 +00:00
self.assertEqual(log.count(), 3)
2022-12-29 14:30:31 -05:00
shelf_manager.move_item(book1, ShelfType.PROGRESS, metadata={"progress": 1})
2022-12-13 18:12:43 +00:00
log = shelf_manager.get_log_for_item(book1)
2022-12-11 23:20:28 +00:00
self.assertEqual(log.count(), 3)
2022-12-29 14:30:31 -05:00
shelf_manager.move_item(book1, ShelfType.PROGRESS, metadata={"progress": 10})
2022-12-13 18:12:43 +00:00
log = shelf_manager.get_log_for_item(book1)
2022-12-11 23:20:28 +00:00
self.assertEqual(log.count(), 4)
2022-12-17 16:18:16 -05:00
shelf_manager.move_item(book1, ShelfType.PROGRESS)
2022-12-13 18:12:43 +00:00
log = shelf_manager.get_log_for_item(book1)
2022-12-11 23:20:28 +00:00
self.assertEqual(log.count(), 4)
2022-12-29 14:30:31 -05:00
self.assertEqual(log.last().metadata, {"progress": 10})
shelf_manager.move_item(book1, ShelfType.PROGRESS, metadata={"progress": 90})
2022-12-13 18:12:43 +00:00
log = shelf_manager.get_log_for_item(book1)
2022-12-11 23:20:28 +00:00
self.assertEqual(log.count(), 5)
2022-12-16 01:08:10 -05:00
self.assertEqual(Mark(user, book1).visibility, 0)
2022-12-29 14:30:31 -05:00
shelf_manager.move_item(
book1, ShelfType.PROGRESS, metadata={"progress": 90}, visibility=1
)
2022-12-16 01:08:10 -05:00
self.assertEqual(Mark(user, book1).visibility, 1)
self.assertEqual(shelf_manager.get_log_for_item(book1).count(), 5)
2022-12-12 16:46:37 +00:00
class TagTest(TestCase):
def setUp(self):
self.book1 = Edition.objects.create(title="Hyperion")
self.book2 = Edition.objects.create(title="Andymion")
self.movie1 = Edition.objects.create(title="Hyperion, The Movie")
self.user1 = User.objects.create(mastodon_site="site", username="name")
self.user2 = User.objects.create(mastodon_site="site2", username="name2")
self.user3 = User.objects.create(mastodon_site="site2", username="name3")
pass
2022-12-17 16:18:16 -05:00
def test_user_tag(self):
2022-12-29 14:30:31 -05:00
t1 = "tag-1"
t2 = "tag-2"
t3 = "tag-3"
2022-12-17 16:18:16 -05:00
TagManager.tag_item_by_user(self.book1, self.user2, [t1, t3])
2022-12-17 17:18:08 -05:00
self.assertEqual(self.book1.tags, [t1, t3])
2022-12-17 16:18:16 -05:00
TagManager.tag_item_by_user(self.book1, self.user2, [t2, t3])
2022-12-17 17:18:08 -05:00
self.assertEqual(self.book1.tags, [t2, t3])
2022-12-17 16:18:16 -05:00
2022-12-12 16:46:37 +00:00
def test_tag(self):
2022-12-29 14:30:31 -05:00
t1 = "tag-1"
t2 = "tag-2"
t3 = "tag-3"
2022-12-15 17:29:35 -05:00
TagManager.add_tag_by_user(self.book1, t3, self.user2)
TagManager.add_tag_by_user(self.book1, t1, self.user1)
TagManager.add_tag_by_user(self.book1, t1, self.user2)
2022-12-17 16:18:16 -05:00
self.assertEqual(self.book1.tags, [t1, t3])
2022-12-15 17:29:35 -05:00
TagManager.add_tag_by_user(self.book1, t2, self.user1, default_visibility=2)
2022-12-12 16:46:37 +00:00
self.assertEqual(self.book1.tags, [t1, t3])
2022-12-15 17:29:35 -05:00
TagManager.add_tag_by_user(self.book1, t3, self.user1)
TagManager.add_tag_by_user(self.book1, t3, self.user3)
2022-12-17 17:18:08 -05:00
self.assertEqual(self.book1.tags, [t1, t3])
2022-12-15 17:29:35 -05:00
TagManager.add_tag_by_user(self.book1, t3, self.user3)
TagManager.add_tag_by_user(self.book1, t3, self.user3)
2022-12-12 16:46:37 +00:00
self.assertEqual(Tag.objects.count(), 6)
2022-12-15 17:29:35 -05:00
TagManager.add_tag_by_user(self.book2, t1, self.user2)
2022-12-12 16:46:37 +00:00
self.assertEqual(self.user2.tags, [t1, t3])
2022-12-15 17:29:35 -05:00
TagManager.add_tag_by_user(self.book2, t3, self.user2)
TagManager.add_tag_by_user(self.movie1, t3, self.user2)
2022-12-28 01:09:55 -05:00
self.assertEqual(sorted(self.user2.tags), [t1, t3])
2022-12-15 17:29:35 -05:00
class MarkTest(TestCase):
def setUp(self):
self.book1 = Edition.objects.create(title="Hyperion")
self.user1 = User.objects.create(mastodon_site="site", username="name")
pass
def test_mark(self):
mark = Mark(self.user1, self.book1)
self.assertEqual(mark.shelf_type, None)
self.assertEqual(mark.shelf_label, None)
self.assertEqual(mark.text, None)
self.assertEqual(mark.rating, None)
self.assertEqual(mark.visibility, None)
self.assertEqual(mark.review, None)
self.assertEqual(mark.tags, [])
2022-12-29 14:30:31 -05:00
mark.update(ShelfType.WISHLIST, "a gentle comment", 9, 1)
2022-12-15 17:29:35 -05:00
mark = Mark(self.user1, self.book1)
2022-12-17 16:18:16 -05:00
self.assertEqual(mark.shelf_type, ShelfType.WISHLIST)
2022-12-29 23:49:28 -05:00
self.assertEqual(mark.shelf_label, "想读的书")
2022-12-29 14:30:31 -05:00
self.assertEqual(mark.text, "a gentle comment")
2022-12-15 17:29:35 -05:00
self.assertEqual(mark.rating, 9)
self.assertEqual(mark.visibility, 1)
self.assertEqual(mark.review, None)
self.assertEqual(mark.tags, [])
2022-12-29 14:30:31 -05:00
review = Review.review_item_by_user(self.book1, self.user1, "Critic", "Review")
2022-12-15 17:29:35 -05:00
mark = Mark(self.user1, self.book1)
self.assertEqual(mark.review, review)
2022-12-29 14:30:31 -05:00
TagManager.tag_item_by_user(self.book1, self.user1, [" Sci-Fi ", " fic "])
2022-12-15 17:29:35 -05:00
mark = Mark(self.user1, self.book1)
2022-12-29 14:30:31 -05:00
self.assertEqual(mark.tags, ["fic", "sci-fi"])