bookwyrm/bookwyrm/tests/models/test_status_model.py

''' testing models '''
from io import BytesIO
import pathlib

from PIL import Image
from django.core.files.base import ContentFile
from django.test import TestCase
from django.utils import timezone

from bookwyrm import models, settings


class Status(TestCase):
    ''' lotta types of statuses '''
    def setUp(self):
        ''' useful things for creating a status '''
        self.user = models.User.objects.create_user(
            'mouse', 'mouse@mouse.mouse', 'mouseword', local=True)
        self.book = models.Edition.objects.create(title='Test Edition')

        image_file = pathlib.Path(__file__).parent.joinpath(
            '../../static/images/default_avi.jpg')
        image = Image.open(image_file)
        output = BytesIO()
        image.save(output, format=image.format)
        self.book.cover.save(
            'test.jpg',
            ContentFile(output.getvalue())
        )

    def test_status_generated_fields(self):
        ''' setting remote id '''
        status = models.Status.objects.create(content='bleh', user=self.user)
        expected_id = 'https://%s/user/mouse/status/%d' % \
                (settings.DOMAIN, status.id)
        self.assertEqual(status.remote_id, expected_id)
        self.assertEqual(status.privacy, 'public')

    def test_replies(self):
        ''' get a list of replies '''
        parent = models.Status.objects.create(content='hi', user=self.user)
        child = models.Status.objects.create(
            content='hello', reply_parent=parent, user=self.user)
        models.Review.objects.create(
            content='hey', reply_parent=parent, user=self.user, book=self.book)
        models.Status.objects.create(
            content='hi hello', reply_parent=child, user=self.user)

        replies = models.Status.replies(parent)
        self.assertEqual(replies.count(), 2)
        self.assertEqual(replies.first(), child)
        # should select subclasses
        self.assertIsInstance(replies.last(), models.Review)

    def test_status_type(self):
        ''' class name '''
        self.assertEqual(models.Status().status_type, 'Note')
        self.assertEqual(models.Review().status_type, 'Review')
        self.assertEqual(models.Quotation().status_type, 'Quotation')
        self.assertEqual(models.Comment().status_type, 'Comment')
        self.assertEqual(models.Boost().status_type, 'Boost')

    def test_boostable(self):
        ''' can a status be boosted, based on privacy '''
        self.assertTrue(models.Status(privacy='public').boostable)
        self.assertTrue(models.Status(privacy='unlisted').boostable)
        self.assertFalse(models.Status(privacy='followers').boostable)
        self.assertFalse(models.Status(privacy='direct').boostable)

    def test_to_replies(self):
        ''' activitypub replies collection '''
        parent = models.Status.objects.create(content='hi', user=self.user)
        child = models.Status.objects.create(
            content='hello', reply_parent=parent, user=self.user)
        models.Review.objects.create(
            content='hey', reply_parent=parent, user=self.user, book=self.book)
        models.Status.objects.create(
            content='hi hello', reply_parent=child, user=self.user)

        replies = parent.to_replies()
        self.assertEqual(replies['id'], '%s/replies' % parent.remote_id)
        self.assertEqual(replies['totalItems'], 2)

    def test_status_to_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.Status.objects.create(
            content='test content', user=self.user)
        activity = status.to_activity()
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'Note')
        self.assertEqual(activity['content'], 'test content')
        self.assertEqual(activity['sensitive'], False)

    def test_status_to_activity_tombstone(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.Status.objects.create(
            content='test content', user=self.user,
            deleted=True, deleted_date=timezone.now())
        activity = status.to_activity()
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'Tombstone')
        self.assertFalse(hasattr(activity, 'content'))

    def test_status_to_pure_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.Status.objects.create(
            content='test content', user=self.user)
        activity = status.to_activity(pure=True)
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'Note')
        self.assertEqual(activity['content'], 'test content')
        self.assertEqual(activity['sensitive'], False)
        self.assertEqual(activity['attachment'], [])

    def test_generated_note_to_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.GeneratedNote.objects.create(
            content='test content', user=self.user)
        status.mention_books.set([self.book])
        status.mention_users.set([self.user])
        activity = status.to_activity()
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'GeneratedNote')
        self.assertEqual(activity['content'], 'test content')
        self.assertEqual(activity['sensitive'], False)
        self.assertEqual(len(activity['tag']), 2)

    def test_generated_note_to_pure_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.GeneratedNote.objects.create(
            content='test content', user=self.user)
        status.mention_books.set([self.book])
        status.mention_users.set([self.user])
        activity = status.to_activity(pure=True)
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(
            activity['content'],
            'mouse test content <a href="%s">"Test Edition"</a>' % \
                self.book.remote_id)
        self.assertEqual(len(activity['tag']), 2)
        self.assertEqual(activity['type'], 'Note')
        self.assertEqual(activity['sensitive'], False)
        self.assertIsInstance(activity['attachment'], list)
        self.assertEqual(activity['attachment'][0].type, 'Image')
        self.assertEqual(activity['attachment'][0].url, 'https://%s%s' % \
                (settings.DOMAIN, self.book.cover.url))
        self.assertEqual(
            activity['attachment'][0].name, 'Test Edition cover')

    def test_comment_to_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.Comment.objects.create(
            content='test content', user=self.user, book=self.book)
        activity = status.to_activity()
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'Comment')
        self.assertEqual(activity['content'], 'test content')
        self.assertEqual(activity['inReplyToBook'], self.book.remote_id)

    def test_comment_to_pure_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.Comment.objects.create(
            content='test content', user=self.user, book=self.book)
        activity = status.to_activity(pure=True)
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'Note')
        self.assertEqual(
            activity['content'],
            '<p>test content</p><p>' \
                    '(comment on <a href="%s">"Test Edition"</a>)</p>' %
            self.book.remote_id)
        self.assertEqual(activity['attachment'][0].type, 'Image')
        self.assertEqual(activity['attachment'][0].url, 'https://%s%s' % \
                (settings.DOMAIN, self.book.cover.url))
        self.assertEqual(
            activity['attachment'][0].name, 'Test Edition cover')

    def test_quotation_to_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.Quotation.objects.create(
            quote='a sickening sense', content='test content',
            user=self.user, book=self.book)
        activity = status.to_activity()
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'Quotation')
        self.assertEqual(activity['quote'], 'a sickening sense')
        self.assertEqual(activity['content'], 'test content')
        self.assertEqual(activity['inReplyToBook'], self.book.remote_id)

    def test_quotation_to_pure_activity(self):
        ''' subclass of the base model version with a "pure" serializer '''
        status = models.Quotation.objects.create(
            quote='a sickening sense', content='test content',
            user=self.user, book=self.book)
        activity = status.to_activity(pure=True)
        self.assertEqual(activity['id'], status.remote_id)
        self.assertEqual(activity['type'], 'Note')
        self.assertEqual(
            activity['content'],
            '<p>"a sickening sense"<br>-- <a href="%s">"Test Edition"</a></p>' \
                    '<p>test content</p>' % self.book.remote_id)
        self.assertEqual(activity['attachment'][0].type, 'Image')
        self.assertEqual(activity['attachment'][0].url, 'https://%s%s' % \
                (settings.DOMAIN, self.book.cover.url))
        self.assertEqual(
            activity['attachment'][0].name, 'Test Edition cover')