diff src/test/helpers.py @ 1412:979210da778a

test: fix the tests
author souliane <souliane@mailoo.org>
date Fri, 17 Apr 2015 19:06:39 +0200
parents 069ad98b360d
children 7e749e8eefd0
line wrap: on
line diff
--- a/src/test/helpers.py	Fri Apr 17 19:05:37 2015 +0200
+++ b/src/test/helpers.py	Fri Apr 17 19:06:39 2015 +0200
@@ -339,6 +339,7 @@
     def __init__(self, host, parent):
         SatRosterProtocol.__init__(self, host)
         self.parent = parent
+        self._jids = {}
         self.addItem(parent.jid)
 
     def addItem(self, jid, *args, **kwargs):
@@ -351,7 +352,8 @@
         if roster_item.name:
             attrs['name'] = roster_item.name
         self.host.bridge.expectCall("newContact", jid.full(), attrs, roster_item.groups, self.parent.profile)
-        self.onRosterSet(roster_item)
+        self._jids[jid] = roster_item
+        self._registerItem(roster_item)
 
 
 class FakeXmlStream(object):