EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Lib / test / test___all__.py

from __future__ import print_function\r
\r
import unittest\r
from test import test_support as support\r
import os\r
import sys\r
\r
# Setup bsddb warnings\r
try:\r
    bsddb = support.import_module('bsddb', deprecated=True)\r
except unittest.SkipTest:\r
    pass\r
\r
\r
class NoAll(RuntimeError):\r
    pass\r
\r
class FailedImport(RuntimeError):\r
    pass\r
\r
\r
class AllTest(unittest.TestCase):\r
\r
    def check_all(self, modname):\r
        names = {}\r
        with support.check_warnings((".* (module|package)",\r
                                     DeprecationWarning), quiet=True):\r
            try:\r
                exec "import %s" % modname in names\r
            except:\r
                # Silent fail here seems the best route since some modules\r
                # may not be available or not initialize properly in all\r
                # environments.\r
                raise FailedImport(modname)\r
        if not hasattr(sys.modules[modname], "__all__"):\r
            raise NoAll(modname)\r
        names = {}\r
        try:\r
            exec "from %s import *" % modname in names\r
        except Exception as e:\r
            # Include the module name in the exception string\r
            self.fail("__all__ failure in {}: {}: {}".format(\r
                      modname, e.__class__.__name__, e))\r
        if "__builtins__" in names:\r
            del names["__builtins__"]\r
        keys = set(names)\r
        all = set(sys.modules[modname].__all__)\r
        self.assertEqual(keys, all)\r
\r
    def walk_modules(self, basedir, modpath):\r
        for fn in sorted(os.listdir(basedir)):\r
            path = os.path.join(basedir, fn)\r
            if os.path.isdir(path):\r
                pkg_init = os.path.join(path, '__init__.py')\r
                if os.path.exists(pkg_init):\r
                    yield pkg_init, modpath + fn\r
                    for p, m in self.walk_modules(path, modpath + fn + "."):\r
                        yield p, m\r
                continue\r
            if not fn.endswith('.py') or fn == '__init__.py':\r
                continue\r
            yield path, modpath + fn[:-3]\r
\r
    def test_all(self):\r
        # Blacklisted modules and packages\r
        blacklist = set([\r
            # Will raise a SyntaxError when compiling the exec statement\r
            '__future__',\r
        ])\r
\r
        if not sys.platform.startswith('java'):\r
            # In case _socket fails to build, make this test fail more gracefully\r
            # than an AttributeError somewhere deep in CGIHTTPServer.\r
            import _socket\r
\r
        # rlcompleter needs special consideration; it import readline which\r
        # initializes GNU readline which calls setlocale(LC_CTYPE, "")... :-(\r
        try:\r
            import rlcompleter\r
            import locale\r
        except ImportError:\r
            pass\r
        else:\r
            locale.setlocale(locale.LC_CTYPE, 'C')\r
\r
        ignored = []\r
        failed_imports = []\r
        lib_dir = os.path.dirname(os.path.dirname(__file__))\r
        for path, modname in self.walk_modules(lib_dir, ""):\r
            m = modname\r
            blacklisted = False\r
            while m:\r
                if m in blacklist:\r
                    blacklisted = True\r
                    break\r
                m = m.rpartition('.')[0]\r
            if blacklisted:\r
                continue\r
            if support.verbose:\r
                print(modname)\r
            try:\r
                # This heuristic speeds up the process by removing, de facto,\r
                # most test modules (and avoiding the auto-executing ones).\r
                with open(path, "rb") as f:\r
                    if "__all__" not in f.read():\r
                        raise NoAll(modname)\r
                    self.check_all(modname)\r
            except NoAll:\r
                ignored.append(modname)\r
            except FailedImport:\r
                failed_imports.append(modname)\r
\r
        if support.verbose:\r
            print('Following modules have no __all__ and have been ignored:',\r
                  ignored)\r
            print('Following modules failed to be imported:', failed_imports)\r
\r
\r
def test_main():\r
    support.run_unittest(AllTest)\r
\r
if __name__ == "__main__":\r
    test_main()\r