Clean up tests.storage.test_main to remove use of legacy code. (#11493)

This commit is contained in:
reivilibre 2021-12-02 18:13:43 +00:00 committed by GitHub
parent 8a4c296987
commit 16d39a5490
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 17 deletions

1
changelog.d/11493.misc Normal file
View File

@ -0,0 +1 @@
Clean up `tests.storage.test_main` to remove use of legacy code.

View File

@ -13,42 +13,35 @@
# limitations under the License.
from twisted.internet import defer
from synapse.types import UserID
from tests import unittest
from tests.utils import setup_test_homeserver
class DataStoreTestCase(unittest.TestCase):
@defer.inlineCallbacks
def setUp(self):
hs = yield setup_test_homeserver(self.addCleanup)
class DataStoreTestCase(unittest.HomeserverTestCase):
def setUp(self) -> None:
super(DataStoreTestCase, self).setUp()
self.store = hs.get_datastore()
self.store = self.hs.get_datastore()
self.user = UserID.from_string("@abcde:test")
self.displayname = "Frank"
@defer.inlineCallbacks
def test_get_users_paginate(self):
yield defer.ensureDeferred(
self.store.register_user(self.user.to_string(), "pass")
)
yield defer.ensureDeferred(self.store.create_profile(self.user.localpart))
yield defer.ensureDeferred(
def test_get_users_paginate(self) -> None:
self.get_success(self.store.register_user(self.user.to_string(), "pass"))
self.get_success(self.store.create_profile(self.user.localpart))
self.get_success(
self.store.set_profile_displayname(self.user.localpart, self.displayname)
)
users, total = yield defer.ensureDeferred(
users, total = self.get_success(
self.store.get_users_paginate(0, 10, name="bc", guests=False)
)
self.assertEquals(1, total)
self.assertEquals(self.displayname, users.pop()["displayname"])
users, total = yield defer.ensureDeferred(
users, total = self.get_success(
self.store.get_users_paginate(0, 10, name="BC", guests=False)
)