66 lines
3 KiB
Python
66 lines
3 KiB
Python
|
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)
|
||
|
|
collection.append_item(self.book1)
|
||
|
|
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])
|
||
|
|
|
||
|
|
|
||
|
|
class QueueTest(TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_queue(self):
|
||
|
|
user = User.objects.create(mastodon_site="site", username="name")
|
||
|
|
queue_manager = QueueManager(user=user)
|
||
|
|
queue_manager.initialize()
|
||
|
|
self.assertEqual(user.queues.all().count(), 30)
|
||
|
|
book1 = Edition.objects.create(title="Hyperion")
|
||
|
|
book2 = Edition.objects.create(title="Andymion")
|
||
|
|
q1 = queue_manager.get_queue(ItemCategory.Book, QueueType.WISHED)
|
||
|
|
q2 = queue_manager.get_queue(ItemCategory.Book, QueueType.STARTED)
|
||
|
|
self.assertIsNotNone(q1)
|
||
|
|
self.assertIsNotNone(q2)
|
||
|
|
self.assertEqual(q1.members.all().count(), 0)
|
||
|
|
self.assertEqual(q2.members.all().count(), 0)
|
||
|
|
queue_manager.update_for_item(book1, QueueType.WISHED)
|
||
|
|
queue_manager.update_for_item(book2, QueueType.WISHED)
|
||
|
|
self.assertEqual(q1.members.all().count(), 2)
|
||
|
|
queue_manager.update_for_item(book1, QueueType.STARTED)
|
||
|
|
self.assertEqual(q1.members.all().count(), 1)
|
||
|
|
self.assertEqual(q2.members.all().count(), 1)
|
||
|
|
queue_manager.update_for_item(book1, QueueType.STARTED, metadata={'progress': 1})
|
||
|
|
self.assertEqual(q1.members.all().count(), 1)
|
||
|
|
self.assertEqual(q2.members.all().count(), 1)
|
||
|
|
log = queue_manager.get_log_for_item(book1)
|
||
|
|
self.assertEqual(log.count(), 3)
|
||
|
|
queue_manager.update_for_item(book1, QueueType.STARTED, metadata={'progress': 1})
|
||
|
|
log = queue_manager.get_log_for_item(book1)
|
||
|
|
self.assertEqual(log.count(), 3)
|
||
|
|
queue_manager.update_for_item(book1, QueueType.STARTED, metadata={'progress': 10})
|
||
|
|
log = queue_manager.get_log_for_item(book1)
|
||
|
|
self.assertEqual(log.count(), 4)
|
||
|
|
queue_manager.update_for_item(book1, QueueType.STARTED)
|
||
|
|
log = queue_manager.get_log_for_item(book1)
|
||
|
|
self.assertEqual(log.count(), 4)
|
||
|
|
self.assertEqual(log.order_by('queued_time').last().metadata, {'progress': 10})
|
||
|
|
queue_manager.update_for_item(book1, QueueType.STARTED, metadata={'progress': 100})
|
||
|
|
log = queue_manager.get_log_for_item(book1)
|
||
|
|
self.assertEqual(log.count(), 5)
|