mirror of
https://github.com/python/cpython.git
synced 2024-12-14 20:34:12 +08:00
73 lines
1.7 KiB
Python
73 lines
1.7 KiB
Python
# Copyright (C) 2003 Python Software Foundation
|
|
|
|
import unittest
|
|
import macostools
|
|
import Carbon.File
|
|
import MacOS
|
|
import os
|
|
import sys
|
|
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()
|