yul/bookwyrm-mastodon
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove boost notification on delete

Browse Source
This commit is contained in:
Mouse Reeve
2021-01-10 19:51:58 -08:00
parent ee96c01cc1
commit 62fd118016
3 changed files with 39 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
bookwyrm/tests/test_outgoing.py
View File

@ -688,3 +688,18 @@ class Outgoing(TestCase):
outgoing.handle_boost(self.remote_user, status)
outgoing.handle_boost(self.remote_user, status)
self.assertEqual(models.Boost.objects.count(), 1)
def test_handle_unboost(self):
''' undo a boost '''
status = models.Status.objects.create(
user=self.local_user, content='hi')
with patch('bookwyrm.broadcast.broadcast_task.delay'):
outgoing.handle_boost(self.remote_user, status)
self.assertEqual(models.Boost.objects.count(), 1)
self.assertEqual(models.Notification.objects.count(), 1)
with patch('bookwyrm.broadcast.broadcast_task.delay'):
outgoing.handle_unboost(self.remote_user, status)
self.assertEqual(models.Boost.objects.count(), 0)
self.assertEqual(models.Notification.objects.count(), 0)