Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 1 | """Tests for sys.audit and sys.addaudithook |
| 2 | """ |
| 3 | |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 4 | import subprocess |
| 5 | import sys |
| 6 | import unittest |
| 7 | from test import support |
| 8 | |
| 9 | if not hasattr(sys, "addaudithook") or not hasattr(sys, "audit"): |
| 10 | raise unittest.SkipTest("test only relevant when sys.audit is available") |
| 11 | |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 12 | AUDIT_TESTS_PY = support.findfile("audit-tests.py") |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 13 | |
| 14 | |
| 15 | class AuditTest(unittest.TestCase): |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 16 | def do_test(self, *args): |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 17 | with subprocess.Popen( |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 18 | [sys.executable, "-X utf8", AUDIT_TESTS_PY, *args], |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 19 | encoding="utf-8", |
| 20 | stdout=subprocess.PIPE, |
| 21 | stderr=subprocess.PIPE, |
| 22 | ) as p: |
| 23 | p.wait() |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 24 | sys.stdout.writelines(p.stdout) |
| 25 | sys.stderr.writelines(p.stderr) |
| 26 | if p.returncode: |
| 27 | self.fail(''.join(p.stderr)) |
| 28 | |
| 29 | def test_basic(self): |
| 30 | self.do_test("test_basic") |
| 31 | |
| 32 | def test_block_add_hook(self): |
| 33 | self.do_test("test_block_add_hook") |
| 34 | |
| 35 | def test_block_add_hook_baseexception(self): |
| 36 | self.do_test("test_block_add_hook_baseexception") |
| 37 | |
| 38 | def test_finalize_hooks(self): |
| 39 | events = [] |
| 40 | with subprocess.Popen( |
| 41 | [sys.executable, "-X utf8", AUDIT_TESTS_PY, "test_finalize_hooks"], |
| 42 | encoding="utf-8", |
| 43 | stdout=subprocess.PIPE, |
| 44 | stderr=subprocess.PIPE, |
| 45 | ) as p: |
| 46 | p.wait() |
| 47 | for line in p.stdout: |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 48 | events.append(line.strip().partition(" ")) |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 49 | sys.stderr.writelines(p.stderr) |
| 50 | if p.returncode: |
| 51 | self.fail(''.join(p.stderr)) |
| 52 | |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 53 | firstId = events[0][2] |
| 54 | self.assertSequenceEqual( |
| 55 | [ |
| 56 | ("Created", " ", firstId), |
| 57 | ("cpython._PySys_ClearAuditHooks", " ", firstId), |
| 58 | ], |
| 59 | events, |
| 60 | ) |
| 61 | |
| 62 | def test_pickle(self): |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 63 | support.import_module("pickle") |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 64 | |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 65 | self.do_test("test_pickle") |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 66 | |
| 67 | def test_monkeypatch(self): |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 68 | self.do_test("test_monkeypatch") |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 69 | |
| 70 | def test_open(self): |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 71 | self.do_test("test_open", support.TESTFN) |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 72 | |
| 73 | def test_cantrace(self): |
Steve Dower | 9ddc416 | 2019-05-29 08:20:35 -0700 | [diff] [blame] | 74 | self.do_test("test_cantrace") |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 75 | |
Steve Dower | 6c79477 | 2019-06-21 09:45:13 -0700 | [diff] [blame] | 76 | def test_mmap(self): |
| 77 | self.do_test("test_mmap") |
| 78 | |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 79 | |
| 80 | if __name__ == "__main__": |
Steve Dower | b82e17e | 2019-05-23 08:45:22 -0700 | [diff] [blame] | 81 | unittest.main() |