// SPDX-License-Identifier: GPL-2.0+ /* * Copyright (C) 2020 Sean Anderson */ #include #include #include #include #include #include #include static int dm_test_part(struct unit_test_state *uts) { char str_disk_guid[UUID_STR_LEN + 1]; struct blk_desc *mmc_dev_desc; struct disk_partition part_info; struct disk_partition parts[2] = { { .start = 48, /* GPT data takes up the first 34 blocks or so */ .size = 1, .name = "test1", }, { .start = 49, .size = 1, .name = "test2", }, }; ut_asserteq(1, blk_get_device_by_str("mmc", "1", &mmc_dev_desc)); if (CONFIG_IS_ENABLED(RANDOM_UUID)) { gen_rand_uuid_str(parts[0].uuid, UUID_STR_FORMAT_STD); gen_rand_uuid_str(parts[1].uuid, UUID_STR_FORMAT_STD); gen_rand_uuid_str(str_disk_guid, UUID_STR_FORMAT_STD); } ut_assertok(gpt_restore(mmc_dev_desc, str_disk_guid, parts, ARRAY_SIZE(parts))); #define test(expected, part_str, whole) \ ut_asserteq(expected, \ part_get_info_by_dev_and_name_or_num("mmc", part_str, \ &mmc_dev_desc, \ &part_info, whole)) test(-ENODEV, "", true); env_set("bootdevice", "0"); test(0, "", true); env_set("bootdevice", "1"); test(1, "", false); test(1, "-", false); env_set("bootdevice", ""); test(-EPROTONOSUPPORT, "0", false); test(0, "0", true); test(0, ":0", true); test(0, ".0", true); test(0, ".0:0", true); test(-EINVAL, "#test1", true); test(1, "1", false); test(1, "1", true); test(-ENOENT, "1:0", false); test(0, "1:0", true); test(1, "1:1", false); test(2, "1:2", false); test(1, "1.0", false); test(0, "1.0:0", true); test(1, "1.0:1", false); test(2, "1.0:2", false); test(-EINVAL, "1#bogus", false); test(1, "1#test1", false); test(2, "1#test2", false); return 0; } DM_TEST(dm_test_part, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);