maxmustermann/synapse
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Update tests

Browse source
This commit is contained in:
David Robertson 2021-10-11 18:28:08 +01:00
parent 1b84f7dfb0
commit 03db1e551d
No known key found for this signature in database
GPG key ID: 903ECE108A39DEDD
1 changed files with 13 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/handlers/test_user_directory.py
View file

@ -181,14 +181,22 @@ class UserDirectoryTestCase(unittest.HomeserverTestCase):
) )
self._check_only_one_user_in_directory(user, public) self._check_only_one_user_in_directory(user, public)
def test_excludes_appservice_sender(self) -> None: def test_includes_appservice_sender(self) -> None:
user = self.register_user("user", "pass") user = self.register_user("user", "pass")
token = self.login(user, "pass") token = self.login(user, "pass")
room = self.helper.create_room_as( room = self.helper.create_room_as(
self.appservice.sender, is_public=True, tok=self.appservice.token self.appservice.sender, is_public=True, tok=self.appservice.token
) )
self.helper.join(room, user, tok=token) self.helper.join(room, user, tok=token)
self._check_only_one_user_in_directory(user, room) users = self.get_success(self.user_dir_helper.get_users_in_user_directory())
in_public = self.get_success(self.user_dir_helper.get_users_in_public_rooms())
in_private = self.get_success(
self.user_dir_helper.get_users_who_share_private_rooms()
)
self.assertEqual(users, {user, self.appservice.sender})
self.assertEqual(set(in_public), {(user, room), (self.appservice.sender, room)})
self.assertEqual(in_private, [])
def _create_rooms_and_inject_memberships( def _create_rooms_and_inject_memberships(
self, creator: str, token: str, joiner: str self, creator: str, token: str, joiner: str
@ -314,18 +322,18 @@ class UserDirectoryTestCase(unittest.HomeserverTestCase):
self.assertIsNone(profile) self.assertIsNone(profile)
# update profile # update profile
profile_info = ProfileInfo(avatar_url="avatar_url", display_name="4L1c3") profile_info = ProfileInfo(avatar_url="avatar_url", display_name="beep boop")
self.get_success( self.get_success(
self.handler.handle_local_profile_change( self.handler.handle_local_profile_change(
self.appservice.sender, profile_info self.appservice.sender, profile_info
) )
) )
# profile is still not in directory # profile is now set
profile = self.get_success( profile = self.get_success(
self.store.get_user_in_directory(self.appservice.sender) self.store.get_user_in_directory(self.appservice.sender)
) )
self.assertIsNone(profile) self.assertEqual(profile, {"display_name": "beep boop", "avatar_url": "avatar_url"})
def test_handle_user_deactivated_support_user(self) -> None: def test_handle_user_deactivated_support_user(self) -> None:
s_user_id = "@support:test" s_user_id = "@support:test"