Updates tests

This commit is contained in:
Mouse Reeve
2021-08-04 18:22:06 -07:00
parent 9d75bc3982
commit 9d19092205
4 changed files with 18 additions and 10 deletions

View File

@ -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