From f2bf6072ec447cf5cc331b4057b29078b1c932b5 Mon Sep 17 00:00:00 2001
From: Robert Robinson <rerobins@gmail.com>
Date: Mon, 14 Sep 2015 16:35:47 -0600
Subject: Add plugin (cherry picked from commit 2296d56)

---
 tests/test_stanza_xep_0122.py | 196 +++++++++++++++++++++++-------------------
 1 file changed, 108 insertions(+), 88 deletions(-)

(limited to 'tests/test_stanza_xep_0122.py')

diff --git a/tests/test_stanza_xep_0122.py b/tests/test_stanza_xep_0122.py
index 6e1e4d7b..5576c45f 100644
--- a/tests/test_stanza_xep_0122.py
+++ b/tests/test_stanza_xep_0122.py
@@ -1,8 +1,7 @@
 import unittest
+
 from sleekxmpp import Message
 from sleekxmpp.test import SleekTest
-from sleekxmpp.thirdparty import OrderedDict
-
 import sleekxmpp.plugins.xep_0004 as xep_0004
 import sleekxmpp.plugins.xep_0122 as xep_0122
 from sleekxmpp.xmlstream import register_stanza_plugin
@@ -46,92 +45,113 @@ class TestDataForms(SleekTest):
           </message>
         """)
 
-    # def test_open_validation(self):
-    #     """Testing using multiple instructions elements in a data form."""
-    #     msg = self.Message()
-    #     form = msg['form']
-    #     field = form.addField(var='f1',
-    #                           ftype='text-single',
-    #                           label='Text',
-    #                           desc='A text field',
-    #                           required=True,
-    #                           value='Some text!')
-    #
-    #     validation = field['validate']
-    #     validation.set_open(True)
-    #
-    #     self.check(msg, """
-    #       <message>
-    #         <x xmlns="jabber:x:data" type="form">
-    #           <field var="f1" type="text-single" label="Text">
-    #             <desc>A text field</desc>
-    #             <required />
-    #             <value>Some text!</value>
-    #             <validate xmlns="http://jabber.org/protocol/xdata-validate">
-    #                 <open/>
-    #             </validate>
-    #           </field>
-    #         </x>
-    #       </message>
-    #     """)
-    #
-    # def test_regex_validation(self):
-    #     """Testing using multiple instructions elements in a data form."""
-    #     msg = self.Message()
-    #     form = msg['form']
-    #     field = form.addField(var='f1',
-    #                           ftype='text-single',
-    #                           label='Text',
-    #                           desc='A text field',
-    #                           required=True,
-    #                           value='Some text!')
-    #
-    #     validation = field['validate']
-    #     validation.set_regex('[0-9]+')
-    #
-    #     self.check(msg, """
-    #       <message>
-    #         <x xmlns="jabber:x:data" type="form">
-    #           <field var="f1" type="text-single" label="Text">
-    #             <desc>A text field</desc>
-    #             <required />
-    #             <value>Some text!</value>
-    #             <validate xmlns="http://jabber.org/protocol/xdata-validate">
-    #                 <regex>[0-9]+</regex>
-    #             </validate>
-    #           </field>
-    #         </x>
-    #       </message>
-    #     """)
-    #
-    # def test_range_validation(self):
-    #     """Testing using multiple instructions elements in a data form."""
-    #     msg = self.Message()
-    #     form = msg['form']
-    #     field = form.addField(var='f1',
-    #                           ftype='text-single',
-    #                           label='Text',
-    #                           desc='A text field',
-    #                           required=True,
-    #                           value='Some text!')
-    #
-    #     validation = field['validate']
-    #     validation.set_range(True, minimum=0, maximum=10)
-    #
-    #     self.check(msg, """
-    #       <message>
-    #         <x xmlns="jabber:x:data" type="form">
-    #           <field var="f1" type="text-single" label="Text">
-    #             <desc>A text field</desc>
-    #             <required />
-    #             <value>Some text!</value>
-    #             <validate xmlns="http://jabber.org/protocol/xdata-validate">
-    #                 <range min="0" max="10" />
-    #             </validate>
-    #           </field>
-    #         </x>
-    #       </message>
-    #     """)
+        self.assertTrue(validation.get_basic())
+        self.assertFalse(validation.get_open())
+        self.assertFalse(validation.get_range())
+        self.assertFalse(validation.get_regex())
+
+    def test_open_validation(self):
+        """Testing using multiple instructions elements in a data form."""
+        msg = self.Message()
+        form = msg['form']
+        field = form.addField(var='f1',
+                              ftype='text-single',
+                              label='Text',
+                              desc='A text field',
+                              required=True,
+                              value='Some text!')
+
+        validation = field['validate']
+        validation.set_open(True)
+
+        self.check(msg, """
+          <message>
+            <x xmlns="jabber:x:data" type="form">
+              <field var="f1" type="text-single" label="Text">
+                <desc>A text field</desc>
+                <required />
+                <value>Some text!</value>
+                <validate xmlns="http://jabber.org/protocol/xdata-validate">
+                    <open />
+                </validate>
+              </field>
+            </x>
+          </message>
+        """)
+
+        self.assertFalse(validation.get_basic())
+        self.assertTrue(validation.get_open())
+        self.assertFalse(validation.get_range())
+        self.assertFalse(validation.get_regex())
+
+    def test_regex_validation(self):
+        """Testing using multiple instructions elements in a data form."""
+        msg = self.Message()
+        form = msg['form']
+        field = form.addField(var='f1',
+                              ftype='text-single',
+                              label='Text',
+                              desc='A text field',
+                              required=True,
+                              value='Some text!')
+
+        regex_value = '[0-9]+'
+
+        validation = field['validate']
+        validation.set_regex(regex_value)
+
+        self.check(msg, """
+          <message>
+            <x xmlns="jabber:x:data" type="form">
+              <field var="f1" type="text-single" label="Text">
+                <desc>A text field</desc>
+                <required />
+                <value>Some text!</value>
+                <validate xmlns="http://jabber.org/protocol/xdata-validate">
+                    <regex>[0-9]+</regex>
+                </validate>
+              </field>
+            </x>
+          </message>
+        """)
+
+        self.assertFalse(validation.get_basic())
+        self.assertFalse(validation.get_open())
+        self.assertFalse(validation.get_range())
+        self.assertTrue(validation.get_regex())
+
+        self.assertEqual(regex_value, validation.get_regex())
+
+    def test_range_validation(self):
+        """Testing using multiple instructions elements in a data form."""
+        msg = self.Message()
+        form = msg['form']
+        field = form.addField(var='f1',
+                              ftype='text-single',
+                              label='Text',
+                              desc='A text field',
+                              required=True,
+                              value='Some text!')
+
+        validation = field['validate']
+        validation.set_range(True, minimum=0, maximum=10)
+
+        self.check(msg, """
+          <message>
+            <x xmlns="jabber:x:data" type="form">
+              <field var="f1" type="text-single" label="Text">
+                <desc>A text field</desc>
+                <required />
+                <value>Some text!</value>
+                <validate xmlns="http://jabber.org/protocol/xdata-validate">
+                    <range min="0" max="10" />
+                </validate>
+              </field>
+            </x>
+          </message>
+        """)
+
+        self.assertDictEqual(dict(minimum=str(0), maximum=str(10)), validation.get_range())
 
 
 suite = unittest.TestLoader().loadTestsFromTestCase(TestDataForms)
-- 
cgit v1.2.3