diff options
Diffstat (limited to 'tests/storage/test_profile.py')
| -rw-r--r-- | tests/storage/test_profile.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/storage/test_profile.py b/tests/storage/test_profile.py index bca056b295..45c69dafae 100644 --- a/tests/storage/test_profile.py +++ b/tests/storage/test_profile.py @@ -53,11 +53,14 @@ class SQLiteMemoryDbPool(ConnectionPool, object): class ProfileStoreTestCase(unittest.TestCase): + + @defer.inlineCallbacks def setUp(self): hs = HomeServer("test", db_pool=SQLiteMemoryDbPool(), ) - hs.get_db_pool().prepare() + + yield hs.get_db_pool().prepare() self.store = ProfileStore(hs) |