moviewyrm/bookwyrm/tests/test_templatetags.py

''' style fixes and lookups for templates '''
import re
from unittest.mock import patch

from dateutil.relativedelta import relativedelta
from django.test import TestCase
from django.utils import timezone

from bookwyrm import models
from bookwyrm.templatetags import bookwyrm_tags


class TemplateTags(TestCase):
    ''' lotta different things here '''
    def setUp(self):
        ''' create some filler objects '''
        self.user = models.User.objects.create_user(
            'mouse@example.com', 'mouse@mouse.mouse', 'mouseword',
            local=True, localname='mouse')
        with patch('bookwyrm.models.user.set_remote_server.delay'):
            self.remote_user = models.User.objects.create_user(
                'rat', 'rat@rat.rat', 'ratword',
                remote_id='http://example.com/rat', local=False)
        self.book = models.Edition.objects.create(title='Test Book')


    def test_dict_key(self):
        ''' just getting a value out of a dict '''
        test_dict = {'a': 1, 'b': 3}
        self.assertEqual(
            bookwyrm_tags.dict_key(test_dict, 'a'), 1)
        self.assertEqual(
            bookwyrm_tags.dict_key(test_dict, 'c'), 0)


    def test_get_user_rating(self):
        ''' get a user's most recent rating of a book '''
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            models.Review.objects.create(
                user=self.user, book=self.book, rating=3)
        self.assertEqual(
            bookwyrm_tags.get_user_rating(self.book, self.user), 3)


    def test_get_user_rating_doesnt_exist(self):
        ''' there is no rating available '''
        self.assertEqual(
            bookwyrm_tags.get_user_rating(self.book, self.user), 0)


    def test_get_user_identifer_local(self):
        ''' fall back to the simplest uid available '''
        self.assertNotEqual(self.user.username, self.user.localname)
        self.assertEqual(
            bookwyrm_tags.get_user_identifier(self.user), 'mouse')

    def test_get_user_identifer_remote(self):
        ''' for a remote user, should be their full username '''
        self.assertEqual(
            bookwyrm_tags.get_user_identifier(self.remote_user),
            'rat@example.com')

    def test_get_notification_count(self):
        ''' just countin' '''
        self.assertEqual(bookwyrm_tags.get_notification_count(self.user), 0)

        models.Notification.objects.create(
            user=self.user, notification_type='FAVORITE')
        models.Notification.objects.create(
            user=self.user, notification_type='MENTION')

        models.Notification.objects.create(
            user=self.remote_user, notification_type='FOLLOW')

        self.assertEqual(bookwyrm_tags.get_notification_count(self.user), 2)


    def test_get_replies(self):
        ''' direct replies to a status '''
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            parent = models.Review.objects.create(
                user=self.user, book=self.book, content='hi')
            first_child = models.Status.objects.create(
                reply_parent=parent, user=self.user, content='hi')
            second_child = models.Status.objects.create(
                reply_parent=parent, user=self.user, content='hi')
            third_child = models.Status.objects.create(
                reply_parent=parent, user=self.user,
                deleted=True, deleted_date=timezone.now())

        replies = bookwyrm_tags.get_replies(parent)
        self.assertEqual(len(replies), 2)
        self.assertTrue(first_child in replies)
        self.assertTrue(second_child in replies)
        self.assertFalse(third_child in replies)


    def test_get_parent(self):
        ''' get the reply parent of a status '''
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            parent = models.Review.objects.create(
                user=self.user, book=self.book, content='hi')
            child = models.Status.objects.create(
                reply_parent=parent, user=self.user, content='hi')

        result = bookwyrm_tags.get_parent(child)
        self.assertEqual(result, parent)
        self.assertIsInstance(result, models.Review)


    def test_get_user_liked(self):
        ''' did a user like a status '''
        status = models.Review.objects.create(
            user=self.remote_user, book=self.book)

        self.assertFalse(bookwyrm_tags.get_user_liked(self.user, status))
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            models.Favorite.objects.create(
                user=self.user,
                status=status
            )
        self.assertTrue(bookwyrm_tags.get_user_liked(self.user, status))


    def test_get_user_boosted(self):
        ''' did a user boost a status '''
        status = models.Review.objects.create(
            user=self.remote_user, book=self.book)

        self.assertFalse(bookwyrm_tags.get_user_boosted(self.user, status))
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            models.Boost.objects.create(
                user=self.user,
                boosted_status=status
            )
        self.assertTrue(bookwyrm_tags.get_user_boosted(self.user, status))


    def test_follow_request_exists(self):
        ''' does a user want to follow '''
        self.assertFalse(
            bookwyrm_tags.follow_request_exists(self.user, self.remote_user))

        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            models.UserFollowRequest.objects.create(
                user_subject=self.user,
                user_object=self.remote_user)

        self.assertFalse(
            bookwyrm_tags.follow_request_exists(self.user, self.remote_user))
        self.assertTrue(
            bookwyrm_tags.follow_request_exists(self.remote_user, self.user))


    def test_get_boosted(self):
        ''' load a boosted status '''
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            status = models.Review.objects.create(
                user=self.remote_user, book=self.book)
            boost = models.Boost.objects.create(
                user=self.user,
                boosted_status=status
            )
        boosted = bookwyrm_tags.get_boosted(boost)
        self.assertIsInstance(boosted, models.Review)
        self.assertEqual(boosted, status)


    def test_get_book_description(self):
        ''' grab it from the edition or the parent '''
        work = models.Work.objects.create(title='Test Work')
        self.book.parent_work = work
        self.book.save()

        self.assertIsNone(bookwyrm_tags.get_book_description(self.book))

        work.description = 'hi'
        work.save()
        self.assertEqual(bookwyrm_tags.get_book_description(self.book), 'hi')

        self.book.description = 'hello'
        self.book.save()
        self.assertEqual(bookwyrm_tags.get_book_description(self.book), 'hello')


    def test_get_uuid(self):
        ''' uuid functionality '''
        uuid = bookwyrm_tags.get_uuid('hi')
        self.assertTrue(re.match(r'hi[A-Za-z0-9\-]', uuid))


    def test_time_since(self):
        ''' ultraconcise timestamps '''
        self.assertEqual(bookwyrm_tags.time_since('bleh'), '')

        now = timezone.now()
        self.assertEqual(bookwyrm_tags.time_since(now), '0s')

        seconds_ago = now - relativedelta(seconds=4)
        self.assertEqual(bookwyrm_tags.time_since(seconds_ago), '4s')

        minutes_ago = now - relativedelta(minutes=8)
        self.assertEqual(bookwyrm_tags.time_since(minutes_ago), '8m')

        hours_ago = now - relativedelta(hours=9)
        self.assertEqual(bookwyrm_tags.time_since(hours_ago), '9h')

        days_ago = now - relativedelta(days=3)
        self.assertEqual(bookwyrm_tags.time_since(days_ago), '3d')

        # I am not going to figure out how to mock dates tonight.
        months_ago = now - relativedelta(months=5)
        self.assertTrue(re.match(
            r'[A-Z][a-z]{2} \d?\d',
            bookwyrm_tags.time_since(months_ago)
        ))

        years_ago = now - relativedelta(years=10)
        self.assertTrue(re.match(
            r'[A-Z][a-z]{2} \d?\d \d{4}',
            bookwyrm_tags.time_since(years_ago)
        ))


    def test_get_markdown(self):
        ''' mardown format data '''
        result = bookwyrm_tags.get_markdown('_hi_')
        self.assertEqual(result, '<p><em>hi</em></p>')

        result = bookwyrm_tags.get_markdown('<marquee>_hi_</marquee>')
        self.assertEqual(result, '<p><em>hi</em></p>')


    def test_get_mentions(self):
        ''' list of people mentioned '''
        status = models.Status.objects.create(
            content='hi', user=self.remote_user)
        result = bookwyrm_tags.get_mentions(status, self.user)
        self.assertEqual(result, '@rat@example.com ')


    def test_get_status_preview_name(self):
        ''' status context string '''
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            status = models.Status.objects.create(content='hi', user=self.user)
            result = bookwyrm_tags.get_status_preview_name(status)
            self.assertEqual(result, 'status')

            status = models.Review.objects.create(
                content='hi', user=self.user, book=self.book)
            result = bookwyrm_tags.get_status_preview_name(status)
            self.assertEqual(result, 'review of <em>Test Book</em>')

            status = models.Comment.objects.create(
                content='hi', user=self.user, book=self.book)
            result = bookwyrm_tags.get_status_preview_name(status)
            self.assertEqual(result, 'comment on <em>Test Book</em>')

            status = models.Quotation.objects.create(
                content='hi', user=self.user, book=self.book)
        result = bookwyrm_tags.get_status_preview_name(status)
        self.assertEqual(result, 'quotation from <em>Test Book</em>')


    def test_related_status(self):
        ''' gets the subclass model for a notification status '''
        with patch('bookwyrm.models.activitypub_mixin.broadcast_task.delay'):
            status = models.Status.objects.create(content='hi', user=self.user)
        notification = models.Notification.objects.create(
            user=self.user, notification_type='MENTION',
            related_status=status)

        result = bookwyrm_tags.related_status(notification)
        self.assertIsInstance(result, models.Status)