/Lib/test/test_applesingle.py
http://unladen-swallow.googlecode.com/ · Python · 71 lines · 58 code · 12 blank · 1 comment · 7 complexity · 9d183a2b9243c3a84b55bedb16902c66 MD5 · raw file
- # Copyright (C) 2003 Python Software Foundation
- import unittest
- import macostools
- import Carbon.File
- import MacOS
- import os
- from test import test_support
- import struct
- import applesingle
- AS_MAGIC=0x00051600
- AS_VERSION=0x00020000
- dataforkdata = 'hello\r\0world\n'
- resourceforkdata = 'goodbye\ncruel\0world\r'
- applesingledata = struct.pack(">ll16sh", AS_MAGIC, AS_VERSION, "foo", 2) + \
- struct.pack(">llllll", 1, 50, len(dataforkdata),
- 2, 50+len(dataforkdata), len(resourceforkdata)) + \
- dataforkdata + \
- resourceforkdata
- TESTFN2 = test_support.TESTFN + '2'
- class TestApplesingle(unittest.TestCase):
- def setUp(self):
- fp = open(test_support.TESTFN, 'w')
- fp.write(applesingledata)
- fp.close()
- def tearDown(self):
- try:
- os.unlink(test_support.TESTFN)
- except:
- pass
- try:
- os.unlink(TESTFN2)
- except:
- pass
- def compareData(self, isrf, data):
- if isrf:
- fp = MacOS.openrf(TESTFN2, '*rb')
- else:
- fp = open(TESTFN2, 'rb')
- filedata = fp.read(1000)
- self.assertEqual(data, filedata)
- def test_applesingle(self):
- try:
- os.unlink(TESTFN2)
- except:
- pass
- applesingle.decode(test_support.TESTFN, TESTFN2)
- self.compareData(False, dataforkdata)
- self.compareData(True, resourceforkdata)
- def test_applesingle_resonly(self):
- try:
- os.unlink(TESTFN2)
- except:
- pass
- applesingle.decode(test_support.TESTFN, TESTFN2, resonly=True)
- self.compareData(False, resourceforkdata)
- def test_main():
- test_support.run_unittest(TestApplesingle)
- if __name__ == '__main__':
- test_main()