test (RSM): fix according to modification in sat.tmp.wokkel.rsm in rev 1771 (b77dc676a4df)

This commit is contained in:
souliane 2016-01-19 14:59:13 +01:00
commit b51b9a8887

View file

@ -22,6 +22,10 @@ from sat.tmp.wokkel.rsm import NS_RSM, RSMRequest, RSMResponse, PubSubClient, Pu
import uuid
RSMResponse.__eq__ = lambda self, other: self.first == other.first and\
self.last == other.last and\
self.index == other.index and\
self.count == other.count
class RSMRequestTest(unittest.TestCase):
"""
@ -174,7 +178,7 @@ class RSMRequestTest(unittest.TestCase):
element.addElement('items')['max_items'] = u'10'
RSMRequest(3).render(element)
self.assertEqual(u'3', element.items['max_items']) # changed
self.assertEqual(u'10', element.items['max_items']) # not changed
self.assertEqual(NS_RSM, element.set.uri)
self.assertEqual(u'3', ''.join(element.set.max.children))
@ -187,7 +191,7 @@ class RSMRequestTest(unittest.TestCase):
Embed a page request in the element, specify items.
"""
element = domish.Element(('jabber:iq:search', 'query'))
RSMRequest(5, 127).render(element)
RSMRequest(5, index=127).render(element)
self.assertEqual(NS_RSM, element.set.uri)
self.assertEqual(u'5', ''.join(element.set.max.children))
self.assertIdentical(None, element.set.after)
@ -228,16 +232,8 @@ class RSMResponseTest(unittest.TestCase):
"""
Fail to initialize a RSMResponse with wrong attribute values.
"""
self.assertRaises(AssertionError, RSMResponse, count=u'1')
self.assertRaises(AssertionError, RSMResponse, index=127, first=u'127')
self.assertRaises(AssertionError, RSMResponse, index=127, last=u'351')
self.assertRaises(AssertionError, RSMResponse, first=u'127', last=u'351')
self.assertRaises(AssertionError, RSMResponse, index=u'127',
first=u'127', last=u'351')
self.assertRaises(AssertionError, RSMResponse, index=127,
first=127, last=u'351')
self.assertRaises(AssertionError, RSMResponse, index=127,
first=u'127', last=351)
def test_parse(self):
"""
@ -280,8 +276,7 @@ class RSMResponseTest(unittest.TestCase):
Embed a page response in the element.
"""
element = domish.Element(('jabber:iq:search', 'query'))
RSMResponse(800, 20, u'stpeter@jabber.org',
u'peterpan@neverland.lit').render(element)
RSMResponse(u'stpeter@jabber.org', u'peterpan@neverland.lit', 20, 800).render(element)
self.assertEqual(NS_RSM, element.set.uri)
self.assertEqual(u'800', ''.join(element.set.count.children))
@ -296,7 +291,7 @@ class RSMResponseTest(unittest.TestCase):
Embed a page response in the element, for empty set.
"""
element = domish.Element(('jabber:iq:search', 'query'))
RSMResponse(800).render(element)
RSMResponse(count=800).render(element)
self.assertEqual(NS_RSM, element.set.uri)
self.assertEqual(u'800', ''.join(element.set.count.children))
@ -324,8 +319,7 @@ class PubSubClientTest(unittest.TestCase):
items, rsm = response
self.assertEquals(2, len(items))
self.assertEquals([item1, item2], items)
self.assertDictEqual(rsm, {'count': '800', 'index': '0',
'first': 'item1', 'last': 'item2'})
self.assertEquals(rsm, RSMResponse('item1', 'item2', 0, 800))
d = self.protocol.items(JID('pubsub.example.org'), 'test',
rsm_request=RSMRequest(2))
@ -356,7 +350,7 @@ class PubSubClientTest(unittest.TestCase):
item1['id'] = 'item1'
item2 = items.addElement('item')
item2['id'] = 'item2'
RSMResponse(800, 0, u'item1', u'item2').render(response.pubsub)
RSMResponse(u'item1', u'item2', 0, 800).render(response.pubsub)
self.stub.send(response)
return d
@ -369,8 +363,7 @@ class PubSubClientTest(unittest.TestCase):
items, rsm = response
self.assertEquals(2, len(items))
self.assertEquals([item1, item2], items)
self.assertDictEqual(rsm, {'count': '800', 'index': '2',
'first': 'item3', 'last': 'item4'})
self.assertEquals(rsm, RSMResponse('item3', 'item4', 2, 800))
d = self.protocol.items(JID('pubsub.example.org'), 'test',
rsm_request=RSMRequest(2, after=u'item2'))
@ -402,7 +395,7 @@ class PubSubClientTest(unittest.TestCase):
item1['id'] = 'item3'
item2 = items.addElement('item')
item2['id'] = 'item4'
RSMResponse(800, 2, u'item3', u'item4').render(response.pubsub)
RSMResponse(u'item3', u'item4', 2, 800).render(response.pubsub)
self.stub.send(response)
return d
@ -415,8 +408,7 @@ class PubSubClientTest(unittest.TestCase):
items, rsm = response
self.assertEquals(2, len(items))
self.assertEquals([item1, item2], items)
self.assertDictEqual(rsm, {'count': '800', 'index': '0',
'first': 'item1', 'last': 'item2'})
self.assertEquals(rsm, RSMResponse('item1', 'item2', 0, 800))
d = self.protocol.items(JID('pubsub.example.org'), 'test',
rsm_request=RSMRequest(2, before=u'item3'))
@ -448,7 +440,7 @@ class PubSubClientTest(unittest.TestCase):
item1['id'] = 'item1'
item2 = items.addElement('item')
item2['id'] = 'item2'
RSMResponse(800, 0, u'item1', u'item2').render(response.pubsub)
RSMResponse(u'item1', u'item2', 0, 800).render(response.pubsub)
self.stub.send(response)
return d
@ -461,11 +453,10 @@ class PubSubClientTest(unittest.TestCase):
items, rsm = response
self.assertEquals(3, len(items))
self.assertEquals([item1, item2, item3], items)
self.assertDictEqual(rsm, {'count': '800', 'index': '3',
'first': 'item4', 'last': 'item6'})
self.assertEquals(rsm, RSMResponse('item4', 'item6', 3, 800))
d = self.protocol.items(JID('pubsub.example.org'), 'test',
rsm_response=RSMRequest(3, 3))
rsm_request=RSMRequest(3, index=3))
d.addCallback(cb)
iq = self.stub.output[-1]
@ -496,7 +487,7 @@ class PubSubClientTest(unittest.TestCase):
item2['id'] = 'item5'
item3 = items.addElement('item')
item3['id'] = 'item6'
RSMResponse(800, 3, u'item4', u'item6').render(response.pubsub)
RSMResponse(u'item4', u'item6', 3, 800).render(response.pubsub)
self.stub.send(response)
return d
@ -508,7 +499,7 @@ class PubSubClientTest(unittest.TestCase):
def cb(response):
items, rsm = response
self.assertEquals(0, len(items))
self.assertDictEqual(rsm, {'count': '800'})
self.assertEquals(rsm, RSMResponse(count=800))
d = self.protocol.items(JID('pubsub.example.org'), 'test',
rsm_request=RSMRequest(0))
@ -533,7 +524,7 @@ class PubSubClientTest(unittest.TestCase):
response = toResponse(iq, 'result')
response.addElement((pubsub.NS_PUBSUB, 'pubsub'))
RSMResponse(800).render(response.pubsub)
RSMResponse(count=800).render(response.pubsub)
self.stub.send(response)
return d
@ -564,7 +555,7 @@ class PubSubServiceTest(unittest.TestCase, TestableRequestHandlerMixin):
"""
def items(request):
rsm = RSMResponse(800, 0, u'item', u'item').toElement()
rsm = RSMResponse(u'item', u'item', 0, 800).toElement()
return defer.succeed([pubsub.Item('current'), rsm])
def cb(element):
@ -606,7 +597,7 @@ class PubSubServiceTest(unittest.TestCase, TestableRequestHandlerMixin):
"""
def items(request):
rsm = RSMResponse(800, 3, u'i1', u'i2').toElement()
rsm = RSMResponse(u'i1', u'i2', 3, 800).toElement()
return defer.succeed([pubsub.Item('i1'), pubsub.Item('i2'), rsm])
def cb(element):
@ -652,7 +643,7 @@ class PubSubServiceTest(unittest.TestCase, TestableRequestHandlerMixin):
"""
def items(request):
rsm = RSMResponse(800).toElement()
rsm = RSMResponse(count=800).toElement()
return defer.succeed([rsm])
def cb(element):