Merge branch 'main' into validate-username

bookwyrm/tests/test_view_actions.py

@@ -42,6 +42,9 @@ class ViewActions(TestCase):
             content='Test status',
             remote_id='https://example.com/status/1',
         )
+        self.work = models.Work.objects.create(title='Test Work')
+        self.book = models.Edition.objects.create(
+            title='Test Book', parent_work=self.work)
         self.settings = models.SiteSettings.objects.create(id=1)
         self.factory = RequestFactory()
 
@@ -352,3 +355,43 @@ class ViewActions(TestCase):
         author.refresh_from_db()
         self.assertEqual(author.name, 'Test Author')
         self.assertEqual(resp.template_name, 'edit_author.html')
+
+
+    def test_tag(self):
+        ''' add a tag to a book '''
+        request = self.factory.post(
+            '', {
+                'name': 'A Tag!?',
+                'book': self.book.id,
+            })
+        request.user = self.local_user
+
+        with patch('bookwyrm.broadcast.broadcast_task.delay'):
+            actions.tag(request)
+
+        tag = models.Tag.objects.get()
+        user_tag = models.UserTag.objects.get()
+        self.assertEqual(tag.name, 'A Tag!?')
+        self.assertEqual(tag.identifier, 'A+Tag%21%3F')
+        self.assertEqual(user_tag.user, self.local_user)
+        self.assertEqual(user_tag.book, self.book)
+
+
+    def test_untag(self):
+        ''' remove a tag from a book '''
+        tag = models.Tag.objects.create(name='A Tag!?')
+        user_tag = models.UserTag.objects.create(
+            user=self.local_user, book=self.book, tag=tag)
+        request = self.factory.post(
+            '', {
+                'user': self.local_user.id,
+                'book': self.book.id,
+                'name': tag.name,
+            })
+        request.user = self.local_user
+
+        with patch('bookwyrm.broadcast.broadcast_task.delay'):
+            actions.untag(request)
+
+        self.assertTrue(models.Tag.objects.filter(name='A Tag!?').exists())
+        self.assertFalse(models.UserTag.objects.exists())