diff wokkel/test/test_rsm.py @ 32:cfee3d472359

test (RSM): fix according to modification in sat.tmp.wokkel.rsm in rev 1771 (54f834e40341)
author souliane <souliane@mailoo.org>
date Tue, 19 Jan 2016 14:59:13 +0100
parents e79eedd9c56a
children
line wrap: on
line diff
--- a/wokkel/test/test_rsm.py	Tue Jan 19 12:18:29 2016 +0100
+++ b/wokkel/test/test_rsm.py	Tue Jan 19 14:59:13 2016 +0100
@@ -22,6 +22,10 @@
 
 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 @@
         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 @@
         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 @@
         """
         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 @@
         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 @@
         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 @@
             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 @@
         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 @@
             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 @@
         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 @@
             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 @@
         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 @@
             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 @@
         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 @@
         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 @@
 
         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 @@
         """
 
         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 @@
         """
 
         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 @@
         """
 
         def items(request):
-            rsm = RSMResponse(800).toElement()
+            rsm = RSMResponse(count=800).toElement()
             return defer.succeed([rsm])
 
         def cb(element):