EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Lib / test / test_macos.py

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