from django.test import TestCase from fedireads import models, broadcast from fedireads.settings import DOMAIN class Book(TestCase): def setUp(self): self.user = models.User.objects.create_user( 'mouse', 'mouse@mouse.mouse', 'mouseword') follower = models.User.objects.create_user( 'rat', 'rat@mouse.mouse', 'ratword', local=False, actor='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, actor='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', actor='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', fedireads_user=False, local=False) self.user.followers.add(non_fr_follower) local_follower = models.User.objects.create_user( 'joe', 'joe@mouse.mouse', 'jeoword') self.user.followers.add(local_follower) models.User.objects.create_user( 'nutria', 'nutria@mouse.mouse', 'nuword', actor='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='fedireads') 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)