AppPkg/Applications/Python/Python-2.7.2/Lib/test/test_applesingle.py

   1 # Copyright (C) 2003 Python Software Foundation
   2
   3 import unittest
   4 import os
   5 from test import test_support
   6 import struct
   7
   8 MacOS = test_support.import_module('MacOS')
   9 # The following should exist if MacOS does.
  10 import applesingle
  11
  12 AS_MAGIC=0x00051600
  13 AS_VERSION=0x00020000
  14 dataforkdata = 'hello\r\0world\n'
  15 resourceforkdata = 'goodbye\ncruel\0world\r'
  16
  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'
  23
  24 class TestApplesingle(unittest.TestCase):
  25
  26     def setUp(self):
  27         fp = open(test_support.TESTFN, 'w')
  28         fp.write(applesingledata)
  29         fp.close()
  30
  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
  40
  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)
  48
  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)
  57
  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)
  65
  66 def test_main():
  67     test_support.run_unittest(TestApplesingle)
  68
  69
  70 if __name__ == '__main__':
  71     test_main()