def test_remove_nsfw_collection_srnames_on_frontpage(self, get_nsfw_collections_srnames):
get_nsfw_collections_srnames.return_value = set(nsfw_collection.sr_names)
srname = "test1"
subreddit = Subreddit(name=srname)
Subreddit.user_subreddits = MagicMock(return_value=[
Subreddit(name=nice_srname),
Subreddit(name=questionably_nsfw),
])
frontpage_srnames = srnames_from_site(self.logged_in, Frontpage)
swf_srnames = srnames_from_site(self.logged_in, subreddit)
self.assertEqual(frontpage_srnames, {Frontpage.name, nice_srname})
self.assertTrue(len(frontpage_srnames & {questionably_nsfw}) == 0)
def test_quarantined_subscriptions_are_never_included(self, user_subreddits):
user_subreddits.return_value = naughty_subscriptions
subreddit = Frontpage
srnames = srnames_from_site(self.logged_in, subreddit)
self.assertEqual(srnames, {subreddit.name} | {nice_srname})
self.assertTrue(len(srnames & {quarantined_srname}) == 0)
def test_multi_logged_in(self, user_subreddits):
user_subreddits.return_value = subscriptions
multi = MultiReddit(path="/user/test/m/multi_test",
srs=multi_subreddits)
srnames = srnames_from_site(self.logged_in, multi)
self.assertEqual(srnames, set(multi_srnames))
def test_nsfw_subscriptions_arent_included_when_viewing_frontpage(
self, user_subreddits):
user_subreddits.return_value = naughty_subscriptions
srnames = srnames_from_site(self.logged_in, Frontpage)
self.assertEqual(srnames, {Frontpage.name} | {nice_srname})
self.assertTrue(len(srnames & {nsfw_srname}) == 0)
def test_subreddit_logged_in(self, user_subreddits):
user_subreddits.return_value = subscriptions
srname = "test1"
subreddit = Subreddit(name=srname)
srnames = srnames_from_site(self.logged_in, subreddit)
self.assertEqual(srnames, {srname})
def test_nsfw_subscriptions_arent_included_when_viewing_sfw(self, user_subreddits):
user_subreddits.return_value = naughty_subscriptions
srname = "test1"
subreddit = Subreddit(name=srname)
srnames = srnames_from_site(self.logged_in, subreddit)
self.assertEqual(srnames, {srname} | {nice_srname})
self.assertTrue(len(srnames & {nsfw_srname}) == 0)
def test_nsfw_subscriptions_arent_included_when_viewing_sfw(
self, user_subreddits):
user_subreddits.return_value = naughty_subscriptions
srname = "test1"
subreddit = Subreddit(name=srname)
srnames = srnames_from_site(self.logged_in, subreddit)
self.assertEqual(srnames, {srname} | {nice_srname})
self.assertTrue(len(srnames & {nsfw_srname}) == 0)
def test_only_nsfw_subscriptions_are_included_when_viewing_nswf(self, user_subreddits):
user_subreddits.return_value = naughty_subscriptions
srname = "bad"
subreddit = Subreddit(name=srname, over_18=True)
srnames = srnames_from_site(self.logged_in, subreddit)
self.assertEqual(srnames, {srname} | {nsfw_srname})
self.assertTrue(len(srnames & {nsfw_srname}) == 1)
self.assertTrue(len(srnames & {nice_srname}) == 0)
def test_only_nsfw_subscriptions_are_included_when_viewing_nswf(
self, user_subreddits):
user_subreddits.return_value = naughty_subscriptions
srname = "bad"
subreddit = Subreddit(name=srname, over_18=True)
srnames = srnames_from_site(self.logged_in, subreddit)
self.assertEqual(srnames, {srname} | {nsfw_srname})
self.assertTrue(len(srnames & {nsfw_srname}) == 1)
self.assertTrue(len(srnames & {nice_srname}) == 0)
def test_remove_swf_subscriptions_when_viewing_sr_in_nsfw_collection(self, get_nsfw_collections_srnames):
get_nsfw_collections_srnames.return_value = set(nsfw_collection.sr_names)
questionable_subreddit = Subreddit(name=questionably_nsfw)
Subreddit.user_subreddits = MagicMock(
return_value=[Subreddit(name=nice_srname), Subreddit(name=questionably_nsfw)]
)
srnames = srnames_from_site(self.logged_in, questionable_subreddit)
self.assertEqual(srnames, {questionably_nsfw})
self.assertTrue(len(srnames & {nice_srname}) == 0)
def test_remove_swf_subscriptions_when_viewing_sr_in_nsfw_collection(
self, get_nsfw_collections_srnames):
get_nsfw_collections_srnames.return_value = set(
nsfw_collection.sr_names)
questionable_subreddit = Subreddit(name=questionably_nsfw)
Subreddit.user_subreddits = MagicMock(return_value=[
Subreddit(name=nice_srname),
Subreddit(name=questionably_nsfw),
])
srnames = srnames_from_site(self.logged_in, questionable_subreddit)
self.assertEqual(srnames, {questionably_nsfw})
self.assertTrue(len(srnames & {nice_srname}) == 0)
def test_nsfw_subscriptions_arent_included_when_viewing_frontpage(self, user_subreddits):
user_subreddits.return_value = naughty_subscriptions
srnames = srnames_from_site(self.logged_in, Frontpage)
self.assertEqual(srnames, {Frontpage.name} | {nice_srname})
self.assertTrue(len(srnames & {nsfw_srname}) == 0)
def test_subreddit_logged_out(self):
srname = "test1"
subreddit = Subreddit(name=srname)
srnames = srnames_from_site(self.logged_out, subreddit)
self.assertEqual(srnames, {srname})
def test_multi_logged_in(self, user_subreddits):
user_subreddits.return_value = subscriptions
multi = MultiReddit(path="/user/test/m/multi_test", srs=multi_subreddits)
srnames = srnames_from_site(self.logged_in, multi)
self.assertEqual(srnames, set(multi_srnames))
def test_multi_logged_out(self):
multi = MultiReddit(path="/user/test/m/multi_test", srs=multi_subreddits)
srnames = srnames_from_site(self.logged_out, multi)
self.assertEqual(srnames, set(multi_srnames))
def test_frontpage_logged_in(self, user_subreddits):
user_subreddits.return_value = subscriptions
srnames = srnames_from_site(self.logged_in, Frontpage)
self.assertEqual(srnames, set(subscriptions_srnames) | {Frontpage.name})
def test_frontpage_logged_out(self):
srnames = srnames_from_site(self.logged_out, Frontpage)
self.assertEqual(srnames, {Frontpage.name})
