Fixes mocks for tests

This commit is contained in:
Mouse Reeve 2023-02-20 12:23:18 -08:00
parent 0211dee0ff
commit 12ed0f46f3

View file

@ -95,7 +95,8 @@ class Signature(TestCase):
def test_correct_signature(self): def test_correct_signature(self):
"""this one should just work""" """this one should just work"""
response = self.send_test_request(sender=self.mouse) with patch("bookwyrm.models.relationship.UserFollowRequest.accept"):
response = self.send_test_request(sender=self.mouse)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
def test_wrong_signature(self): def test_wrong_signature(self):
@ -124,8 +125,12 @@ class Signature(TestCase):
) )
with patch("bookwyrm.models.user.get_remote_reviews.delay"): with patch("bookwyrm.models.user.get_remote_reviews.delay"):
response = self.send_test_request(sender=self.fake_remote) with patch(
"bookwyrm.models.relationship.UserFollowRequest.accept"
) as accept_mock:
response = self.send_test_request(sender=self.fake_remote)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertTrue(accept_mock.called)
@responses.activate @responses.activate
def test_key_needs_refresh(self): def test_key_needs_refresh(self):
@ -148,16 +153,28 @@ class Signature(TestCase):
with patch("bookwyrm.models.user.get_remote_reviews.delay"): with patch("bookwyrm.models.user.get_remote_reviews.delay"):
# Key correct: # Key correct:
response = self.send_test_request(sender=self.fake_remote) with patch(
"bookwyrm.models.relationship.UserFollowRequest.accept"
) as accept_mock:
response = self.send_test_request(sender=self.fake_remote)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertTrue(accept_mock.called)
# Old key is cached, so still works: # Old key is cached, so still works:
response = self.send_test_request(sender=self.fake_remote) with patch(
"bookwyrm.models.relationship.UserFollowRequest.accept"
) as accept_mock:
response = self.send_test_request(sender=self.fake_remote)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertTrue(accept_mock.called)
# Try with new key: # Try with new key:
response = self.send_test_request(sender=new_sender) with patch(
"bookwyrm.models.relationship.UserFollowRequest.accept"
) as accept_mock:
response = self.send_test_request(sender=new_sender)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertTrue(accept_mock.called)
# Now the old key will fail: # Now the old key will fail:
response = self.send_test_request(sender=self.fake_remote) response = self.send_test_request(sender=self.fake_remote)