forked from mirrors/bookwyrm
78 lines
2.9 KiB
Python
78 lines
2.9 KiB
Python
from unittest.mock import patch
|
|
from django.test import TestCase
|
|
|
|
from bookwyrm import models, broadcast
|
|
|
|
|
|
class Book(TestCase):
|
|
def setUp(self):
|
|
self.user = models.User.objects.create_user(
|
|
'mouse', 'mouse@mouse.mouse', 'mouseword', local=True)
|
|
|
|
local_follower = models.User.objects.create_user(
|
|
'joe', 'joe@mouse.mouse', 'jeoword', local=True)
|
|
self.user.followers.add(local_follower)
|
|
|
|
with patch('bookwyrm.models.user.set_remote_server.delay'):
|
|
follower = models.User.objects.create_user(
|
|
'rat', 'rat@mouse.mouse', 'ratword', local=False,
|
|
remote_id='http://example.com/u/1',
|
|
outbox='http://example.com/u/1/o',
|
|
shared_inbox='http://example.com/inbox',
|
|
inbox='http://example.com/u/1/inbox')
|
|
self.user.followers.add(follower)
|
|
|
|
no_inbox_follower = models.User.objects.create_user(
|
|
'hamster', 'hamster@mouse.mouse', 'hamword',
|
|
shared_inbox=None, local=False,
|
|
remote_id='http://example.com/u/2',
|
|
outbox='http://example.com/u/2/o',
|
|
inbox='http://example.com/u/2/inbox')
|
|
self.user.followers.add(no_inbox_follower)
|
|
|
|
non_fr_follower = models.User.objects.create_user(
|
|
'gerbil', 'gerb@mouse.mouse', 'gerbword',
|
|
remote_id='http://example.com/u/3',
|
|
outbox='http://example2.com/u/3/o',
|
|
inbox='http://example2.com/u/3/inbox',
|
|
shared_inbox='http://example2.com/inbox',
|
|
bookwyrm_user=False, local=False)
|
|
self.user.followers.add(non_fr_follower)
|
|
|
|
models.User.objects.create_user(
|
|
'nutria', 'nutria@mouse.mouse', 'nuword',
|
|
remote_id='http://example.com/u/4',
|
|
outbox='http://example.com/u/4/o',
|
|
shared_inbox='http://example.com/inbox',
|
|
inbox='http://example.com/u/4/inbox',
|
|
local=False)
|
|
|
|
|
|
def test_get_public_recipients(self):
|
|
expected = [
|
|
'http://example2.com/inbox',
|
|
'http://example.com/inbox',
|
|
'http://example.com/u/2/inbox',
|
|
]
|
|
|
|
recipients = broadcast.get_public_recipients(self.user)
|
|
self.assertEqual(recipients, expected)
|
|
|
|
|
|
def test_get_public_recipients_software(self):
|
|
expected = [
|
|
'http://example.com/inbox',
|
|
'http://example.com/u/2/inbox',
|
|
]
|
|
|
|
recipients = broadcast.get_public_recipients(self.user, software='bookwyrm')
|
|
self.assertEqual(recipients, expected)
|
|
|
|
|
|
def test_get_public_recipients_software_other(self):
|
|
expected = [
|
|
'http://example2.com/inbox',
|
|
]
|
|
|
|
recipients = broadcast.get_public_recipients(self.user, software='mastodon')
|
|
self.assertEqual(recipients, expected)
|