unittests: use UnitTest.addCleanup a bit

pull/9628/head
Dylan Baker 3 years ago committed by Nirbheek Chauhan
parent 905a361ca0
commit 457e2d9812
  1. 21
      unittests/allplatformstests.py

@ -1499,19 +1499,22 @@ class AllPlatformTests(BasePlatformTests):
else:
shlibfile = os.path.join(tdir, 'libalexandria.' + shared_suffix)
self.build_shared_lib(cc, source, objectfile, shlibfile, impfile)
if is_windows():
def cleanup() -> None:
"""Clean up all the garbage MSVC writes in the source tree."""
for fname in glob(os.path.join(tdir, 'alexandria.*')):
if os.path.splitext(fname)[1] not in {'.c', '.h'}:
os.unlink(fname)
self.addCleanup(cleanup)
else:
self.addCleanup(os.unlink, shlibfile)
# Run the test
try:
self.init(tdir)
self.build()
self.run_tests()
finally:
os.unlink(shlibfile)
if is_windows():
# Clean up all the garbage MSVC writes in the
# source tree.
for fname in glob(os.path.join(tdir, 'alexandria.*')):
if os.path.splitext(fname)[1] not in ['.c', '.h']:
os.unlink(fname)
@skipIfNoPkgconfig
def test_pkgconfig_static(self):

Loading…
Cancel
Save