mirrors/synapse
1
0
Fork 0
mirror of https://github.com/element-hq/synapse.git synced 2024-11-28 05:39:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

Implement and test presence dropping of remote users

Browse source
This commit is contained in:
Paul "LeoNerd" Evans 2014-08-18 16:43:18 +01:00
parent 6fafa878f6
commit 83f031207e
2 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
synapse/handlers/presence.py
View file

@ -463,9 +463,13 @@ class PresenceHandler(BaseHandler):
deferreds = [] deferreds = []
if target_user: if target_user:
raise NotImplementedError("TODO: remove one user") if target_user not in self._remote_recvmap:
return
target_users = set([target_user])
else:
target_users = self._remote_recvmap.keys()
remoteusers = [u for u in self._remote_recvmap remoteusers = [u for u in target_users
if user in self._remote_recvmap[u]] if user in self._remote_recvmap[u]]
remoteusers_by_domain = partition(remoteusers, lambda u: u.domain) remoteusers_by_domain = partition(remoteusers, lambda u: u.domain)

8
tests/handlers/test_presence.py
View file

@ -383,6 +383,14 @@ class PresenceInvitesTestCase(unittest.TestCase):
self.mock_stop.assert_called_with( self.mock_stop.assert_called_with(
self.u_apple, target_user=self.u_banana) self.u_apple, target_user=self.u_banana)
@defer.inlineCallbacks
def test_drop_remote(self):
yield self.handler.drop(
observer_user=self.u_apple, observed_user=self.u_cabbage)
self.datastore.del_presence_list.assert_called_with(
"apple", "@cabbage:elsewhere")
@defer.inlineCallbacks @defer.inlineCallbacks
def test_get_presence_list(self): def test_get_presence_list(self):
self.datastore.get_presence_list.return_value = defer.succeed( self.datastore.get_presence_list.return_value = defer.succeed(