Fixes mocks in tests

This commit is contained in:
Mouse Reeve 2023-01-25 09:32:45 -08:00
parent b89cab1ee5
commit 9fdcc7debd
2 changed files with 4 additions and 2 deletions

View file

@ -35,6 +35,7 @@ Sender = namedtuple("Sender", ("remote_id", "key_pair"))
class Signature(TestCase): class Signature(TestCase):
"""signature test""" """signature test"""
# pylint: disable=invalid-name
def setUp(self): def setUp(self):
"""create users and test data""" """create users and test data"""
with patch("bookwyrm.suggested_users.rerank_suggestions_task.delay"), patch( with patch("bookwyrm.suggested_users.rerank_suggestions_task.delay"), patch(
@ -86,7 +87,7 @@ class Signature(TestCase):
data = json.dumps(get_follow_activity(sender, self.rat)) data = json.dumps(get_follow_activity(sender, self.rat))
digest = digest or make_digest(data) digest = digest or make_digest(data)
signature = make_signature(signer or sender, self.rat.inbox, now, digest) signature = make_signature(signer or sender, self.rat.inbox, now, digest)
with patch("bookwyrm.views.inbox.activity_task.delay"): with patch("bookwyrm.views.inbox.activity_task.apply_async"):
with patch("bookwyrm.models.user.set_remote_server.delay"): with patch("bookwyrm.models.user.set_remote_server.delay"):
return self.send(signature, now, send_data or data, digest) return self.send(signature, now, send_data or data, digest)

View file

@ -15,6 +15,7 @@ from bookwyrm import models, views
class Inbox(TestCase): class Inbox(TestCase):
"""readthrough tests""" """readthrough tests"""
# pylint: disable=invalid-name
def setUp(self): def setUp(self):
"""basic user and book data""" """basic user and book data"""
self.client = Client() self.client = Client()
@ -119,7 +120,7 @@ class Inbox(TestCase):
with patch("bookwyrm.views.inbox.has_valid_signature") as mock_valid: with patch("bookwyrm.views.inbox.has_valid_signature") as mock_valid:
mock_valid.return_value = True mock_valid.return_value = True
with patch("bookwyrm.views.inbox.activity_task.delay"): with patch("bookwyrm.views.inbox.activity_task.apply_async"):
result = self.client.post( result = self.client.post(
"/inbox", json.dumps(activity), content_type="application/json" "/inbox", json.dumps(activity), content_type="application/json"
) )