| #!/usr/bin/env python3 |
| """ |
| # nvmept.py |
| # |
| # Test fio's io_uring_cmd ioengine with NVMe pass-through commands. |
| # |
| # USAGE |
| # see python3 nvmept.py --help |
| # |
| # EXAMPLES |
| # python3 t/nvmept.py --dut /dev/ng0n1 |
| # python3 t/nvmept.py --dut /dev/ng1n1 -f ./fio |
| # |
| # REQUIREMENTS |
| # Python 3.6 |
| # |
| """ |
| import os |
| import sys |
| import time |
| import argparse |
| from pathlib import Path |
| from fiotestlib import FioJobCmdTest, run_fio_tests |
| |
| |
| class PassThruTest(FioJobCmdTest): |
| """ |
| NVMe pass-through test class. Check to make sure output for selected data |
| direction(s) is non-zero and that zero data appears for other directions. |
| """ |
| |
| def setup(self, parameters): |
| """Setup a test.""" |
| |
| fio_args = [ |
| "--name=nvmept", |
| "--ioengine=io_uring_cmd", |
| "--cmd_type=nvme", |
| "--iodepth=8", |
| "--iodepth_batch=4", |
| "--iodepth_batch_complete=4", |
| f"--filename={self.fio_opts['filename']}", |
| f"--rw={self.fio_opts['rw']}", |
| f"--output={self.filenames['output']}", |
| f"--output-format={self.fio_opts['output-format']}", |
| ] |
| for opt in ['fixedbufs', 'nonvectored', 'force_async', 'registerfiles', |
| 'sqthread_poll', 'sqthread_poll_cpu', 'hipri', 'nowait', |
| 'time_based', 'runtime', 'verify', 'io_size']: |
| if opt in self.fio_opts: |
| option = f"--{opt}={self.fio_opts[opt]}" |
| fio_args.append(option) |
| |
| super().setup(fio_args) |
| |
| |
| def check_result(self): |
| if 'rw' not in self.fio_opts: |
| return |
| |
| if not self.passed: |
| return |
| |
| job = self.json_data['jobs'][0] |
| |
| if self.fio_opts['rw'] in ['read', 'randread']: |
| self.passed = self.check_all_ddirs(['read'], job) |
| elif self.fio_opts['rw'] in ['write', 'randwrite']: |
| if 'verify' not in self.fio_opts: |
| self.passed = self.check_all_ddirs(['write'], job) |
| else: |
| self.passed = self.check_all_ddirs(['read', 'write'], job) |
| elif self.fio_opts['rw'] in ['trim', 'randtrim']: |
| self.passed = self.check_all_ddirs(['trim'], job) |
| elif self.fio_opts['rw'] in ['readwrite', 'randrw']: |
| self.passed = self.check_all_ddirs(['read', 'write'], job) |
| elif self.fio_opts['rw'] in ['trimwrite', 'randtrimwrite']: |
| self.passed = self.check_all_ddirs(['trim', 'write'], job) |
| else: |
| print(f"Unhandled rw value {self.fio_opts['rw']}") |
| self.passed = False |
| |
| if job['iodepth_level']['8'] < 95: |
| print("Did not achieve requested iodepth") |
| self.passed = False |
| |
| |
| TEST_LIST = [ |
| { |
| "test_id": 1, |
| "fio_opts": { |
| "rw": 'read', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 2, |
| "fio_opts": { |
| "rw": 'randread', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 3, |
| "fio_opts": { |
| "rw": 'write', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 4, |
| "fio_opts": { |
| "rw": 'randwrite', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 5, |
| "fio_opts": { |
| "rw": 'trim', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 6, |
| "fio_opts": { |
| "rw": 'randtrim', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 7, |
| "fio_opts": { |
| "rw": 'write', |
| "io_size": 1024*1024, |
| "verify": "crc32c", |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 8, |
| "fio_opts": { |
| "rw": 'randwrite', |
| "io_size": 1024*1024, |
| "verify": "crc32c", |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 9, |
| "fio_opts": { |
| "rw": 'readwrite', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 10, |
| "fio_opts": { |
| "rw": 'randrw', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 11, |
| "fio_opts": { |
| "rw": 'trimwrite', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 12, |
| "fio_opts": { |
| "rw": 'randtrimwrite', |
| "timebased": 1, |
| "runtime": 3, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 13, |
| "fio_opts": { |
| "rw": 'randread', |
| "timebased": 1, |
| "runtime": 3, |
| "fixedbufs": 1, |
| "nonvectored": 1, |
| "force_async": 1, |
| "registerfiles": 1, |
| "sqthread_poll": 1, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| "test_id": 14, |
| "fio_opts": { |
| "rw": 'randwrite', |
| "timebased": 1, |
| "runtime": 3, |
| "fixedbufs": 1, |
| "nonvectored": 1, |
| "force_async": 1, |
| "registerfiles": 1, |
| "sqthread_poll": 1, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| { |
| # We can't enable fixedbufs because for trim-only |
| # workloads fio actually does not allocate any buffers |
| "test_id": 15, |
| "fio_opts": { |
| "rw": 'randtrim', |
| "timebased": 1, |
| "runtime": 3, |
| "fixedbufs": 0, |
| "nonvectored": 1, |
| "force_async": 1, |
| "registerfiles": 1, |
| "sqthread_poll": 1, |
| "output-format": "json", |
| }, |
| "test_class": PassThruTest, |
| }, |
| ] |
| |
| def parse_args(): |
| """Parse command-line arguments.""" |
| |
| parser = argparse.ArgumentParser() |
| parser.add_argument('-f', '--fio', help='path to file executable (e.g., ./fio)') |
| parser.add_argument('-a', '--artifact-root', help='artifact root directory') |
| parser.add_argument('-s', '--skip', nargs='+', type=int, |
| help='list of test(s) to skip') |
| parser.add_argument('-o', '--run-only', nargs='+', type=int, |
| help='list of test(s) to run, skipping all others') |
| parser.add_argument('--dut', help='target NVMe character device to test ' |
| '(e.g., /dev/ng0n1). WARNING: THIS IS A DESTRUCTIVE TEST', required=True) |
| args = parser.parse_args() |
| |
| return args |
| |
| |
| def main(): |
| """Run tests using fio's io_uring_cmd ioengine to send NVMe pass through commands.""" |
| |
| args = parse_args() |
| |
| artifact_root = args.artifact_root if args.artifact_root else \ |
| f"nvmept-test-{time.strftime('%Y%m%d-%H%M%S')}" |
| os.mkdir(artifact_root) |
| print(f"Artifact directory is {artifact_root}") |
| |
| if args.fio: |
| fio_path = str(Path(args.fio).absolute()) |
| else: |
| fio_path = 'fio' |
| print(f"fio path is {fio_path}") |
| |
| for test in TEST_LIST: |
| test['fio_opts']['filename'] = args.dut |
| |
| test_env = { |
| 'fio_path': fio_path, |
| 'fio_root': str(Path(__file__).absolute().parent.parent), |
| 'artifact_root': artifact_root, |
| 'basename': 'readonly', |
| } |
| |
| _, failed, _ = run_fio_tests(TEST_LIST, test_env, args) |
| sys.exit(failed) |
| |
| |
| if __name__ == '__main__': |
| main() |
