zedfrigg/bookwyrm
1
0
Fork 0
Code 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)