tests/qemu-iotests/testenv.py

3 # Copyright (c) 2020-2021 Virtuozzo International GmbH
34 def isxfile(path: str) -> bool:
38 def get_default_machine(qemu_prog: str) -> str:
39     outp = subprocess.run([qemu_prog, '-machine', 'help'], check=True,
63     lower-cased TestEnv attributes.
68     # pylint: disable=too-many-instance-attributes
81     def prepare_subprocess(self, args: List[str]) -> Dict[str, str]:
83             args.append('-d')
85         with open(args[0], encoding="utf-8") as f:
96     def get_env(self) -> Dict[str, str]:
105     def init_directories(self) -> None:
131             self.sock_dir = tempfile.mkdtemp(prefix="qemu-iotests-")
138     def init_binaries(self) -> None:
145         def root(*names: str) -> str:
152         self.qemu_prog = os.getenv('QEMU_PROG', root(f'qemu-system-{arch}'))
154             pattern = root('qemu-system-*')
162         self.qemu_img_prog = os.getenv('QEMU_IMG_PROG', root('qemu-img'))
163         self.qemu_io_prog = os.getenv('QEMU_IO_PROG', root('qemu-io'))
164         self.qemu_nbd_prog = os.getenv('QEMU_NBD_PROG', root('qemu-nbd'))
165         self.qsd_prog = os.getenv('QSD_PROG', root('storage-daemon',
166                                                    'qemu-storage-daemon'))
184                  dry_run: bool = False) -> None:
234         self.qemu_options = '-nodefaults -display none -accel qtest'
243             ('rx', 'gdbsim-r5f562n8'),
249             if self.qemu_prog.endswith(f'qemu-system-{suffix}'):
250                 self.qemu_options += f' -machine {machine}'
258         is_generic = self.imgfmt not in ['bochs', 'cloop', 'dmg', 'vvfat']
261         self.qemu_io_options = f'--cache {self.cachemode} --aio {self.aiomode}'
263             self.qemu_io_options += ' --misalign'
271                 self.imgopts = 'iter-time=10'
272             elif 'iter-time=' not in self.imgopts:
273                 self.imgopts += ',iter-time=10'
276             self.qemu_io_options += ' -f ' + self.imgfmt
284     def close(self) -> None:
288     def __enter__(self) -> 'TestEnv':
291     def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
294     def print_env(self, prefix: str = '') -> None:
296 {prefix}QEMU          -- "{QEMU_PROG}" {QEMU_OPTIONS}
297 {prefix}QEMU_IMG      -- "{QEMU_IMG_PROG}" {QEMU_IMG_OPTIONS}
298 {prefix}QEMU_IO       -- "{QEMU_IO_PROG}" {QEMU_IO_OPTIONS}
299 {prefix}QEMU_NBD      -- "{QEMU_NBD_PROG}" {QEMU_NBD_OPTIONS}
300 {prefix}IMGFMT        -- {IMGFMT}{imgopts}
301 {prefix}IMGPROTO      -- {IMGPROTO}
302 {prefix}PLATFORM      -- {platform}
303 {prefix}TEST_DIR      -- {TEST_DIR}
304 {prefix}SOCK_DIR      -- {SOCK_DIR}
305 {prefix}GDB_OPTIONS   -- {GDB_OPTIONS}
306 {prefix}VALGRIND_QEMU -- {VALGRIND_QEMU}
307 {prefix}PRINT_QEMU_OUTPUT -- {PRINT_QEMU}