From 9d19092205bb2b0db2e120455e11edb563db9d80 Mon Sep 17 00:00:00 2001
From: Mouse Reeve <mousereeve@riseup.net>
Date: Wed, 4 Aug 2021 18:22:06 -0700
Subject: [PATCH] Updates tests

---
 bookwyrm/activitystreams.py                    | 18 ++++++++++++------
 .../tests/management/test_populate_streams.py  |  2 +-
 bookwyrm/tests/views/test_feed.py              |  2 +-
 bookwyrm/views/feed.py                         |  6 ++++--
 4 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/bookwyrm/activitystreams.py b/bookwyrm/activitystreams.py
index 2f155538..0a966971 100644
--- a/bookwyrm/activitystreams.py
+++ b/bookwyrm/activitystreams.py
@@ -7,6 +7,7 @@ from bookwyrm.redis_store import RedisStore, r
 from bookwyrm.settings import STREAMS
 from bookwyrm.views.helpers import privacy_filter
 
+
 class ActivityStream(RedisStore):
     """a category of activity stream (like home, local, federated)"""
 
@@ -219,12 +220,17 @@ class BooksStream(ActivityStream):
 
 
 available_streams = [s["key"] for s in STREAMS]
-streams = {k:v for (k, v) in {
-    "home": HomeStream(),
-    "local": LocalStream(),
-    "federated": FederatedStream(),
-    "books": BooksStream(),
-}.items() if k in available_streams}
+streams = {
+    k: v
+    for (k, v) in {
+        "home": HomeStream(),
+        "local": LocalStream(),
+        "federated": FederatedStream(),
+        "books": BooksStream(),
+    }.items()
+    if k in available_streams
+}
+
 
 @receiver(signals.post_save)
 # pylint: disable=unused-argument
diff --git a/bookwyrm/tests/management/test_populate_streams.py b/bookwyrm/tests/management/test_populate_streams.py
index ee7a96d7..c080e3fe 100644
--- a/bookwyrm/tests/management/test_populate_streams.py
+++ b/bookwyrm/tests/management/test_populate_streams.py
@@ -46,4 +46,4 @@ class Activitystreams(TestCase):
             "bookwyrm.activitystreams.ActivityStream.populate_store"
         ) as redis_mock:
             populate_streams()
-        self.assertEqual(redis_mock.call_count, 6)  # 2 users x 3 streams
+        self.assertEqual(redis_mock.call_count, 4)  # 2 users x 2 streams
diff --git a/bookwyrm/tests/views/test_feed.py b/bookwyrm/tests/views/test_feed.py
index 8a38b808..666c3e02 100644
--- a/bookwyrm/tests/views/test_feed.py
+++ b/bookwyrm/tests/views/test_feed.py
@@ -45,7 +45,7 @@ class FeedViews(TestCase):
         view = views.Feed.as_view()
         request = self.factory.get("")
         request.user = self.local_user
-        result = view(request, "local")
+        result = view(request, "home")
         self.assertIsInstance(result, TemplateResponse)
         result.render()
         self.assertEqual(result.status_code, 200)
diff --git a/bookwyrm/views/feed.py b/bookwyrm/views/feed.py
index d28166a3..7a46ca57 100644
--- a/bookwyrm/views/feed.py
+++ b/bookwyrm/views/feed.py
@@ -24,9 +24,11 @@ class Feed(View):
     def get(self, request, tab):
         """user's homepage with activity feed"""
         tab = [s for s in STREAMS if s["key"] == tab]
-        tab = tab[0] or STREAMS[0]
+        tab = tab[0] if tab else STREAMS[0]
 
-        activities = activitystreams.streams[tab["key"]].get_activity_stream(request.user)
+        activities = activitystreams.streams[tab["key"]].get_activity_stream(
+            request.user
+        )
         paginated = Paginator(activities, PAGE_LENGTH)
 
         suggestions = suggested_users.get_suggestions(request.user)