tests/test_stanza_xep_0059.py

   1 from sleekxmpp.test import *
   2 from sleekxmpp.plugins.xep_0059 import Set
   3
   4
   5 class TestSetStanzas(SleekTest):
   6
   7     def testSetFirstIndex(self):
   8         s = Set()
   9         s['first'] = 'id'
  10         s.set_first_index('10')
  11         self.check(s, """
  12           <set xmlns="http://jabber.org/protocol/rsm">
  13             <first index="10">id</first>
  14           </set>
  15         """)
  16
  17     def testGetFirstIndex(self):
  18         xml_string = """
  19           <set xmlns="http://jabber.org/protocol/rsm">
  20             <first index="10">id</first>
  21           </set>
  22         """
  23         s = Set(ET.fromstring(xml_string))
  24         expected = '10'
  25         self.failUnless(s['first_index'] == expected)
  26
  27     def testDelFirstIndex(self):
  28         xml_string = """
  29           <set xmlns="http://jabber.org/protocol/rsm">
  30             <first index="10">id</first>
  31           </set>
  32         """
  33         s = Set(ET.fromstring(xml_string))
  34         del s['first_index']
  35         self.check(s, """
  36           <set xmlns="http://jabber.org/protocol/rsm">
  37             <first>id</first>
  38           </set>
  39         """)
  40
  41     def testSetBefore(self):
  42         s = Set()
  43         s['before'] = True
  44         self.check(s, """
  45           <set xmlns="http://jabber.org/protocol/rsm">
  46             <before />
  47           </set>
  48         """)
  49
  50     def testGetBefore(self):
  51         xml_string = """
  52           <set xmlns="http://jabber.org/protocol/rsm">
  53             <before />
  54           </set>
  55         """
  56         s = Set(ET.fromstring(xml_string))
  57         expected = True
  58         self.failUnless(s['before'] == expected)
  59
  60     def testGetBefore(self):
  61         xml_string = """
  62           <set xmlns="http://jabber.org/protocol/rsm">
  63             <before />
  64           </set>
  65         """
  66         s = Set(ET.fromstring(xml_string))
  67         del s['before']
  68         self.check(s, """
  69           <set xmlns="http://jabber.org/protocol/rsm">
  70           </set>
  71         """)
  72
  73     def testSetBeforeVal(self):
  74         s = Set()
  75         s['before'] = 'id'
  76         self.check(s, """
  77           <set xmlns="http://jabber.org/protocol/rsm">
  78             <before>id</before>
  79           </set>
  80         """)
  81
  82     def testGetBeforeVal(self):
  83         xml_string = """
  84           <set xmlns="http://jabber.org/protocol/rsm">
  85             <before>id</before>
  86           </set>
  87         """
  88         s = Set(ET.fromstring(xml_string))
  89         expected = 'id'
  90         self.failUnless(s['before'] == expected)
  91
  92     def testGetBeforeVal(self):
  93         xml_string = """
  94           <set xmlns="http://jabber.org/protocol/rsm">
  95             <before>id</before>
  96           </set>
  97         """
  98         s = Set(ET.fromstring(xml_string))
  99         del s['before']
 100         self.check(s, """
 101           <set xmlns="http://jabber.org/protocol/rsm">
 102           </set>
 103         """)
 104
 105
 106 suite = unittest.TestLoader().loadTestsFromTestCase(TestSetStanzas)