示例#1
0
class TestProfileAllStream(SocialhomeTestCase):
    @classmethod
    def setUpTestData(cls):
        super().setUpTestData()
        cls.create_local_and_remote_user()
        cls.create_content_set()
        cls.remote_profile_content = PublicContentFactory(author=cls.remote_profile)

    def setUp(self):
        super().setUp()
        self.stream = ProfileAllStream(user=self.user, profile=self.remote_profile)

    def test_get_content_ids__uses_cached_ids(self):
        with patch.object(self.stream, "get_cached_content_ids", return_value=([], {})) as mock_cached:
            self.stream.get_content_ids()
            mock_cached.assert_called_once_with()

    def test_get_queryset(self):
        qs = self.stream.get_queryset()
        self.assertEqual(set(qs), {self.remote_profile_content})

    def test_get_target_streams(self):
        self.assertEqual(
            len(ProfileAllStream.get_target_streams(self.public_content, self.user, self.public_content.author)), 1,
        )

    def test_key(self):
        self.assertEqual(self.stream.key, "sh:streams:profile_all:%s:%s" % (self.remote_profile.id, self.user.id))
示例#2
0
 def get_content(self):
     stream = ProfileAllStream(
         last_id=self.last_id,
         profile=self.profile,
         user=self.request.user,
         accept_ids=self.accept_ids,
     )
     return stream.get_content()
示例#3
0
 def test_get_target_streams(self):
     self.assertEqual(
         len(
             ProfileAllStream.get_target_streams(
                 self.public_content, self.user,
                 self.public_content.author)),
         1,
     )
示例#4
0
 def setUp(self):
     super().setUp()
     self.stream = ProfileAllStream(user=self.user,
                                    profile=self.remote_profile)
示例#5
0
 def test_get_target_streams(self):
     self.assertEqual(
         len(ProfileAllStream.get_target_streams(self.public_content, self.user, self.public_content.author)), 1,
     )
示例#6
0
 def setUp(self):
     super().setUp()
     self.stream = ProfileAllStream(user=self.user, profile=self.remote_profile)