import unittest from test import test_support import os import subprocess MacOS = test_support.import_module('MacOS') TESTFN2 = test_support.TESTFN + '2' class TestMacOS(unittest.TestCase): def testGetCreatorAndType(self): if not os.path.exists('/Developer/Tools/SetFile'): return try: fp = open(test_support.TESTFN, 'w') fp.write('\n') fp.close() subprocess.call( ['/Developer/Tools/SetFile', '-t', 'ABCD', '-c', 'EFGH', test_support.TESTFN]) cr, tp = MacOS.GetCreatorAndType(test_support.TESTFN) self.assertEqual(tp, 'ABCD') self.assertEqual(cr, 'EFGH') finally: os.unlink(test_support.TESTFN) def testSetCreatorAndType(self): if not os.path.exists('/Developer/Tools/GetFileInfo'): return try: fp = open(test_support.TESTFN, 'w') fp.write('\n') fp.close() MacOS.SetCreatorAndType(test_support.TESTFN, 'ABCD', 'EFGH') cr, tp = MacOS.GetCreatorAndType(test_support.TESTFN) self.assertEqual(cr, 'ABCD') self.assertEqual(tp, 'EFGH') data = subprocess.Popen(["/Developer/Tools/GetFileInfo", test_support.TESTFN], stdout=subprocess.PIPE).communicate()[0] tp = None cr = None for ln in data.splitlines(): if ln.startswith('type:'): tp = ln.split()[-1][1:-1] if ln.startswith('creator:'): cr = ln.split()[-1][1:-1] self.assertEqual(cr, 'ABCD') self.assertEqual(tp, 'EFGH') finally: os.unlink(test_support.TESTFN) def testOpenRF(self): try: fp = open(test_support.TESTFN, 'w') fp.write('hello world\n') fp.close() rfp = MacOS.openrf(test_support.TESTFN, '*wb') rfp.write('goodbye world\n') rfp.close() fp = open(test_support.TESTFN, 'r') data = fp.read() fp.close() self.assertEqual(data, 'hello world\n') rfp = MacOS.openrf(test_support.TESTFN, '*rb') data = rfp.read(100) data2 = rfp.read(100) rfp.close() self.assertEqual(data, 'goodbye world\n') self.assertEqual(data2, '') finally: os.unlink(test_support.TESTFN) def test_main(): test_support.run_unittest(TestMacOS) if __name__ == '__main__': test_main()