Fix iterating over keys when changelog topic is set. (#106)

* check assigned topic name for a changelog * fix testcase to mock property Co-authored-by: alexanderoberegger <alexander.oberegger@smaxtec.com> Co-authored-by: Vikram Patki <54442035+patkivikram@users.noreply.github.com>
faust-streaming · Feb 27, 2021 · 7d29cad · 7d29cad
1 parent 788f310
commit 7d29cad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/faust/stores/rocksdb.py b/faust/stores/rocksdb.py
@@ -453,7 +453,7 @@ def _dbs_for_key(self, key: bytes) -> Iterable[DB]:
 
     def _dbs_for_actives(self) -> Iterator[DB]:
         actives = self.app.assignor.assigned_actives()
-        topic = self.table._changelog_topic_name()
+        topic = self.table.changelog_topic_name
         for partition, db in self._dbs.items():
             tp = TP(topic=topic, partition=partition)
             # for global tables, keys from all

diff --git a/tests/unit/stores/test_rocksdb.py b/tests/unit/stores/test_rocksdb.py
@@ -427,7 +427,7 @@ def test__dbs_for_key(self, *, store):
         assert list(store._dbs_for_key(b"key")) == [dbs[2]]
 
     def test__dbs_for_actives(self, *, store, table):
-        table._changelog_topic_name.return_value = "clog"
+        table.changelog_topic_name = "clog"
         store.app.assignor.assigned_actives = Mock(
             return_value=[
                 TP("clog", 1),