Updates models

bookwyrm/tests/models/test_link.py

@@ -0,0 +1,41 @@
+""" 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")