EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

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

""" Tests for the linecache module """\r
\r
import linecache\r
import unittest\r
import os.path\r
from test import test_support as support\r
\r
\r
FILENAME = linecache.__file__\r
INVALID_NAME = '!@$)(!@#_1'\r
EMPTY = ''\r
TESTS = 'inspect_fodder inspect_fodder2 mapping_tests'\r
TESTS = TESTS.split()\r
TEST_PATH = os.path.dirname(support.__file__)\r
MODULES = "linecache abc".split()\r
MODULE_PATH = os.path.dirname(FILENAME)\r
\r
SOURCE_1 = '''\r
" Docstring "\r
\r
def function():\r
    return result\r
\r
'''\r
\r
SOURCE_2 = '''\r
def f():\r
    return 1 + 1\r
\r
a = f()\r
\r
'''\r
\r
SOURCE_3 = '''\r
def f():\r
    return 3''' # No ending newline\r
\r
\r
class LineCacheTests(unittest.TestCase):\r
\r
    def test_getline(self):\r
        getline = linecache.getline\r
\r
        # Bad values for line number should return an empty string\r
        self.assertEqual(getline(FILENAME, 2**15), EMPTY)\r
        self.assertEqual(getline(FILENAME, -1), EMPTY)\r
\r
        # Float values currently raise TypeError, should it?\r
        self.assertRaises(TypeError, getline, FILENAME, 1.1)\r
\r
        # Bad filenames should return an empty string\r
        self.assertEqual(getline(EMPTY, 1), EMPTY)\r
        self.assertEqual(getline(INVALID_NAME, 1), EMPTY)\r
\r
        # Check whether lines correspond to those from file iteration\r
        for entry in TESTS:\r
            filename = os.path.join(TEST_PATH, entry) + '.py'\r
            for index, line in enumerate(open(filename)):\r
                self.assertEqual(line, getline(filename, index + 1))\r
\r
        # Check module loading\r
        for entry in MODULES:\r
            filename = os.path.join(MODULE_PATH, entry) + '.py'\r
            for index, line in enumerate(open(filename)):\r
                self.assertEqual(line, getline(filename, index + 1))\r
\r
        # Check that bogus data isn't returned (issue #1309567)\r
        empty = linecache.getlines('a/b/c/__init__.py')\r
        self.assertEqual(empty, [])\r
\r
    def test_no_ending_newline(self):\r
        self.addCleanup(support.unlink, support.TESTFN)\r
        with open(support.TESTFN, "w") as fp:\r
            fp.write(SOURCE_3)\r
        lines = linecache.getlines(support.TESTFN)\r
        self.assertEqual(lines, ["\n", "def f():\n", "    return 3\n"])\r
\r
    def test_clearcache(self):\r
        cached = []\r
        for entry in TESTS:\r
            filename = os.path.join(TEST_PATH, entry) + '.py'\r
            cached.append(filename)\r
            linecache.getline(filename, 1)\r
\r
        # Are all files cached?\r
        cached_empty = [fn for fn in cached if fn not in linecache.cache]\r
        self.assertEqual(cached_empty, [])\r
\r
        # Can we clear the cache?\r
        linecache.clearcache()\r
        cached_empty = [fn for fn in cached if fn in linecache.cache]\r
        self.assertEqual(cached_empty, [])\r
\r
    def test_checkcache(self):\r
        getline = linecache.getline\r
        # Create a source file and cache its contents\r
        source_name = support.TESTFN + '.py'\r
        self.addCleanup(support.unlink, source_name)\r
        with open(source_name, 'w') as source:\r
            source.write(SOURCE_1)\r
        getline(source_name, 1)\r
\r
        # Keep a copy of the old contents\r
        source_list = []\r
        with open(source_name) as source:\r
            for index, line in enumerate(source):\r
                self.assertEqual(line, getline(source_name, index + 1))\r
                source_list.append(line)\r
\r
        with open(source_name, 'w') as source:\r
            source.write(SOURCE_2)\r
\r
        # Try to update a bogus cache entry\r
        linecache.checkcache('dummy')\r
\r
        # Check that the cache matches the old contents\r
        for index, line in enumerate(source_list):\r
            self.assertEqual(line, getline(source_name, index + 1))\r
\r
        # Update the cache and check whether it matches the new source file\r
        linecache.checkcache(source_name)\r
        with open(source_name) as source:\r
            for index, line in enumerate(source):\r
                self.assertEqual(line, getline(source_name, index + 1))\r
                source_list.append(line)\r
\r
def test_main():\r
    support.run_unittest(LineCacheTests)\r
\r
if __name__ == "__main__":\r
    test_main()\r