forked from mirrors/bookwyrm
Fixes peers endpoint
This commit is contained in:
parent
9d886254d8
commit
09e49c70ed
2 changed files with 23 additions and 4 deletions
|
@ -10,7 +10,7 @@ from django.test.client import RequestFactory
|
||||||
from bookwyrm import models, views
|
from bookwyrm import models, views
|
||||||
|
|
||||||
|
|
||||||
class UserViews(TestCase):
|
class WellknownViews(TestCase):
|
||||||
"""view user and edit profile"""
|
"""view user and edit profile"""
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
|
@ -95,3 +95,22 @@ class UserViews(TestCase):
|
||||||
self.assertIsInstance(result, JsonResponse)
|
self.assertIsInstance(result, JsonResponse)
|
||||||
self.assertEqual(data["stats"]["user_count"], 2)
|
self.assertEqual(data["stats"]["user_count"], 2)
|
||||||
self.assertEqual(models.User.objects.count(), 3)
|
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")
|
||||||
|
|
|
@ -95,7 +95,7 @@ def instance_info(_):
|
||||||
status_count = models.Status.objects.filter(user__local=True, deleted=False).count()
|
status_count = models.Status.objects.filter(user__local=True, deleted=False).count()
|
||||||
|
|
||||||
site = models.SiteSettings.get()
|
site = models.SiteSettings.get()
|
||||||
logo_path = site.logo_small or "images/logo-small.png"
|
logo_path = site.logo or "images/logo.png"
|
||||||
logo = f"{MEDIA_FULL_URL}{logo_path}"
|
logo = f"{MEDIA_FULL_URL}{logo_path}"
|
||||||
return JsonResponse(
|
return JsonResponse(
|
||||||
{
|
{
|
||||||
|
@ -120,8 +120,8 @@ def instance_info(_):
|
||||||
@require_GET
|
@require_GET
|
||||||
def peers(_):
|
def peers(_):
|
||||||
"""list of federated servers this instance connects with"""
|
"""list of federated servers this instance connects with"""
|
||||||
names = models.FederatedServer.objects.values_list(
|
names = models.FederatedServer.objects.filter(status="federated").values_list(
|
||||||
"server_name", flat=True, status="federated"
|
"server_name", flat=True
|
||||||
)
|
)
|
||||||
return JsonResponse(list(names), safe=False)
|
return JsonResponse(list(names), safe=False)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue