diff --git a/synapse/storage/databases/main/profile.py b/synapse/storage/databases/main/profile.py index 12f984d433..4503c49997 100644 --- a/synapse/storage/databases/main/profile.py +++ b/synapse/storage/databases/main/profile.py @@ -99,9 +99,12 @@ class ProfileWorkerStore(SQLBaseStore): else: raise - async def create_profile(self, user_localpart: str) -> None: + async def create_profile(self, user_id: str) -> None: + user_localpart = UserID.from_string(user_id).localpart await self.db_pool.simple_insert( - table="profiles", values={"user_id": user_localpart}, desc="create_profile" + table="profiles", + values={"user_id": user_localpart, "full_user_id": user_id}, + desc="create_profile", ) async def set_profile_displayname( diff --git a/tests/handlers/test_profile.py b/tests/handlers/test_profile.py index 2cf3fd2119..9fb2a1ea8a 100644 --- a/tests/handlers/test_profile.py +++ b/tests/handlers/test_profile.py @@ -166,7 +166,7 @@ class ProfileTestCase(unittest.HomeserverTestCase): ) def test_incoming_fed_query(self) -> None: - self.get_success(self.store.create_profile("caroline")) + self.get_success(self.store.create_profile("@caroline:test")) self.get_success(self.store.set_profile_displayname("caroline", "Caroline")) response = self.get_success( diff --git a/tests/storage/test_main.py b/tests/storage/test_main.py index 5806cb0e4b..f83c1e6d62 100644 --- a/tests/storage/test_main.py +++ b/tests/storage/test_main.py @@ -29,7 +29,7 @@ class DataStoreTestCase(unittest.HomeserverTestCase): 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.create_profile(self.user.to_string())) self.get_success( self.store.set_profile_displayname(self.user.localpart, self.displayname) ) diff --git a/tests/storage/test_profile.py b/tests/storage/test_profile.py index 136352f838..4af733d6ba 100644 --- a/tests/storage/test_profile.py +++ b/tests/storage/test_profile.py @@ -27,7 +27,7 @@ class ProfileStoreTestCase(unittest.HomeserverTestCase): self.u_frank = UserID.from_string("@frank:test") def test_displayname(self) -> None: - self.get_success(self.store.create_profile(self.u_frank.localpart)) + self.get_success(self.store.create_profile(self.u_frank.to_string())) self.get_success( self.store.set_profile_displayname(self.u_frank.localpart, "Frank") @@ -54,7 +54,7 @@ class ProfileStoreTestCase(unittest.HomeserverTestCase): ) def test_avatar_url(self) -> None: - self.get_success(self.store.create_profile(self.u_frank.localpart)) + self.get_success(self.store.create_profile(self.u_frank.to_string())) self.get_success( self.store.set_profile_avatar_url(