24 lines
921 B
Python
24 lines
921 B
Python
|
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)
|