diff options
| -rwxr-xr-x | bin/storage_test.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/bin/storage_test.py b/bin/storage_test.py index 3dbc87f8..dbc0552c 100755 --- a/bin/storage_test.py +++ b/bin/storage_test.py @@ -31,6 +31,9 @@ def find_largest_partition(device): blk_devs = [BlkDev(*p.strip().split()) for p in out.decode(sys.stdout.encoding).splitlines()] blk_devs[:] = [bd for bd in blk_devs if bd.type == 'part'] + if not blk_devs: + raise SystemExit( + 'ERROR: No partitions found on device {}'.format(device)) blk_devs.sort(key=lambda bd: int(bd.size)) return blk_devs[-1].name |