diff --git a/tests/handlers/test_profile.py b/tests/handlers/test_profile.py
index 5330a9b34e..8834d21f0d 100644
--- a/tests/handlers/test_profile.py
+++ b/tests/handlers/test_profile.py
@@ -56,7 +56,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
def test_get_my_name(self):
self.get_success(
- self.store.set_profile_displayname(self.frank.localpart, "Frank")
+ self.store.set_profile_displayname(self.frank.localpart, "Frank", 1)
)
displayname = self.get_success(self.handler.get_displayname(self.frank))
@@ -111,7 +111,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
# Setting displayname for the first time is allowed
self.get_success(
- self.store.set_profile_displayname(self.frank.localpart, "Frank")
+ self.store.set_profile_displayname(self.frank.localpart, "Frank", 1)
)
self.assertEquals(
@@ -156,7 +156,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
def test_incoming_fed_query(self):
self.get_success(self.store.create_profile("caroline"))
- self.get_success(self.store.set_profile_displayname("caroline", "Caroline"))
+ self.get_success(self.store.set_profile_displayname("caroline", "Caroline", 1))
response = self.get_success(
self.query_handlers["profile"](
@@ -173,7 +173,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
def test_get_my_avatar(self):
self.get_success(
self.store.set_profile_avatar_url(
- self.frank.localpart, "http://my.server/me.png"
+ self.frank.localpart, "http://my.server/me.png", 1
)
)
avatar_url = self.get_success(self.handler.get_avatar_url(self.frank))
@@ -227,7 +227,7 @@ class ProfileTestCase(unittest.HomeserverTestCase):
# Setting displayname for the first time is allowed
self.get_success(
self.store.set_profile_avatar_url(
- self.frank.localpart, "http://my.server/me.png"
+ self.frank.localpart, "http://my.server/me.png", 1
)
)
|
