diff options
| author | David Robertson <davidr@element.io> | 2023-02-06 13:10:22 +0000 |
|---|---|---|
| committer | David Robertson <davidr@element.io> | 2023-02-06 13:55:08 +0000 |
| commit | 3ad65ac778e73b027cbb2d90da4a30b6a988bc0f (patch) | |
| tree | c8484db4cb4e47edadf2fdba5e791f94ba186747 | |
| parent | Fix force_load_metadata return type (diff) | |
| download | synapse-3ad65ac778e73b027cbb2d90da4a30b6a988bc0f.tar.xz | |
Missing get_success-es
| -rw-r--r-- | tests/storage/test_appservice.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/storage/test_appservice.py b/tests/storage/test_appservice.py index 5e1324a169..2b09700bfe 100644 --- a/tests/storage/test_appservice.py +++ b/tests/storage/test_appservice.py @@ -151,7 +151,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase): outfile.write(yaml.dump(as_yaml)) self.as_yaml_files.append(as_token) - def _set_state(self, id: str, state: ApplicationServiceState) -> defer.Deferred: + def _set_state( + self, id: str, state: ApplicationServiceState + ) -> "defer.Deferred[None]": return self.db_pool.runOperation( self.engine.convert_param_style( "INSERT INTO application_services_state(as_id, state) VALUES(?,?)" @@ -297,7 +299,9 @@ class ApplicationServiceTransactionStoreTestCase(unittest.HomeserverTestCase): service = Mock(id=self.as_list[0]["id"]) events = [Mock(event_id="e1"), Mock(event_id="e2")] txn_id = 5 - self._set_state(self.as_list[0]["id"], ApplicationServiceState.UP) + self.get_success( + self._set_state(self.as_list[0]["id"], ApplicationServiceState.UP) + ) self.get_success(self._insert_txn(service.id, txn_id, events)) self.get_success( self.store.complete_appservice_txn(txn_id=txn_id, service=service) |