diff options
| author | Pablo Galindo Salgado <Pablogsal@gmail.com> | 2021-12-09 13:53:44 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-12-09 13:53:44 (GMT) |
| commit | f0d290d25cad66e615ada68ba190b8a23ac1deaa (patch) | |
| tree | 3a95b2b47cd9b56b49e0a3ef4645db3e4b6087a3 /Lib/test/_test_atexit.py | |
| parent | af6b4068859a5d0c8afd696f3c0c0155660211a4 (diff) | |
| download | cpython-f0d290d25cad66e615ada68ba190b8a23ac1deaa.zip cpython-f0d290d25cad66e615ada68ba190b8a23ac1deaa.tar.gz cpython-f0d290d25cad66e615ada68ba190b8a23ac1deaa.tar.bz2 | |
bpo-46025: Fix a crash in the atexit module for auto-unregistering functions (GH-30002)
Diffstat (limited to 'Lib/test/_test_atexit.py')
| -rw-r--r-- | Lib/test/_test_atexit.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/Lib/test/_test_atexit.py b/Lib/test/_test_atexit.py index a316585..55d2808 100644 --- a/Lib/test/_test_atexit.py +++ b/Lib/test/_test_atexit.py @@ -116,6 +116,21 @@ class GeneralTest(unittest.TestCase): atexit._run_exitfuncs() self.assertEqual(l, [5]) + def test_atexit_with_unregistered_function(self): + # See bpo-46025 for more info + def func(): + atexit.unregister(func) + 1/0 + atexit.register(func) + try: + with support.catch_unraisable_exception() as cm: + atexit._run_exitfuncs() + self.assertEqual(cm.unraisable.object, func) + self.assertEqual(cm.unraisable.exc_type, ZeroDivisionError) + self.assertEqual(type(cm.unraisable.exc_value), ZeroDivisionError) + finally: + atexit.unregister(func) + if __name__ == "__main__": unittest.main() |