move sections
[python/dscho.git] / Lib / test / test_applesingle.py
blob1beffe0f2f465919e71eec50a238eaf4a841c93c
1 # Copyright (C) 2003 Python Software Foundation
3 import unittest
4 import os
5 from test import test_support
6 import struct
8 MacOS = test_support.import_module('MacOS')
9 # The following should exist if MacOS does.
10 import applesingle
12 AS_MAGIC=0x00051600
13 AS_VERSION=0x00020000
14 dataforkdata = 'hello\r\0world\n'
15 resourceforkdata = 'goodbye\ncruel\0world\r'
17 applesingledata = struct.pack(">ll16sh", AS_MAGIC, AS_VERSION, "foo", 2) + \
18 struct.pack(">llllll", 1, 50, len(dataforkdata),
19 2, 50+len(dataforkdata), len(resourceforkdata)) + \
20 dataforkdata + \
21 resourceforkdata
22 TESTFN2 = test_support.TESTFN + '2'
24 class TestApplesingle(unittest.TestCase):
26 def setUp(self):
27 fp = open(test_support.TESTFN, 'w')
28 fp.write(applesingledata)
29 fp.close()
31 def tearDown(self):
32 try:
33 os.unlink(test_support.TESTFN)
34 except:
35 pass
36 try:
37 os.unlink(TESTFN2)
38 except:
39 pass
41 def compareData(self, isrf, data):
42 if isrf:
43 fp = MacOS.openrf(TESTFN2, '*rb')
44 else:
45 fp = open(TESTFN2, 'rb')
46 filedata = fp.read(1000)
47 self.assertEqual(data, filedata)
49 def test_applesingle(self):
50 try:
51 os.unlink(TESTFN2)
52 except:
53 pass
54 applesingle.decode(test_support.TESTFN, TESTFN2)
55 self.compareData(False, dataforkdata)
56 self.compareData(True, resourceforkdata)
58 def test_applesingle_resonly(self):
59 try:
60 os.unlink(TESTFN2)
61 except:
62 pass
63 applesingle.decode(test_support.TESTFN, TESTFN2, resonly=True)
64 self.compareData(False, resourceforkdata)
66 def test_main():
67 test_support.run_unittest(TestApplesingle)
70 if __name__ == '__main__':
71 test_main()