Fix-up test.

matrix-org · Oct 24, 2023 · 36559c8 · 36559c8
1 parent 13783ff
commit 36559c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/storage/test_base.py b/tests/storage/test_base.py
@@ -146,7 +146,7 @@ def test_select_one_missing(
     @defer.inlineCallbacks
     def test_select_list(self) -> Generator["defer.Deferred[object]", object, None]:
         self.mock_txn.rowcount = 3
-        self.mock_txn.__iter__ = Mock(return_value=iter([(1,), (2,), (3,)]))
+        self.mock_txn.fetchall.return_value = [(1,), (2,), (3,)]
         self.mock_txn.description = (("colA", None, None, None, None, None, None),)
 
         ret = yield defer.ensureDeferred(