rudimentary debug tests added

2 files changed, 48 insertions, 5 deletions

diff --git a/test/suite.py b/test/suite.py
index 2242316..7633c0e 100644
--- a/test/suite.py
+++ b/test/suite.py
@@ -4,6 +4,8 @@
 # Copyright (c) 2005-2016, Ilya Etingof <ilya@glas.net>
 # License: http://pyasn1.sf.net/license.html
 #
+import test_debug
+
 from sys import path, version_info
 from os.path import sep
 
@@ -23,12 +25,12 @@ else:
     import unittest
 
 suite = unittest.TestSuite()
-for m in (
-        type.suite,
-        codec.suite
-):
-    suite.addTest(getattr(m, 'suite'))
 
+loader = unittest.TestLoader()
+suite.addTest(loader.loadTestsFromModule(test_debug))
+
+for m in type.suite, codec.suite:
+    suite.addTest(getattr(m, 'suite'))
 
 def runTests():
     unittest.TextTestRunner(verbosity=2).run(suite)
diff --git a/test/test_debug.py b/test/test_debug.py
new file mode 100644
index 0000000..b6bbb6c
--- /dev/null
+++ b/test/test_debug.py
@@ -0,0 +1,41 @@
+#
+# This file is part of pyasn1 software.
+#
+# Copyright (c) 2005-2016, Ilya Etingof <ilya@glas.net>
+# License: http://pyasn1.sf.net/license.html
+#
+from pyasn1 import debug
+from pyasn1 import error
+from sys import version_info
+
+if version_info[0:2] < (2, 7) or \
+        version_info[0:2] in ((3, 0), (3, 1)):
+    try:
+        import unittest2 as unittest
+    except ImportError:
+        import unittest
+else:
+    import unittest
+
+
+class DebugCaseBase(unittest.TestCase):
+    def testKnownFlags(self):
+        debug.setLogger(debug.Debug('all', 'encoder', 'decoder'))
+        debug.setLogger(0)
+
+    def testUnknownFlags(self):
+        try:
+            debug.setLogger(debug.Debug('all', 'unknown', loggerName='xxx'))
+
+        except error.PyAsn1Error:
+            debug.setLogger(0)
+            return
+
+        else:
+            debug.setLogger(0)
+            assert 0, 'unknown debug flag tolerated'
+
+
+
+if __name__ == '__main__':
+    unittest.main()