tests/test_stanza_xep_0059.py

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