zulip-bots: Yield cache storage instead bot storage.

Fixes the bug that the context manager doesn't actually manage the storage.
2021-05-26 23:09:35 +08:00 · 2021-05-26 23:09:35 +08:00 · bd27631dd1
parent 06bbfd752e
commit bd27631dd1
1 changed files with 1 additions and 1 deletions
--- a/zulip_bots/zulip_bots/lib.py
+++ b/zulip_bots/zulip_bots/lib.py
@ -164,7 +164,7 @@ def use_storage(storage: BotStorage, keys: List[Text]) -> Iterator[BotStorage]:
    # calling flush or getting some values that are not previously fetched.
    data = {key: storage.get(key) for key in keys}
    cache = CachedStorage(storage, data)
-    yield storage
+    yield cache
    cache.flush()

 class BotHandler(Protocol):