forked from mirrors/bookwyrm
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
|
""" testing models """
|
||
|
|
from unittest.mock import patch
|
||
|
|
from django.test import TestCase
|
||
|
|
|
||
|
|
from bookwyrm import models
|
||
|
|
|
||
|
|
|
||
|
|
@patch("bookwyrm.models.activitypub_mixin.broadcast_task.apply_async")
|
||
|
|
class Link(TestCase):
|
||
|
|
"""some activitypub oddness ahead"""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""look, a list"""
|
||
|
|
with patch("bookwyrm.suggested_users.rerank_suggestions_task.delay"), patch(
|
||
|
|
"bookwyrm.activitystreams.populate_stream_task.delay"
|
||
|
|
), patch("bookwyrm.lists_stream.populate_lists_task.delay"):
|
||
|
|
self.local_user = models.User.objects.create_user(
|
||
|
|
"mouse", "mouse@mouse.mouse", "mouseword", local=True, localname="mouse"
|
||
|
|
)
|
||
|
|
work = models.Work.objects.create(title="hello")
|
||
|
|
self.book = models.Edition.objects.create(title="hi", parent_work=work)
|
||
|
|
|
||
|
|
def test_create_domain(self, _):
|
||
|
|
"""generated default name"""
|
||
|
|
domain = models.LinkDomain.objects.create(domain="beep.com")
|
||
|
|
self.assertEqual(domain.name, "beep.com")
|
||
|
|
self.assertEqual(domain.status, "pending")
|
||
|
|
|
||
|
|
def test_create_link_new_domain(self, _):
|
||
|
|
"""generates link and sets domain"""
|
||
|
|
link = models.Link.objects.create(url="https://www.hello.com/hi-there")
|
||
|
|
self.assertEqual(link.domain.domain, "www.hello.com")
|
||
|
|
self.assertEqual(link.name, "www.hello.com")
|
||
|
|
|
||
|
|
def test_create_link_existing_domain(self, _):
|
||
|
|
"""generate link with a known domain"""
|
||
|
|
domain = models.LinkDomain.objects.create(domain="www.hello.com", name="Hi")
|
||
|
|
|
||
|
|
link = models.Link.objects.create(url="https://www.hello.com/hi-there")
|
||
|
|
self.assertEqual(link.domain, domain)
|
||
|
|
self.assertEqual(link.name, "Hi")
|