diff options
| author | Erik Johnston <erik@matrix.org> | 2019-12-06 13:08:40 +0000 |
|---|---|---|
| committer | Erik Johnston <erik@matrix.org> | 2019-12-06 13:30:06 +0000 |
| commit | 9a4fb457cf5918c85068ea249cd2d58b3e2e3cfc (patch) | |
| tree | 4bb85c953cb523732bcc53ec01b7c1f8209ac1ad /tests/storage | |
| parent | Merge pull request #6469 from matrix-org/erikj/make_database_class (diff) | |
| download | synapse-9a4fb457cf5918c85068ea249cd2d58b3e2e3cfc.tar.xz | |
Change DataStores to accept 'database' param.
Diffstat (limited to 'tests/storage')
| -rw-r--r-- | tests/storage/test_appservice.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/storage/test_appservice.py b/tests/storage/test_appservice.py index dfeea24599..1679112d82 100644 --- a/tests/storage/test_appservice.py +++ b/tests/storage/test_appservice.py @@ -28,6 +28,7 @@ from synapse.storage.data_stores.main.appservice import ( ApplicationServiceStore, ApplicationServiceTransactionStore, ) +from synapse.storage.database import Database from tests import unittest from tests.utils import setup_test_homeserver @@ -382,8 +383,8 @@ class ApplicationServiceTransactionStoreTestCase(unittest.TestCase): # required for ApplicationServiceTransactionStoreTestCase tests class TestTransactionStore(ApplicationServiceTransactionStore, ApplicationServiceStore): - def __init__(self, db_conn, hs): - super(TestTransactionStore, self).__init__(db_conn, hs) + def __init__(self, database: Database, db_conn, hs): + super(TestTransactionStore, self).__init__(database, db_conn, hs) class ApplicationServiceStoreConfigTestCase(unittest.TestCase): |