On Fri, Feb 16, 2018 at 02:46:35PM -0600, Eric Blake wrote: > On 02/15/2018 07:51 AM, Vladimir Sementsov-Ogievskiy wrote: > >Simple way to have auto generated filenames with auto clenup. Like > > s/clenup/cleanup/ > > >FilePath but without using 'with' statement and without additional > >indentation of the whole test. > > > >Signed-off-by: Vladimir Sementsov-Ogievskiy <vsement...@virtuozzo.com> > >--- > > tests/qemu-iotests/iotests.py | 32 ++++++++++++++++++++++++++++++++ > > 1 file changed, 32 insertions(+) > > Jeff, where do we stand on your iotests cleanups? Is this something you > want to use? >
This patch has niceties that my patch does not, in that explicit clean-up is not required. This is also a slight design difference; my patch series explicitly avoided auto-cleanup, so that ./check could handle it, or (optionally) leave behind all the mouse droppings for later analysis, in the case of test failure. Ideally, I'd love clean-up to be optional; if a test fails, I'd like to have the option of examining all the intermediate files and output of the test (and once it resides in its own subdir, that is also side-affect free for the other tests). > > > >diff --git a/tests/qemu-iotests/iotests.py b/tests/qemu-iotests/iotests.py > >index c1302a2f9b..f2d05ca3fd 100644 > >--- a/tests/qemu-iotests/iotests.py > >+++ b/tests/qemu-iotests/iotests.py > >@@ -27,6 +27,7 @@ import struct > > import json > > import signal > > import logging > >+import atexit > > sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', > > 'scripts')) > > import qtest > >@@ -250,6 +251,37 @@ class FilePath(object): > > return False > >+def file_path_remover(): > >+ for path in reversed(file_path_remover.paths): > >+ try: > >+ os.remove(path) > >+ except OSError: > >+ pass > >+ > >+ > >+def file_path(*names): > >+ ''' Another way to get auto-generated filename that cleans itself up. > >+ > >+ Use it as simple as: > >+ > >+ img_a, img_b = file_path('a.img', 'b.img') > >+ sock = file_path('socket') > >+ ''' > >+ > >+ if not hasattr(file_path_remover, 'paths'): > >+ file_path_remover.paths = [] > >+ atexit.register(file_path_remover) > >+ > >+ paths = [] > >+ for name in names: > >+ filename = '{0}-{1}'.format(os.getpid(), name) > >+ path = os.path.join(test_dir, filename) > >+ file_path_remover.paths.append(path) > >+ paths.append(path) > >+ > >+ return paths[0] if len(paths) == 1 else paths > >+ > >+ > > class VM(qtest.QEMUQtestMachine): > > '''A QEMU VM''' > > > > -- > Eric Blake, Principal Software Engineer > Red Hat, Inc. +1-919-301-3266 > Virtualization: qemu.org | libvirt.org >