iotests: create generic qemu_tool() function

reimplement qemu_img() in terms of qemu_tool() in preparation for doing
the same with qemu_io().

Signed-off-by: John Snow <jsnow@redhat.com>
Reviewed-by: Eric Blake <eblake@redhat.com>
Reviewed-by: Hanna Reitz <hreitz@redhat.com>
Message-Id: <20220418211504.943969-7-jsnow@redhat.com>
Signed-off-by: Hanna Reitz <hreitz@redhat.com>
This commit is contained in:
John Snow 2022-04-18 17:14:58 -04:00 committed by Hanna Reitz
parent 4897629173
commit b2d68a8e56

View file

@ -206,15 +206,13 @@ def qemu_img_create_prepare_args(args: List[str]) -> List[str]:
return result return result
def qemu_img(*args: str, check: bool = True, combine_stdio: bool = True
def qemu_tool(*args: str, check: bool = True, combine_stdio: bool = True
) -> 'subprocess.CompletedProcess[str]': ) -> 'subprocess.CompletedProcess[str]':
""" """
Run qemu_img and return the status code and console output. Run a qemu tool and return its status code and console output.
This function always prepends QEMU_IMG_OPTIONS and may further alter :param args: full command line to run.
the args for 'create' commands.
:param args: command-line arguments to qemu-img.
:param check: Enforce a return code of zero. :param check: Enforce a return code of zero.
:param combine_stdio: set to False to keep stdout/stderr separated. :param combine_stdio: set to False to keep stdout/stderr separated.
@ -231,10 +229,8 @@ def qemu_img(*args: str, check: bool = True, combine_stdio: bool = True
properties. If streams are not combined, it will also have a properties. If streams are not combined, it will also have a
stderr property. stderr property.
""" """
full_args = qemu_img_args + qemu_img_create_prepare_args(list(args))
subp = subprocess.run( subp = subprocess.run(
full_args, args,
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.STDOUT if combine_stdio else subprocess.PIPE, stderr=subprocess.STDOUT if combine_stdio else subprocess.PIPE,
universal_newlines=True, universal_newlines=True,
@ -243,7 +239,7 @@ def qemu_img(*args: str, check: bool = True, combine_stdio: bool = True
if check and subp.returncode or (subp.returncode < 0): if check and subp.returncode or (subp.returncode < 0):
raise VerboseProcessError( raise VerboseProcessError(
subp.returncode, full_args, subp.returncode, args,
output=subp.stdout, output=subp.stdout,
stderr=subp.stderr, stderr=subp.stderr,
) )
@ -251,6 +247,20 @@ def qemu_img(*args: str, check: bool = True, combine_stdio: bool = True
return subp return subp
def qemu_img(*args: str, check: bool = True, combine_stdio: bool = True
) -> 'subprocess.CompletedProcess[str]':
"""
Run QEMU_IMG_PROG and return its status code and console output.
This function always prepends QEMU_IMG_OPTIONS and may further alter
the args for 'create' commands.
See `qemu_tool()` for greater detail.
"""
full_args = qemu_img_args + qemu_img_create_prepare_args(list(args))
return qemu_tool(*full_args, check=check, combine_stdio=combine_stdio)
def ordered_qmp(qmsg, conv_keys=True): def ordered_qmp(qmsg, conv_keys=True):
# Dictionaries are not ordered prior to 3.6, therefore: # Dictionaries are not ordered prior to 3.6, therefore:
if isinstance(qmsg, list): if isinstance(qmsg, list):