diff options
Diffstat (limited to 'tests')
| -rwxr-xr-x | tests/setup_disk_test | 34 |
1 files changed, 32 insertions, 2 deletions
diff --git a/tests/setup_disk_test b/tests/setup_disk_test index f935796..1fee4aa 100755 --- a/tests/setup_disk_test +++ b/tests/setup_disk_test @@ -8,6 +8,7 @@ init_tests \ setup_disk_func_find_disks \ setup_disk_func_setup_partitions_dos \ setup_disk_func_setup_partitions_gpt \ + setup_disk_func_find_efi_size \ setup_disk_non_existing_block_dev setup_disk_usage_body() { @@ -27,10 +28,20 @@ setup_disk_none_body() { } fake_disk() { - mkdir -p sys/block/$1/device \ - sys/block/$1/holders + local dev="$1" + local size="${2:-1000}" # size in MiB + local blocksize="${3:-512}" + + mkdir -p dev \ + sys/block/$dev/device \ + sys/block/$dev/holders \ + sys/block/$dev/queue + + echo $(( $size * 1000000 / $blocksize )) > sys/block/$dev/size + echo $(( $blocksize )) > sys/block/$dev/queue/logical_block_size } + fake_partition() { mkdir -p sys/block/$1/$1$2/holders } @@ -160,6 +171,25 @@ setup_disk_func_setup_partitions_gpt_body() { setup-disk dev/vda "lvm" } +setup_disk_func_find_efi_size_body() { + init_env + # size:blocksize:expected + for i in 10000::512 4000::264 1000::160 \ + 10000:4096:512 4000:4096:264 1000:4096:264; do + + local diskdata=${i%:*} + local disksize=${diskdata%:*} + local blocksize=${diskdata#*:} + local expected=${i##*:} + + fake_disk vda $disksize $blocksize + SETUP_DISK_TESTFUNC=find_efi_size \ + atf_check -s exit:0 \ + -o match:"^$expected$" \ + setup-disk /dev/vda + done +} + setup_disk_non_existing_block_dev_body() { init_env atf_check -s not-exit:0 \ |