EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

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

# Tests of the full ZIP64 functionality of zipfile\r
# The test_support.requires call is the only reason for keeping this separate\r
# from test_zipfile\r
from test import test_support\r
\r
# XXX(nnorwitz): disable this test by looking for extra largfile resource\r
# which doesn't exist.  This test takes over 30 minutes to run in general\r
# and requires more disk space than most of the buildbots.\r
test_support.requires(\r
        'extralargefile',\r
        'test requires loads of disk-space bytes and a long time to run'\r
    )\r
\r
# We can test part of the module without zlib.\r
try:\r
    import zlib\r
except ImportError:\r
    zlib = None\r
\r
import zipfile, os, unittest\r
import time\r
import sys\r
\r
from tempfile import TemporaryFile\r
\r
from test.test_support import TESTFN, run_unittest\r
\r
TESTFN2 = TESTFN + "2"\r
\r
# How much time in seconds can pass before we print a 'Still working' message.\r
_PRINT_WORKING_MSG_INTERVAL = 5 * 60\r
\r
class TestsWithSourceFile(unittest.TestCase):\r
    def setUp(self):\r
        # Create test data.\r
        # xrange() is important here -- don't want to create immortal space\r
        # for a million ints.\r
        line_gen = ("Test of zipfile line %d." % i for i in xrange(1000000))\r
        self.data = '\n'.join(line_gen)\r
\r
        # And write it to a file.\r
        fp = open(TESTFN, "wb")\r
        fp.write(self.data)\r
        fp.close()\r
\r
    def zipTest(self, f, compression):\r
        # Create the ZIP archive.\r
        zipfp = zipfile.ZipFile(f, "w", compression, allowZip64=True)\r
\r
        # It will contain enough copies of self.data to reach about 6GB of\r
        # raw data to store.\r
        filecount = 6*1024**3 // len(self.data)\r
\r
        next_time = time.time() + _PRINT_WORKING_MSG_INTERVAL\r
        for num in range(filecount):\r
            zipfp.writestr("testfn%d" % num, self.data)\r
            # Print still working message since this test can be really slow\r
            if next_time <= time.time():\r
                next_time = time.time() + _PRINT_WORKING_MSG_INTERVAL\r
                print >>sys.__stdout__, (\r
                   '  zipTest still writing %d of %d, be patient...' %\r
                   (num, filecount))\r
                sys.__stdout__.flush()\r
        zipfp.close()\r
\r
        # Read the ZIP archive\r
        zipfp = zipfile.ZipFile(f, "r", compression)\r
        for num in range(filecount):\r
            self.assertEqual(zipfp.read("testfn%d" % num), self.data)\r
            # Print still working message since this test can be really slow\r
            if next_time <= time.time():\r
                next_time = time.time() + _PRINT_WORKING_MSG_INTERVAL\r
                print >>sys.__stdout__, (\r
                   '  zipTest still reading %d of %d, be patient...' %\r
                   (num, filecount))\r
                sys.__stdout__.flush()\r
        zipfp.close()\r
\r
    def testStored(self):\r
        # Try the temp file first.  If we do TESTFN2 first, then it hogs\r
        # gigabytes of disk space for the duration of the test.\r
        for f in TemporaryFile(), TESTFN2:\r
            self.zipTest(f, zipfile.ZIP_STORED)\r
\r
    if zlib:\r
        def testDeflated(self):\r
            # Try the temp file first.  If we do TESTFN2 first, then it hogs\r
            # gigabytes of disk space for the duration of the test.\r
            for f in TemporaryFile(), TESTFN2:\r
                self.zipTest(f, zipfile.ZIP_DEFLATED)\r
\r
    def tearDown(self):\r
        for fname in TESTFN, TESTFN2:\r
            if os.path.exists(fname):\r
                os.remove(fname)\r
\r
\r
class OtherTests(unittest.TestCase):\r
    def testMoreThan64kFiles(self):\r
        # This test checks that more than 64k files can be added to an archive,\r
        # and that the resulting archive can be read properly by ZipFile\r
        zipf = zipfile.ZipFile(TESTFN, mode="w")\r
        zipf.debug = 100\r
        numfiles = (1 << 16) * 3/2\r
        for i in xrange(numfiles):\r
            zipf.writestr("foo%08d" % i, "%d" % (i**3 % 57))\r
        self.assertEqual(len(zipf.namelist()), numfiles)\r
        zipf.close()\r
\r
        zipf2 = zipfile.ZipFile(TESTFN, mode="r")\r
        self.assertEqual(len(zipf2.namelist()), numfiles)\r
        for i in xrange(numfiles):\r
            self.assertEqual(zipf2.read("foo%08d" % i), "%d" % (i**3 % 57))\r
        zipf.close()\r
\r
    def tearDown(self):\r
        test_support.unlink(TESTFN)\r
        test_support.unlink(TESTFN2)\r
\r
def test_main():\r
    run_unittest(TestsWithSourceFile, OtherTests)\r
\r
if __name__ == "__main__":\r
    test_main()\r