Changeset 171:4f5ee7b81ebf for wokkel/test/test_generic.py

Timestamp:

May 9, 2012, 2:24:28 PM (10 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Message:

Add convenience method Request.parseRequest.

The new method Request.parseRequest is called from Request.fromElement
with the child element of the iq that is passed to fromElement, providing
a convenient hook for parsing the payload of incoming requests.

File:

• wokkel/test/test_generic.py (modified) (2 diffs)

wokkel/test/test_generic.py

@@ -91 +91 @@
 
 
+class StanzaTest(unittest.TestCase):
+    """
+    Tests for L{generic.Stanza}.
+    """
+
+    def test_fromElement(self):
+        xml = """
+        <message type='chat' from='other@example.org' to='user@example.org'/>
+        """
+
+        stanza = generic.Stanza.fromElement(generic.parseXml(xml))
+        self.assertEqual('chat', stanza.stanzaType)
+        self.assertEqual(JID('other@example.org'), stanza.sender)
+        self.assertEqual(JID('user@example.org'), stanza.recipient)
+
+
+    def test_fromElementChildParser(self):
+        """
+        Child elements for which no parser is defined are ignored.
+        """
+        xml = """
+        <message from='other@example.org' to='user@example.org'>
+          <x xmlns='http://example.org/'/>
+        </message>
+        """
+
+        class Message(generic.Stanza):
+            childParsers = {('http://example.org/', 'x'): '_childParser_x'}
+            elements = []
+
+            def _childParser_x(self, element):
+                self.elements.append(element)
+
+        message = Message.fromElement(generic.parseXml(xml))
+        self.assertEqual(1, len(message.elements))
+
+
+    def test_fromElementChildParserAll(self):
+        """
+        Child elements for which no parser is defined are ignored.
+        """
+        xml = """
+        <message from='other@example.org' to='user@example.org'>
+          <x xmlns='http://example.org/'/>
+        </message>
+        """
+
+        class Message(generic.Stanza):
+            childParsers = {None: '_childParser'}
+            elements = []
+
+            def _childParser(self, element):
+                self.elements.append(element)
+
+        message = Message.fromElement(generic.parseXml(xml))
+        self.assertEqual(1, len(message.elements))
+
+
+    def test_fromElementChildParserUnknown(self):
+        """
+        Child elements for which no parser is defined are ignored.
+        """
+        xml = """
+        <message from='other@example.org' to='user@example.org'>
+          <x xmlns='http://example.org/'/>
+        </message>
+        """
+        generic.Stanza.fromElement(generic.parseXml(xml))
+
+
+
+
 class RequestTest(unittest.TestCase):
     """
@@ -100 +172 @@
 
 
+    def test_requestParser(self):
+        """
+        The request's child element is passed to requestParser.
+        """
+        xml = """
+        <iq type='get'>
+          <query xmlns='jabber:iq:version'/>
+        </iq>
+        """
+
+        class VersionRequest(generic.Request):
+            elements = []
+
+            def parseRequest(self, element):
+                self.elements.append((element.uri, element.name))
+
+        request = VersionRequest.fromElement(generic.parseXml(xml))
+        self.assertEqual([(NS_VERSION, 'query')], request.elements)
+
+
     def test_toElementStanzaKind(self):
         """