Fixes peers endpoint

This commit is contained in:
Mouse Reeve
2021-09-11 13:30:27 -07:00
parent 9d886254d8
commit 09e49c70ed
2 changed files with 23 additions and 4 deletions

View File

@ -10,7 +10,7 @@ from django.test.client import RequestFactory
from bookwyrm import models, views
class UserViews(TestCase):
class WellknownViews(TestCase):
"""view user and edit profile"""
def setUp(self):
@ -95,3 +95,22 @@ class UserViews(TestCase):
self.assertIsInstance(result, JsonResponse)
self.assertEqual(data["stats"]["user_count"], 2)
self.assertEqual(models.User.objects.count(), 3)
def test_peers(self):
"""who's federating with whom"""
models.FederatedServer.objects.create(
server_name="test.server",
status="federated",
)
models.FederatedServer.objects.create(
server_name="another.test.server",
status="blocked",
)
request = self.factory.get("")
request.user = self.anonymous_user
result = views.peers(request)
data = json.loads(result.getvalue())
self.assertIsInstance(result, JsonResponse)
self.assertEqual(len(data), 1)
self.assertEqual(data[0], "test.server")