From e02ffe854745c6c39127967f5d1dcaff915d7190 Mon Sep 17 00:00:00 2001 From: Lance Stout Date: Thu, 7 Oct 2010 19:42:28 -0400 Subject: [PATCH] Corrected test errors. There was a bug in the XML compare method. --- sleekxmpp/stanza/presence.py | 12 +++++++++++- sleekxmpp/xmlstream/stanzabase.py | 4 +++- tests/__init__.py | 0 tests/test_stanza_element.py | 6 +++--- tests/test_stanza_error.py | 8 +++++--- tests/test_stanza_gmail.py | 2 +- 6 files changed, 23 insertions(+), 9 deletions(-) create mode 100644 tests/__init__.py diff --git a/sleekxmpp/stanza/presence.py b/sleekxmpp/stanza/presence.py index 651bf34d..7eb0c5b1 100644 --- a/sleekxmpp/stanza/presence.py +++ b/sleekxmpp/stanza/presence.py @@ -78,7 +78,9 @@ class Presence(RootStanza): Arguments: show -- Must be one of: away, chat, dnd, or xa. """ - if show in self.showtypes: + if show is None: + self._delSub('show') + elif show in self.showtypes: self._setSubText('show', text=show) return self @@ -99,6 +101,14 @@ class Presence(RootStanza): self['show'] = value return self + def delType(self): + """ + Remove both the type attribute and the element. + """ + self._delAttr('type') + self._delSub('show') + + def setPriority(self, value): """ Set the entity's priority value. Some server use priority to diff --git a/sleekxmpp/xmlstream/stanzabase.py b/sleekxmpp/xmlstream/stanzabase.py index f4a7e506..6851333f 100644 --- a/sleekxmpp/xmlstream/stanzabase.py +++ b/sleekxmpp/xmlstream/stanzabase.py @@ -423,7 +423,9 @@ class ElementBase(object): self._delAttr(attrib) elif attrib in self.plugin_attrib_map: if attrib in self.plugins: + xml = self.plugins[attrib].xml del self.plugins[attrib] + self.xml.remove(xml) return self def _setAttr(self, name, value): @@ -511,7 +513,7 @@ class ElementBase(object): element = self.xml.find(name) if not text and not keep: - return self.__delitem__(name) + return self._delSub(name) if element is None: # We need to add the element. If the provided name was diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/tests/test_stanza_element.py b/tests/test_stanza_element.py index c157e0a8..2a774d6c 100644 --- a/tests/test_stanza_element.py +++ b/tests/test_stanza_element.py @@ -607,7 +607,7 @@ class TestElementBase(SleekTest): - """) + """, use_values=False) self.failUnless(len(stanza) == 1, "Incorrect stanza size with 1 substanza.") @@ -617,7 +617,7 @@ class TestElementBase(SleekTest): - """) + """, use_values=False) self.failUnless(len(stanza) == 2, "Incorrect stanza size with 2 substanzas.") @@ -627,7 +627,7 @@ class TestElementBase(SleekTest): - """) + """, use_values=False) # Test iterating over substanzas stanza.append(substanza1) diff --git a/tests/test_stanza_error.py b/tests/test_stanza_error.py index f76ac62d..d5ed6f5f 100644 --- a/tests/test_stanza_error.py +++ b/tests/test_stanza_error.py @@ -30,11 +30,13 @@ class TestErrorStanzas(SleekTest): self.failUnless(msg['error']['condition'] == 'item-not-found', "Error condition doesn't match.") - del msg['error']['condition'] + msg['error']['condition'] = 'resource-constraint' self.check_message(msg, """ - + + + """) @@ -52,6 +54,6 @@ class TestErrorStanzas(SleekTest): Error! - """) + """, use_values=False) suite = unittest.TestLoader().loadTestsFromTestCase(TestErrorStanzas) diff --git a/tests/test_stanza_gmail.py b/tests/test_stanza_gmail.py index 5c31a4e2..a5630e62 100644 --- a/tests/test_stanza_gmail.py +++ b/tests/test_stanza_gmail.py @@ -25,7 +25,7 @@ class TestGmail(SleekTest): newer-than-tid="11134623426430234" q="is:starred" /> - """) + """, use_values=False) def testMailBox(self): """Testing reading from Gmail mailbox result"""