lib.itmens/social/tests.py

24 lines
921 B
Python
Raw Normal View History

2022-12-13 06:44:29 +00:00
from django.test import TestCase
from catalog.models import *
from journal.models import *
from .models import *
from users.models import User
class SocialTest(TestCase):
def setUp(self):
self.book1 = Edition.objects.create(title="Hyperion")
self.book2 = Edition.objects.create(title="Andymion")
self.alice = User.objects.create(mastodon_site="MySpace", username="Alice")
self.alice.queue_manager.initialize()
self.bob = User.objects.create(mastodon_site="KKCity", username="Bob")
self.bob.queue_manager.initialize()
def test_timeline(self):
timeline = list(self.alice.activity_manager.get_viewable_activities())
self.assertEqual(timeline, [])
self.alice.queue_manager.update_for_item(self.book1, QueueType.WISHED)
timeline = list(self.alice.activity_manager.get_viewable_activities())
self.assertEqual(len(timeline), 1)