diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/suite.py | 12 | ||||
-rw-r--r-- | test/test_debug.py | 41 |
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() |