mirror of https://github.com/python/cpython
63 lines
948 B
Python
63 lines
948 B
Python
# Test the atexit module.
|
|
from test_support import TESTFN, vereq
|
|
import atexit
|
|
import os
|
|
import sys
|
|
|
|
input = """\
|
|
import atexit
|
|
|
|
def handler1():
|
|
print "handler1"
|
|
|
|
def handler2(*args, **kargs):
|
|
print "handler2", args, kargs
|
|
|
|
atexit.register(handler1)
|
|
atexit.register(handler2)
|
|
atexit.register(handler2, 7, kw="abc")
|
|
"""
|
|
|
|
fname = TESTFN + ".py"
|
|
f = file(fname, "w")
|
|
f.write(input)
|
|
f.close()
|
|
|
|
p = os.popen("%s %s" % (sys.executable, fname))
|
|
output = p.read()
|
|
p.close()
|
|
vereq(output, """\
|
|
handler2 (7,) {'kw': 'abc'}
|
|
handler2 () {}
|
|
handler1
|
|
""")
|
|
|
|
input = """\
|
|
def direct():
|
|
print "direct exit"
|
|
|
|
import sys
|
|
sys.exitfunc = direct
|
|
|
|
# Make sure atexit doesn't drop
|
|
def indirect():
|
|
print "indirect exit"
|
|
|
|
import atexit
|
|
atexit.register(indirect)
|
|
"""
|
|
|
|
f = file(fname, "w")
|
|
f.write(input)
|
|
f.close()
|
|
|
|
p = os.popen("%s %s" % (sys.executable, fname))
|
|
output = p.read()
|
|
p.close()
|
|
vereq(output, """\
|
|
indirect exit
|
|
direct exit
|
|
""")
|
|
|
|
os.unlink(fname)
|