diff options
Diffstat (limited to 'tests/test_stream_xep_0059.py')
| -rw-r--r-- | tests/test_stream_xep_0059.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/test_stream_xep_0059.py b/tests/test_stream_xep_0059.py index 3a99842b..5f3ea079 100644 --- a/tests/test_stream_xep_0059.py +++ b/tests/test_stream_xep_0059.py @@ -1,6 +1,7 @@ import threading -from sleekxmpp.test import * +import unittest +from sleekxmpp.test import SleekTest from sleekxmpp.xmlstream import register_stanza_plugin from sleekxmpp.plugins.xep_0030 import DiscoItems from sleekxmpp.plugins.xep_0059 import ResultIterator, Set @@ -17,7 +18,7 @@ class TestStreamSet(SleekTest): def iter(self, rev=False): q = self.xmpp.Iq() q['type'] = 'get' - it = ResultIterator(q, 'disco_items', '1', reverse=rev) + it = ResultIterator(q, 'disco_items', amount='1', reverse=rev) for i in it: for j in i['disco_items']['items']: self.items.append(j[0]) |