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_bz2.py < prev    next >
Text File  |  2003-12-30  |  13KB  |  323 lines
  1. #!/usr/bin/python
  2. from test import test_support
  3. from test.test_support import TESTFN
  4.  
  5. import unittest
  6. from cStringIO import StringIO
  7. import os
  8. import popen2
  9. import sys
  10.  
  11. import bz2
  12. from bz2 import BZ2File, BZ2Compressor, BZ2Decompressor
  13.  
  14. has_cmdline_bunzip2 = sys.platform not in ("win32", "os2emx", "riscos")
  15.  
  16. class BaseTest(unittest.TestCase):
  17.     "Base for other testcases."
  18.     TEXT = 'root:x:0:0:root:/root:/bin/bash\nbin:x:1:1:bin:/bin:\ndaemon:x:2:2:daemon:/sbin:\nadm:x:3:4:adm:/var/adm:\nlp:x:4:7:lp:/var/spool/lpd:\nsync:x:5:0:sync:/sbin:/bin/sync\nshutdown:x:6:0:shutdown:/sbin:/sbin/shutdown\nhalt:x:7:0:halt:/sbin:/sbin/halt\nmail:x:8:12:mail:/var/spool/mail:\nnews:x:9:13:news:/var/spool/news:\nuucp:x:10:14:uucp:/var/spool/uucp:\noperator:x:11:0:operator:/root:\ngames:x:12:100:games:/usr/games:\ngopher:x:13:30:gopher:/usr/lib/gopher-data:\nftp:x:14:50:FTP User:/var/ftp:/bin/bash\nnobody:x:65534:65534:Nobody:/home:\npostfix:x:100:101:postfix:/var/spool/postfix:\nniemeyer:x:500:500::/home/niemeyer:/bin/bash\npostgres:x:101:102:PostgreSQL Server:/var/lib/pgsql:/bin/bash\nmysql:x:102:103:MySQL server:/var/lib/mysql:/bin/bash\nwww:x:103:104::/var/www:/bin/false\n'
  19.     DATA = 'BZh91AY&SY.\xc8N\x18\x00\x01>_\x80\x00\x10@\x02\xff\xf0\x01\x07n\x00?\xe7\xff\xe00\x01\x99\xaa\x00\xc0\x03F\x86\x8c#&\x83F\x9a\x03\x06\xa6\xd0\xa6\x93M\x0fQ\xa7\xa8\x06\x804hh\x12$\x11\xa4i4\xf14S\xd2<Q\xb5\x0fH\xd3\xd4\xdd\xd5\x87\xbb\xf8\x94\r\x8f\xafI\x12\xe1\xc9\xf8/E\x00pu\x89\x12]\xc9\xbbDL\nQ\x0e\t1\x12\xdf\xa0\xc0\x97\xac2O9\x89\x13\x94\x0e\x1c7\x0ed\x95I\x0c\xaaJ\xa4\x18L\x10\x05#\x9c\xaf\xba\xbc/\x97\x8a#C\xc8\xe1\x8cW\xf9\xe2\xd0\xd6M\xa7\x8bXa<e\x84t\xcbL\xb3\xa7\xd9\xcd\xd1\xcb\x84.\xaf\xb3\xab\xab\xad`n}\xa0lh\tE,\x8eZ\x15\x17VH>\x88\xe5\xcd9gd6\x0b\n\xe9\x9b\xd5\x8a\x99\xf7\x08.K\x8ev\xfb\xf7xw\xbb\xdf\xa1\x92\xf1\xdd|/";\xa2\xba\x9f\xd5\xb1#A\xb6\xf6\xb3o\xc9\xc5y\\\xebO\xe7\x85\x9a\xbc\xb6f8\x952\xd5\xd7"%\x89>V,\xf7\xa6z\xe2\x9f\xa3\xdf\x11\x11"\xd6E)I\xa9\x13^\xca\xf3r\xd0\x03U\x922\xf26\xec\xb6\xed\x8b\xc3U\x13\x9d\xc5\x170\xa4\xfa^\x92\xacDF\x8a\x97\xd6\x19\xfe\xdd\xb8\xbd\x1a\x9a\x19\xa3\x80ankR\x8b\xe5\xd83]\xa9\xc6\x08\x82f\xf6\xb9"6l$\xb8j@\xc0\x8a\xb0l1..\xbak\x83ls\x15\xbc\xf4\xc1\x13\xbe\xf8E\xb8\x9d\r\xa8\x9dk\x84\xd3n\xfa\xacQ\x07\xb1%y\xaav\xb4\x08\xe0z\x1b\x16\xf5\x04\xe9\xcc\xb9\x08z\x1en7.G\xfc]\xc9\x14\xe1B@\xbb!8`'
  20.     DATA_CRLF = 'BZh91AY&SY\xaez\xbbN\x00\x01H\xdf\x80\x00\x12@\x02\xff\xf0\x01\x07n\x00?\xe7\xff\xe0@\x01\xbc\xc6`\x86*\x8d=M\xa9\x9a\x86\xd0L@\x0fI\xa6!\xa1\x13\xc8\x88jdi\x8d@\x03@\x1a\x1a\x0c\x0c\x83 \x00\xc4h2\x19\x01\x82D\x84e\t\xe8\x99\x89\x19\x1ah\x00\r\x1a\x11\xaf\x9b\x0fG\xf5(\x1b\x1f?\t\x12\xcf\xb5\xfc\x95E\x00ps\x89\x12^\xa4\xdd\xa2&\x05(\x87\x04\x98\x89u\xe40%\xb6\x19\'\x8c\xc4\x89\xca\x07\x0e\x1b!\x91UIFU%C\x994!DI\xd2\xfa\xf0\xf1N8W\xde\x13A\xf5\x9cr%?\x9f3;I45A\xd1\x8bT\xb1<l\xba\xcb_\xc00xY\x17r\x17\x88\x08\x08@\xa0\ry@\x10\x04$)`\xf2\xce\x89z\xb0s\xec\x9b.iW\x9d\x81\xb5-+t\x9f\x1a\'\x97dB\xf5x\xb5\xbe.[.\xd7\x0e\x81\xe7\x08\x1cN`\x88\x10\xca\x87\xc3!"\x80\x92R\xa1/\xd1\xc0\xe6mf\xac\xbd\x99\xcca\xb3\x8780>\xa4\xc7\x8d\x1a\\"\xad\xa1\xabyBg\x15\xb9l\x88\x88\x91k"\x94\xa4\xd4\x89\xae*\xa6\x0b\x10\x0c\xd6\xd4m\xe86\xec\xb5j\x8a\x86j\';\xca.\x01I\xf2\xaaJ\xe8\x88\x8cU+t3\xfb\x0c\n\xa33\x13r2\r\x16\xe0\xb3(\xbf\x1d\x83r\xe7M\xf0D\x1365\xd8\x88\xd3\xa4\x92\xcb2\x06\x04\\\xc1\xb0\xea//\xbek&\xd8\xe6+t\xe5\xa1\x13\xada\x16\xder5"w]\xa2i\xb7[\x97R \xe2IT\xcd;Z\x04dk4\xad\x8a\t\xd3\x81z\x10\xf1:^`\xab\x1f\xc5\xdc\x91N\x14$+\x9e\xae\xd3\x80'
  21.  
  22.     if has_cmdline_bunzip2:
  23.         def decompress(self, data):
  24.             pop = popen2.Popen3("bunzip2", capturestderr=1)
  25.             pop.tochild.write(data)
  26.             pop.tochild.close()
  27.             ret = pop.fromchild.read()
  28.             pop.fromchild.close()
  29.             if pop.wait() != 0:
  30.                 ret = bz2.decompress(data)
  31.             return ret
  32.  
  33.     else:
  34.         # popen2.Popen3 doesn't exist on Windows, and even if it did, bunzip2
  35.         # isn't available to run.
  36.         def decompress(self, data):
  37.             return bz2.decompress(data)
  38.  
  39. class BZ2FileTest(BaseTest):
  40.     "Test MCRYPT type miscelaneous methods."
  41.  
  42.     def setUp(self):
  43.         self.filename = TESTFN
  44.  
  45.     def tearDown(self):
  46.         if os.path.isfile(self.filename):
  47.             os.unlink(self.filename)
  48.  
  49.     def createTempFile(self, crlf=0):
  50.         f = open(self.filename, "wb")
  51.         if crlf:
  52.             data = self.DATA_CRLF
  53.         else:
  54.             data = self.DATA
  55.         f.write(data)
  56.         f.close()
  57.  
  58.     def testRead(self):
  59.         # "Test BZ2File.read()"
  60.         self.createTempFile()
  61.         bz2f = BZ2File(self.filename)
  62.         self.assertEqual(bz2f.read(), self.TEXT)
  63.         bz2f.close()
  64.  
  65.     def testReadChunk10(self):
  66.         # "Test BZ2File.read() in chunks of 10 bytes"
  67.         self.createTempFile()
  68.         bz2f = BZ2File(self.filename)
  69.         text = ''
  70.         while 1:
  71.             str = bz2f.read(10)
  72.             if not str:
  73.                 break
  74.             text += str
  75.         self.assertEqual(text, text)
  76.         bz2f.close()
  77.  
  78.     def testRead100(self):
  79.         # "Test BZ2File.read(100)"
  80.         self.createTempFile()
  81.         bz2f = BZ2File(self.filename)
  82.         self.assertEqual(bz2f.read(100), self.TEXT[:100])
  83.         bz2f.close()
  84.  
  85.     def testReadLine(self):
  86.         # "Test BZ2File.readline()"
  87.         self.createTempFile()
  88.         bz2f = BZ2File(self.filename)
  89.         sio = StringIO(self.TEXT)
  90.         for line in sio.readlines():
  91.             self.assertEqual(bz2f.readline(), line)
  92.         bz2f.close()
  93.  
  94.     def testReadLines(self):
  95.         # "Test BZ2File.readlines()"
  96.         self.createTempFile()
  97.         bz2f = BZ2File(self.filename)
  98.         sio = StringIO(self.TEXT)
  99.         self.assertEqual(bz2f.readlines(), sio.readlines())
  100.         bz2f.close()
  101.  
  102.     def testIterator(self):
  103.         # "Test iter(BZ2File)"
  104.         self.createTempFile()
  105.         bz2f = BZ2File(self.filename)
  106.         sio = StringIO(self.TEXT)
  107.         self.assertEqual(list(iter(bz2f)), sio.readlines())
  108.         bz2f.close()
  109.  
  110.     def testXReadLines(self):
  111.         # "Test BZ2File.xreadlines()"
  112.         self.createTempFile()
  113.         bz2f = BZ2File(self.filename)
  114.         sio = StringIO(self.TEXT)
  115.         self.assertEqual(list(bz2f.xreadlines()), sio.readlines())
  116.         bz2f.close()
  117.  
  118.     def testUniversalNewlinesLF(self):
  119.         # "Test BZ2File.read() with universal newlines (\\n)"
  120.         self.createTempFile()
  121.         bz2f = BZ2File(self.filename, "rU")
  122.         self.assertEqual(bz2f.read(), self.TEXT)
  123.         self.assertEqual(bz2f.newlines, "\n")
  124.         bz2f.close()
  125.  
  126.     def testUniversalNewlinesCRLF(self):
  127.         # "Test BZ2File.read() with universal newlines (\\r\\n)"
  128.         self.createTempFile(crlf=1)
  129.         bz2f = BZ2File(self.filename, "rU")
  130.         self.assertEqual(bz2f.read(), self.TEXT)
  131.         self.assertEqual(bz2f.newlines, "\r\n")
  132.         bz2f.close()
  133.  
  134.     def testWrite(self):
  135.         # "Test BZ2File.write()"
  136.         bz2f = BZ2File(self.filename, "w")
  137.         bz2f.write(self.TEXT)
  138.         bz2f.close()
  139.         f = open(self.filename, 'rb')
  140.         self.assertEqual(self.decompress(f.read()), self.TEXT)
  141.         f.close()
  142.  
  143.     def testWriteChunks10(self):
  144.         # "Test BZ2File.write() with chunks of 10 bytes"
  145.         bz2f = BZ2File(self.filename, "w")
  146.         n = 0
  147.         while 1:
  148.             str = self.TEXT[n*10:(n+1)*10]
  149.             if not str:
  150.                 break
  151.             bz2f.write(str)
  152.             n += 1
  153.         bz2f.close()
  154.         f = open(self.filename, 'rb')
  155.         self.assertEqual(self.decompress(f.read()), self.TEXT)
  156.         f.close()
  157.  
  158.     def testWriteLines(self):
  159.         # "Test BZ2File.writelines()"
  160.         bz2f = BZ2File(self.filename, "w")
  161.         sio = StringIO(self.TEXT)
  162.         bz2f.writelines(sio.readlines())
  163.         bz2f.close()
  164.         f = open(self.filename, 'rb')
  165.         self.assertEqual(self.decompress(f.read()), self.TEXT)
  166.         f.close()
  167.  
  168.     def testSeekForward(self):
  169.         # "Test BZ2File.seek(150, 0)"
  170.         self.createTempFile()
  171.         bz2f = BZ2File(self.filename)
  172.         bz2f.seek(150)
  173.         self.assertEqual(bz2f.read(), self.TEXT[150:])
  174.         bz2f.close()
  175.  
  176.     def testSeekBackwards(self):
  177.         # "Test BZ2File.seek(-150, 1)"
  178.         self.createTempFile()
  179.         bz2f = BZ2File(self.filename)
  180.         bz2f.read(500)
  181.         bz2f.seek(-150, 1)
  182.         self.assertEqual(bz2f.read(), self.TEXT[500-150:])
  183.         bz2f.close()
  184.  
  185.     def testSeekBackwardsFromEnd(self):
  186.         # "Test BZ2File.seek(-150, 2)"
  187.         self.createTempFile()
  188.         bz2f = BZ2File(self.filename)
  189.         bz2f.seek(-150, 2)
  190.         self.assertEqual(bz2f.read(), self.TEXT[len(self.TEXT)-150:])
  191.         bz2f.close()
  192.  
  193.     def testSeekPostEnd(self):
  194.         # "Test BZ2File.seek(150000)"
  195.         self.createTempFile()
  196.         bz2f = BZ2File(self.filename)
  197.         bz2f.seek(150000)
  198.         self.assertEqual(bz2f.tell(), len(self.TEXT))
  199.         self.assertEqual(bz2f.read(), "")
  200.         bz2f.close()
  201.  
  202.     def testSeekPostEndTwice(self):
  203.         # "Test BZ2File.seek(150000) twice"
  204.         self.createTempFile()
  205.         bz2f = BZ2File(self.filename)
  206.         bz2f.seek(150000)
  207.         bz2f.seek(150000)
  208.         self.assertEqual(bz2f.tell(), len(self.TEXT))
  209.         self.assertEqual(bz2f.read(), "")
  210.         bz2f.close()
  211.  
  212.     def testSeekPreStart(self):
  213.         # "Test BZ2File.seek(-150, 0)"
  214.         self.createTempFile()
  215.         bz2f = BZ2File(self.filename)
  216.         bz2f.seek(-150)
  217.         self.assertEqual(bz2f.tell(), 0)
  218.         self.assertEqual(bz2f.read(), self.TEXT)
  219.         bz2f.close()
  220.  
  221.     def testOpenDel(self):
  222.         # "Test opening and deleting a file many times"
  223.         self.createTempFile()
  224.         for i in xrange(10000):
  225.             o = BZ2File(self.filename)
  226.             del o
  227.  
  228.     def testOpenNonexistent(self):
  229.         # "Test opening a nonexistent file"
  230.         self.assertRaises(IOError, BZ2File, "/non/existent")
  231.  
  232. class BZ2CompressorTest(BaseTest):
  233.     def testCompress(self):
  234.         # "Test BZ2Compressor.compress()/flush()"
  235.         bz2c = BZ2Compressor()
  236.         data = bz2c.compress(self.TEXT)
  237.         data += bz2c.flush()
  238.         self.assertEqual(self.decompress(data), self.TEXT)
  239.  
  240.     def testCompressChunks10(self):
  241.         # "Test BZ2Compressor.compress()/flush() with chunks of 10 bytes"
  242.         bz2c = BZ2Compressor()
  243.         n = 0
  244.         data = ''
  245.         while 1:
  246.             str = self.TEXT[n*10:(n+1)*10]
  247.             if not str:
  248.                 break
  249.             data += bz2c.compress(str)
  250.             n += 1
  251.         data += bz2c.flush()
  252.         self.assertEqual(self.decompress(data), self.TEXT)
  253.  
  254. class BZ2DecompressorTest(BaseTest):
  255.     def testDecompress(self):
  256.         # "Test BZ2Decompressor.decompress()"
  257.         bz2d = BZ2Decompressor()
  258.         text = bz2d.decompress(self.DATA)
  259.         self.assertEqual(text, self.TEXT)
  260.  
  261.     def testDecompressChunks10(self):
  262.         # "Test BZ2Decompressor.decompress() with chunks of 10 bytes"
  263.         bz2d = BZ2Decompressor()
  264.         text = ''
  265.         n = 0
  266.         while 1:
  267.             str = self.DATA[n*10:(n+1)*10]
  268.             if not str:
  269.                 break
  270.             text += bz2d.decompress(str)
  271.             n += 1
  272.         self.assertEqual(text, self.TEXT)
  273.  
  274.     def testDecompressUnusedData(self):
  275.         # "Test BZ2Decompressor.decompress() with unused data"
  276.         bz2d = BZ2Decompressor()
  277.         unused_data = "this is unused data"
  278.         text = bz2d.decompress(self.DATA+unused_data)
  279.         self.assertEqual(text, self.TEXT)
  280.         self.assertEqual(bz2d.unused_data, unused_data)
  281.  
  282.     def testEOFError(self):
  283.         # "Calling BZ2Decompressor.decompress() after EOS must raise EOFError"
  284.         bz2d = BZ2Decompressor()
  285.         text = bz2d.decompress(self.DATA)
  286.         self.assertRaises(EOFError, bz2d.decompress, "anything")
  287.  
  288.  
  289. class FuncTest(BaseTest):
  290.     "Test module functions"
  291.  
  292.     def testCompress(self):
  293.         # "Test compress() function"
  294.         data = bz2.compress(self.TEXT)
  295.         self.assertEqual(self.decompress(data), self.TEXT)
  296.  
  297.     def testDecompress(self):
  298.         # "Test decompress() function"
  299.         text = bz2.decompress(self.DATA)
  300.         self.assertEqual(text, self.TEXT)
  301.  
  302.     def testDecompressEmpty(self):
  303.         # "Test decompress() function with empty string"
  304.         text = bz2.decompress("")
  305.         self.assertEqual(text, "")
  306.  
  307.     def testDecompressIncomplete(self):
  308.         # "Test decompress() function with incomplete data"
  309.         self.assertRaises(ValueError, bz2.decompress, self.DATA[:-10])
  310.  
  311. def test_main():
  312.     test_support.run_unittest(
  313.         BZ2FileTest,
  314.         BZ2CompressorTest,
  315.         BZ2DecompressorTest,
  316.         FuncTest
  317.     )
  318.  
  319. if __name__ == '__main__':
  320.     test_main()
  321.  
  322. # vim:ts=4:sw=4
  323.