summary refs log tree commit diff
path: root/tests/storage/test_base.py
diff options
context:
space:
mode:
authorErik Johnston <erik@matrix.org>2015-04-08 16:57:14 +0100
committerErik Johnston <erik@matrix.org>2015-04-08 16:57:14 +0100
commit22d7a593062eb06081d0cd1dc71e4fc79fc1cb85 (patch)
tree7b9dc27f4d68e8722894426acfaa30dea8b72d10 /tests/storage/test_base.py
parentUse generic db exceptions rather than sqlite3 specific ones (diff)
downloadsynapse-22d7a593062eb06081d0cd1dc71e4fc79fc1cb85.tar.xz
Fix tests after commit 9a0579
Diffstat (limited to 'tests/storage/test_base.py')
-rw-r--r--tests/storage/test_base.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/storage/test_base.py b/tests/storage/test_base.py
index 5c17d30148..a64d2b821e 100644
--- a/tests/storage/test_base.py
+++ b/tests/storage/test_base.py
@@ -33,12 +33,18 @@ class SQLBaseStoreTestCase(unittest.TestCase):
     def setUp(self):
         self.db_pool = Mock(spec=["runInteraction"])
         self.mock_txn = Mock()
+        self.mock_conn = Mock(spec_set=["cursor"])
+        self.mock_conn.cursor.return_value = self.mock_txn
         # Our fake runInteraction just runs synchronously inline
 
         def runInteraction(func, *args, **kwargs):
             return defer.succeed(func(self.mock_txn, *args, **kwargs))
         self.db_pool.runInteraction = runInteraction
 
+        def runWithConnection(func, *args, **kwargs):
+            return defer.succeed(func(self.mock_conn, *args, **kwargs))
+        self.db_pool.runWithConnection = runWithConnection
+
         config = Mock()
         config.event_cache_size = 1
         hs = HomeServer(