Spaces:
Paused
Paused
| # -*- coding: utf-8 -*- | |
| # | |
| # SelfTest/__init__.py: Self-test for PyCrypto | |
| # | |
| # Written in 2008 by Dwayne C. Litzenberger <dlitz@dlitz.net> | |
| # | |
| # =================================================================== | |
| # The contents of this file are dedicated to the public domain. To | |
| # the extent that dedication to the public domain is not available, | |
| # everyone is granted a worldwide, perpetual, royalty-free, | |
| # non-exclusive license to exercise all rights associated with the | |
| # contents of this file for any purpose whatsoever. | |
| # No rights are reserved. | |
| # | |
| # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | |
| # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | |
| # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND | |
| # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS | |
| # BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN | |
| # ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN | |
| # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
| # SOFTWARE. | |
| # =================================================================== | |
| """Self tests | |
| These tests should perform quickly and can ideally be used every time an | |
| application runs. | |
| """ | |
| import sys | |
| import unittest | |
| from importlib import import_module | |
| from Crypto.Util.py3compat import StringIO | |
| class SelfTestError(Exception): | |
| def __init__(self, message, result): | |
| Exception.__init__(self, message, result) | |
| self.message = message | |
| self.result = result | |
| def run(module=None, verbosity=0, stream=None, tests=None, config=None, **kwargs): | |
| """Execute self-tests. | |
| This raises SelfTestError if any test is unsuccessful. | |
| You may optionally pass in a sub-module of SelfTest if you only want to | |
| perform some of the tests. For example, the following would test only the | |
| hash modules: | |
| Crypto.SelfTest.run(Crypto.SelfTest.Hash) | |
| """ | |
| if config is None: | |
| config = {} | |
| suite = unittest.TestSuite() | |
| if module is None: | |
| if tests is None: | |
| tests = get_tests(config=config) | |
| suite.addTests(tests) | |
| else: | |
| if tests is None: | |
| suite.addTests(module.get_tests(config=config)) | |
| else: | |
| raise ValueError("'module' and 'tests' arguments are mutually exclusive") | |
| if stream is None: | |
| kwargs['stream'] = StringIO() | |
| else: | |
| kwargs['stream'] = stream | |
| runner = unittest.TextTestRunner(verbosity=verbosity, **kwargs) | |
| result = runner.run(suite) | |
| if not result.wasSuccessful(): | |
| if stream is None: | |
| sys.stderr.write(kwargs['stream'].getvalue()) | |
| raise SelfTestError("Self-test failed", result) | |
| return result | |
| def get_tests(config={}): | |
| tests = [] | |
| module_names = [ | |
| "Cipher", "Hash", "Protocol", "PublicKey", "Random", | |
| "Util", "Signature", "IO", "Math", | |
| ] | |
| for name in module_names: | |
| module = import_module("Crypto.SelfTest." + name) | |
| tests += module.get_tests(config=config) | |
| return tests | |
| if __name__ == '__main__': | |
| def suite(): | |
| return unittest.TestSuite(get_tests()) | |
| unittest.main(defaultTest='suite') | |
| # vim:set ts=4 sw=4 sts=4 expandtab: | |