home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / test_zipfile.py < prev    next >
Text File  |  2003-12-30  |  3KB  |  100 lines
  1. import zlib # implied prerequisite
  2. import zipfile, os, StringIO, tempfile
  3. from test.test_support import TestFailed
  4.  
  5. srcname = "junk9630"+os.extsep+"tmp"
  6. zipname = "junk9708"+os.extsep+"tmp"
  7.  
  8.  
  9. def zipTest(f, compression, srccontents):
  10.     zip = zipfile.ZipFile(f, "w", compression)   # Create the ZIP archive
  11.     zip.write(srcname, "another"+os.extsep+"name")
  12.     zip.write(srcname, srcname)
  13.     zip.close()
  14.  
  15.     zip = zipfile.ZipFile(f, "r", compression)   # Read the ZIP archive
  16.     readData2 = zip.read(srcname)
  17.     readData1 = zip.read("another"+os.extsep+"name")
  18.     zip.close()
  19.  
  20.     if readData1 != srccontents or readData2 != srccontents:
  21.         raise TestFailed, "Written data doesn't equal read data."
  22.  
  23.  
  24. try:
  25.     fp = open(srcname, "wb")               # Make a source file with some lines
  26.     for i in range(0, 1000):
  27.         fp.write("Test of zipfile line %d.\n" % i)
  28.     fp.close()
  29.  
  30.     fp = open(srcname, "rb")
  31.     writtenData = fp.read()
  32.     fp.close()
  33.  
  34.     for file in (zipname, tempfile.TemporaryFile(), StringIO.StringIO()):
  35.         zipTest(file, zipfile.ZIP_STORED, writtenData)
  36.  
  37.     for file in (zipname, tempfile.TemporaryFile(), StringIO.StringIO()):
  38.         zipTest(file, zipfile.ZIP_DEFLATED, writtenData)
  39.  
  40. finally:
  41.     if os.path.isfile(srcname):           # Remove temporary files
  42.         os.unlink(srcname)
  43.     if os.path.isfile(zipname):
  44.         os.unlink(zipname)
  45.  
  46.  
  47. # This test checks that the ZipFile constructor closes the file object
  48. # it opens if there's an error in the file.  If it doesn't, the traceback
  49. # holds a reference to the ZipFile object and, indirectly, the file object.
  50. # On Windows, this causes the os.unlink() call to fail because the
  51. # underlying file is still open.  This is SF bug #412214.
  52. #
  53. fp = open(srcname, "w")
  54. fp.write("this is not a legal zip file\n")
  55. fp.close()
  56. try:
  57.     zf = zipfile.ZipFile(srcname)
  58. except zipfile.BadZipfile:
  59.     os.unlink(srcname)
  60.  
  61.  
  62. # make sure we don't raise an AttributeError when a partially-constructed
  63. # ZipFile instance is finalized; this tests for regression on SF tracker
  64. # bug #403871.
  65. try:
  66.     zipfile.ZipFile(srcname)
  67. except IOError:
  68.     # The bug we're testing for caused an AttributeError to be raised
  69.     # when a ZipFile instance was created for a file that did not
  70.     # exist; the .fp member was not initialized but was needed by the
  71.     # __del__() method.  Since the AttributeError is in the __del__(),
  72.     # it is ignored, but the user should be sufficiently annoyed by
  73.     # the message on the output that regression will be noticed
  74.     # quickly.
  75.     pass
  76. else:
  77.     raise TestFailed("expected creation of readable ZipFile without\n"
  78.                      "  a file to raise an IOError.")
  79.  
  80.  
  81. # Verify that testzip() doesn't swallow inappropriate exceptions.
  82. data = StringIO.StringIO()
  83. zipf = zipfile.ZipFile(data, mode="w")
  84. zipf.writestr("foo.txt", "O, for a Muse of Fire!")
  85. zipf.close()
  86. zipf = zipfile.ZipFile(data, mode="r")
  87. zipf.close()
  88. try:
  89.     zipf.testzip()
  90. except RuntimeError:
  91.     # This is correct; calling .read on a closed ZipFile should throw
  92.     # a RuntimeError, and so should calling .testzip.  An earlier
  93.     # version of .testzip would swallow this exception (and any other)
  94.     # and report that the first file in the archive was corrupt.
  95.     pass
  96. else:
  97.     raise TestFailed("expected calling .testzip on a closed ZipFile"
  98.                      " to raise a RuntimeError")
  99. del data, zipf
  100.