tests/qemu-iotests/testenv.py

3 # Copyright (c) 2020-2021 Virtuozzo International GmbH
37 def isxfile(path: str) -> bool:
41 def get_default_machine(qemu_prog: str) -> str:
42     outp = subprocess.run([qemu_prog, '-machine', 'help'], check=True,
66     lower-cased TestEnv attributes.
71     # pylint: disable=too-many-instance-attributes
84     def prepare_subprocess(self, args: List[str]) -> Dict[str, str]:
86             args.append('-d')
88         with open(args[0], encoding="utf-8") as f:
99     def get_env(self) -> Dict[str, str]:
108     def init_directories(self) -> None:
134             self.sock_dir = tempfile.mkdtemp(prefix="qemu-iotests-")
141     def init_binaries(self) -> None:
148         def root(*names: str) -> str:
155         self.qemu_prog = os.getenv('QEMU_PROG', root(f'qemu-system-{arch}'))
157             pattern = root('qemu-system-*')
165         self.qemu_img_prog = os.getenv('QEMU_IMG_PROG', root('qemu-img'))
166         self.qemu_io_prog = os.getenv('QEMU_IO_PROG', root('qemu-io'))
167         self.qemu_nbd_prog = os.getenv('QEMU_NBD_PROG', root('qemu-nbd'))
168         self.qsd_prog = os.getenv('QSD_PROG', root('storage-daemon',
169                                                    'qemu-storage-daemon'))
187                  dry_run: bool = False) -> None:
237         self.qemu_options = '-nodefaults -display none -accel qtest'
246             ('rx', 'gdbsim-r5f562n8'),
252             if self.qemu_prog.endswith(f'qemu-system-{suffix}'):
253                 self.qemu_options += f' -machine {machine}'
261         is_generic = self.imgfmt not in ['bochs', 'cloop', 'dmg', 'vvfat']
264         self.qemu_io_options = f'--cache {self.cachemode} --aio {self.aiomode}'
266             self.qemu_io_options += ' --misalign'
274                 self.imgopts = 'iter-time=10'
275             elif 'iter-time=' not in self.imgopts:
276                 self.imgopts += ',iter-time=10'
279             self.qemu_io_options += ' -f ' + self.imgfmt
287     def close(self) -> None:
291     def __enter__(self) -> 'TestEnv':
294     def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
297     def print_env(self, prefix: str = '') -> None:
299 {prefix}QEMU          -- "{QEMU_PROG}" {QEMU_OPTIONS}
300 {prefix}QEMU_IMG      -- "{QEMU_IMG_PROG}" {QEMU_IMG_OPTIONS}
301 {prefix}QEMU_IO       -- "{QEMU_IO_PROG}" {QEMU_IO_OPTIONS}
302 {prefix}QEMU_NBD      -- "{QEMU_NBD_PROG}" {QEMU_NBD_OPTIONS}
303 {prefix}IMGFMT        -- {IMGFMT}{imgopts}
304 {prefix}IMGPROTO      -- {IMGPROTO}
305 {prefix}PLATFORM      -- {platform}
306 {prefix}TEST_DIR      -- {TEST_DIR}
307 {prefix}SOCK_DIR      -- {SOCK_DIR}
308 {prefix}GDB_OPTIONS   -- {GDB_OPTIONS}
309 {prefix}VALGRIND_QEMU -- {VALGRIND_QEMU}
310 {prefix}PRINT_QEMU_OUTPUT -- {PRINT_QEMU}