80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
"""Tests for sys.audit and sys.addaudithook
|
|
"""
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
from test import support
|
|
|
|
if not hasattr(sys, "addaudithook") or not hasattr(sys, "audit"):
|
|
raise unittest.SkipTest("test only relevant when sys.audit is available")
|
|
|
|
AUDIT_TESTS_PY = support.findfile("audit-tests.py")
|
|
|
|
|
|
class AuditTest(unittest.TestCase):
|
|
def do_test(self, *args):
|
|
with subprocess.Popen(
|
|
[sys.executable, "-X utf8", AUDIT_TESTS_PY, *args],
|
|
encoding="utf-8",
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
) as p:
|
|
p.wait()
|
|
sys.stdout.writelines(p.stdout)
|
|
sys.stderr.writelines(p.stderr)
|
|
if p.returncode:
|
|
self.fail(''.join(p.stderr))
|
|
|
|
def test_basic(self):
|
|
self.do_test("test_basic")
|
|
|
|
def test_block_add_hook(self):
|
|
self.do_test("test_block_add_hook")
|
|
|
|
def test_block_add_hook_baseexception(self):
|
|
self.do_test("test_block_add_hook_baseexception")
|
|
|
|
def test_finalize_hooks(self):
|
|
events = []
|
|
with subprocess.Popen(
|
|
[sys.executable, "-X utf8", AUDIT_TESTS_PY, "test_finalize_hooks"],
|
|
encoding="utf-8",
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
) as p:
|
|
p.wait()
|
|
for line in p.stdout:
|
|
events.append(line.strip().partition(" "))
|
|
sys.stderr.writelines(p.stderr)
|
|
if p.returncode:
|
|
self.fail(''.join(p.stderr))
|
|
|
|
firstId = events[0][2]
|
|
self.assertSequenceEqual(
|
|
[
|
|
("Created", " ", firstId),
|
|
("cpython._PySys_ClearAuditHooks", " ", firstId),
|
|
],
|
|
events,
|
|
)
|
|
|
|
def test_pickle(self):
|
|
support.import_module("pickle")
|
|
|
|
self.do_test("test_pickle")
|
|
|
|
def test_monkeypatch(self):
|
|
self.do_test("test_monkeypatch")
|
|
|
|
def test_open(self):
|
|
self.do_test("test_open", support.TESTFN)
|
|
|
|
def test_cantrace(self):
|
|
self.do_test("test_cantrace")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|