github.com/grumpyhome/grumpy@v0.3.1-0.20201208125205-7b775405bdf1/grumpy-runtime-src/third_party/stdlib/test/test_stat.py (about) 1 import unittest 2 import os 3 from test.test_support import TESTFN, run_unittest 4 import stat 5 6 class TestFilemode(unittest.TestCase): 7 file_flags = {'SF_APPEND', 'SF_ARCHIVED', 'SF_IMMUTABLE', 'SF_NOUNLINK', 8 'SF_SNAPSHOT', 'UF_APPEND', 'UF_COMPRESSED', 'UF_HIDDEN', 9 'UF_IMMUTABLE', 'UF_NODUMP', 'UF_NOUNLINK', 'UF_OPAQUE'} 10 11 formats = {'S_IFBLK', 'S_IFCHR', 'S_IFDIR', 'S_IFIFO', 'S_IFLNK', 12 'S_IFREG', 'S_IFSOCK'} 13 14 format_funcs = {'S_ISBLK', 'S_ISCHR', 'S_ISDIR', 'S_ISFIFO', 'S_ISLNK', 15 'S_ISREG', 'S_ISSOCK'} 16 17 stat_struct = { 18 'ST_MODE': 0, 19 'ST_INO': 1, 20 'ST_DEV': 2, 21 'ST_NLINK': 3, 22 'ST_UID': 4, 23 'ST_GID': 5, 24 'ST_SIZE': 6, 25 'ST_ATIME': 7, 26 'ST_MTIME': 8, 27 'ST_CTIME': 9} 28 29 # permission bit value are defined by POSIX 30 permission_bits = { 31 'S_ISUID': 0o4000, 32 'S_ISGID': 0o2000, 33 'S_ENFMT': 0o2000, 34 'S_ISVTX': 0o1000, 35 'S_IRWXU': 0o700, 36 'S_IRUSR': 0o400, 37 'S_IREAD': 0o400, 38 'S_IWUSR': 0o200, 39 'S_IWRITE': 0o200, 40 'S_IXUSR': 0o100, 41 'S_IEXEC': 0o100, 42 'S_IRWXG': 0o070, 43 'S_IRGRP': 0o040, 44 'S_IWGRP': 0o020, 45 'S_IXGRP': 0o010, 46 'S_IRWXO': 0o007, 47 'S_IROTH': 0o004, 48 'S_IWOTH': 0o002, 49 'S_IXOTH': 0o001} 50 51 def setUp(self): 52 try: 53 os.remove(TESTFN) 54 except OSError: 55 try: 56 os.rmdir(TESTFN) 57 except OSError: 58 pass 59 tearDown = setUp 60 61 def get_mode(self, fname=TESTFN): #, lstat=True): 62 # if lstat: 63 # st_mode = os.lstat(fname).st_mode 64 # else: 65 st_mode = os.stat(fname).st_mode 66 return st_mode 67 68 def assertS_IS(self, name, mode): 69 # test format, lstrip is for S_IFIFO 70 # fmt = getattr(stat, "S_IF" + name.lstrip("F")) 71 # self.assertEqual(stat.S_IFMT(mode), fmt) 72 # test that just one function returns true 73 testname = "S_IS" + name 74 for funcname in self.format_funcs: 75 func = getattr(stat, funcname, None) 76 if func is None: 77 if funcname == testname: 78 raise ValueError(funcname) 79 continue 80 if funcname == testname: 81 self.assertTrue(func(mode)) 82 else: 83 self.assertFalse(func(mode)) 84 85 @unittest.skip('grumpy') 86 def test_mode(self): 87 with open(TESTFN, 'w'): 88 pass 89 if os.name == 'posix': 90 os.chmod(TESTFN, 0o700) 91 st_mode = self.get_mode() 92 self.assertS_IS("REG", st_mode) 93 self.assertEqual(stat.S_IMODE(st_mode), 94 stat.S_IRWXU) 95 96 os.chmod(TESTFN, 0o070) 97 st_mode = self.get_mode() 98 self.assertS_IS("REG", st_mode) 99 self.assertEqual(stat.S_IMODE(st_mode), 100 stat.S_IRWXG) 101 102 os.chmod(TESTFN, 0o007) 103 st_mode = self.get_mode() 104 self.assertS_IS("REG", st_mode) 105 self.assertEqual(stat.S_IMODE(st_mode), 106 stat.S_IRWXO) 107 108 os.chmod(TESTFN, 0o444) 109 st_mode = self.get_mode() 110 self.assertS_IS("REG", st_mode) 111 self.assertEqual(stat.S_IMODE(st_mode), 0o444) 112 else: 113 os.chmod(TESTFN, 0o700) 114 st_mode = self.get_mode() 115 self.assertS_IS("REG", st_mode) 116 self.assertEqual(stat.S_IFMT(st_mode), 117 stat.S_IFREG) 118 119 def test_directory(self): 120 os.mkdir(TESTFN) 121 os.chmod(TESTFN, 0o700) 122 st_mode = self.get_mode() 123 self.assertS_IS("DIR", st_mode) 124 125 @unittest.skip('grumpy') 126 @unittest.skipUnless(hasattr(os, 'symlink'), 'os.symlink not available') 127 def test_link(self): 128 try: 129 os.symlink(os.getcwd(), TESTFN) 130 except (OSError, NotImplementedError) as err: 131 raise unittest.SkipTest(str(err)) 132 else: 133 st_mode = self.get_mode() 134 self.assertS_IS("LNK", st_mode) 135 136 @unittest.skip('grumpy') 137 @unittest.skipUnless(hasattr(os, 'mkfifo'), 'os.mkfifo not available') 138 def test_fifo(self): 139 os.mkfifo(TESTFN, 0o700) 140 st_mode = self.get_mode() 141 self.assertS_IS("FIFO", st_mode) 142 143 @unittest.skip('grumpy') 144 @unittest.skipUnless(os.name == 'posix', 'requires Posix') 145 def test_devices(self): 146 if os.path.exists(os.devnull): 147 st_mode = self.get_mode(os.devnull, lstat=False) 148 self.assertS_IS("CHR", st_mode) 149 # Linux block devices, BSD has no block devices anymore 150 for blockdev in ("/dev/sda", "/dev/hda"): 151 if os.path.exists(blockdev): 152 st_mode = self.get_mode(blockdev, lstat=False) 153 self.assertS_IS("BLK", st_mode) 154 break 155 156 @unittest.skip('grumpy') 157 def test_module_attributes(self): 158 for key, value in self.stat_struct.items(): 159 modvalue = getattr(stat, key) 160 self.assertEqual(value, modvalue, key) 161 for key, value in self.permission_bits.items(): 162 modvalue = getattr(stat, key) 163 self.assertEqual(value, modvalue, key) 164 for key in self.file_flags: 165 modvalue = getattr(stat, key) 166 self.assertIsInstance(modvalue, int) 167 for key in self.formats: 168 modvalue = getattr(stat, key) 169 self.assertIsInstance(modvalue, int) 170 for key in self.format_funcs: 171 func = getattr(stat, key) 172 self.assertTrue(callable(func)) 173 self.assertEqual(func(0), 0) 174 175 176 def test_main(): 177 run_unittest(TestFilemode) 178 179 if __name__ == '__main__': 180 test_main()