Fixes references to federation views

This commit is contained in:
Mouse Reeve 2021-09-11 07:27:45 -07:00
parent b26e486888
commit c2ccef20a6

View file

@ -75,7 +75,7 @@ class FederationViews(TestCase):
self.assertEqual(server.status, "federated") self.assertEqual(server.status, "federated")
view = views.federation.block_server view = views.block_server
request = self.factory.post("") request = self.factory.post("")
request.user = self.local_user request.user = self.local_user
request.user.is_superuser = True request.user.is_superuser = True
@ -121,7 +121,7 @@ class FederationViews(TestCase):
request.user.is_superuser = True request.user.is_superuser = True
with patch("bookwyrm.suggested_users.bulk_add_instance_task.delay") as mock: with patch("bookwyrm.suggested_users.bulk_add_instance_task.delay") as mock:
views.federation.unblock_server(request, server.id) views.unblock_server(request, server.id)
self.assertEqual(mock.call_count, 1) self.assertEqual(mock.call_count, 1)
self.assertEqual(mock.call_args[0][0], server.id) self.assertEqual(mock.call_args[0][0], server.id)