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)