Resource getDiskResource(const Megabytes& mb)
{
Resource diskResource;
switch (GetParam()) {
case NONE: {
diskResource = createDiskResource(
stringify(mb.megabytes()),
"role1",
None(),
None());
break;
}
case PATH: {
diskResource = createDiskResource(
stringify(mb.megabytes()),
"role1",
None(),
None(),
createDiskSourcePath(diskPath));
break;
}
}
return diskResource;
}
// Creates a disk with / without a `source` based on the
// parameterization of the test. `id` influences the `root` if one
// is specified so that we can create multiple disks in the tests.
Resource getDiskResource(const Megabytes& mb, size_t id = 1)
{
CHECK_LE(1u, id);
CHECK_GE(NUM_DISKS, id);
Resource diskResource;
switch (GetParam()) {
case NONE: {
diskResource = createDiskResource(
stringify(mb.megabytes()),
"role1",
None(),
None());
break;
}
case PATH: {
diskResource = createDiskResource(
stringify(mb.megabytes()),
"role1",
None(),
None(),
createDiskSourcePath(path::join(diskPath, "disk" + stringify(id))));
break;
}
case MOUNT: {
diskResource = createDiskResource(
stringify(mb.megabytes()),
"role1",
None(),
None(),
createDiskSourceMount(
path::join(diskPath, "disk" + stringify(id))));
break;
}
}
return diskResource;
}
