External/cloud-hypervisor
1
0
Fork 0
mirror of https://github.com/cloud-hypervisor/cloud-hypervisor.git synced 2025-03-07 17:26:14 +00:00
Code

tests: Port vhost_user_block tests to new methodology

Browse Source 
Signed-off-by: Rob Bradford <robert.bradford@intel.com>
This commit is contained in:
Rob Bradford 2020-08-12 12:20:14 +01:00
parent 8f4e1382a9
commit 23174800ca
1 changed files with 157 additions and 163 deletions

320
tests/integration.rs

@ -1188,72 +1188,72 @@ mod tests {
prepare_vhost_user_blk_daemon: Option<&PrepareBlkDaemon>, prepare_vhost_user_blk_daemon: Option<&PrepareBlkDaemon>,
self_spawned: bool, self_spawned: bool,
) { ) {
test_block!(tb, "", { let mut focal = UbuntuDiskConfig::new(FOCAL_IMAGE_NAME.to_string());
let mut focal = UbuntuDiskConfig::new(FOCAL_IMAGE_NAME.to_string()); let guest = Guest::new(&mut focal);
let guest = Guest::new(&mut focal); let api_socket = temp_api_path(&guest.tmp_dir);
let api_socket = temp_api_path(&guest.tmp_dir);
let mut workload_path = dirs::home_dir().unwrap(); let mut workload_path = dirs::home_dir().unwrap();
workload_path.push("workloads"); workload_path.push("workloads");
let kernel_path = direct_kernel_boot_path().unwrap(); let kernel_path = direct_kernel_boot_path().unwrap();
let (blk_params, daemon_child) = if self_spawned { let (blk_params, daemon_child) = if self_spawned {
let mut blk_file_path = workload_path; let mut blk_file_path = workload_path;
blk_file_path.push("blk.img"); blk_file_path.push("blk.img");
let blk_file_path = String::from(blk_file_path.to_str().unwrap()); let blk_file_path = String::from(blk_file_path.to_str().unwrap());
( (
format!( format!(
"vhost_user=true,path={},num_queues={},queue_size=128", "vhost_user=true,path={},num_queues={},queue_size=128",
blk_file_path, num_queues, blk_file_path, num_queues,
), ),
None, None,
)
} else {
let prepare_daemon = prepare_vhost_user_blk_daemon.unwrap();
// Start the daemon
let (daemon_child, vubd_socket_path) =
prepare_daemon(&guest.tmp_dir, "blk.img", num_queues, readonly, direct);
(
format!(
"vhost_user=true,socket={},num_queues={},queue_size=128",
vubd_socket_path, num_queues,
),
Some(daemon_child),
)
};
let mut child = GuestCommand::new(&guest)
.args(&["--cpus", format!("boot={}", num_queues).as_str()])
.args(&["--memory", "size=512M,hotplug_size=2048M,shared=on"])
.args(&["--kernel", kernel_path.to_str().unwrap()])
.args(&["--cmdline", DIRECT_KERNEL_BOOT_CMDLINE])
.args(&[
"--disk",
format!(
"path={}",
guest.disk_config.disk(DiskType::OperatingSystem).unwrap()
) )
} else { .as_str(),
let prepare_daemon = prepare_vhost_user_blk_daemon.unwrap(); format!(
// Start the daemon "path={}",
let (daemon_child, vubd_socket_path) = guest.disk_config.disk(DiskType::CloudInit).unwrap()
prepare_daemon(&guest.tmp_dir, "blk.img", num_queues, readonly, direct);
(
format!(
"vhost_user=true,socket={},num_queues={},queue_size=128",
vubd_socket_path, num_queues,
),
Some(daemon_child),
) )
}; .as_str(),
blk_params.as_str(),
])
.default_net()
.args(&["--api-socket", &api_socket])
.capture_output()
.spawn()
.unwrap();
let mut cloud_child = GuestCommand::new(&guest) thread::sleep(std::time::Duration::new(120, 0));
.args(&["--cpus", format!("boot={}", num_queues).as_str()])
.args(&["--memory", "size=512M,hotplug_size=2048M,shared=on"])
.args(&["--kernel", kernel_path.to_str().unwrap()])
.args(&["--cmdline", DIRECT_KERNEL_BOOT_CMDLINE])
.args(&[
"--disk",
format!(
"path={}",
guest.disk_config.disk(DiskType::OperatingSystem).unwrap()
)
.as_str(),
format!(
"path={}",
guest.disk_config.disk(DiskType::CloudInit).unwrap()
)
.as_str(),
blk_params.as_str(),
])
.default_net()
.args(&["--api-socket", &api_socket])
.spawn()
.unwrap();
thread::sleep(std::time::Duration::new(120, 0));
let r = std::panic::catch_unwind(|| {
// Check both if /dev/vdc exists and if the block size is 16M. // Check both if /dev/vdc exists and if the block size is 16M.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command("lsblk | grep vdc | grep -c 16M") .ssh_command("lsblk | grep vdc | grep -c 16M")
.unwrap_or_default() .unwrap_or_default()
@ -1264,8 +1264,7 @@ mod tests {
); );
// Check if this block is RO or RW. // Check if this block is RO or RW.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command("lsblk | grep vdc | awk '{print $5}'") .ssh_command("lsblk | grep vdc | awk '{print $5}'")
.unwrap_or_default() .unwrap_or_default()
@ -1277,8 +1276,7 @@ mod tests {
// Check if the number of queues in /sys/block/vdc/mq matches the // Check if the number of queues in /sys/block/vdc/mq matches the
// expected num_queues. // expected num_queues.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command("ls -ll /sys/block/vdc/mq | grep ^d | wc -l") .ssh_command("ls -ll /sys/block/vdc/mq | grep ^d | wc -l")
.unwrap_or_default() .unwrap_or_default()
@ -1290,19 +1288,20 @@ mod tests {
// Mount the device // Mount the device
let mount_ro_rw_flag = if readonly { "ro,noload" } else { "rw" }; let mount_ro_rw_flag = if readonly { "ro,noload" } else { "rw" };
guest.ssh_command("mkdir mount_image")?; guest.ssh_command("mkdir mount_image").unwrap();
guest.ssh_command( guest
format!( .ssh_command(
"sudo mount -o {} -t ext4 /dev/vdc mount_image/", format!(
mount_ro_rw_flag "sudo mount -o {} -t ext4 /dev/vdc mount_image/",
mount_ro_rw_flag
)
.as_str(),
) )
.as_str(), .unwrap();
)?;
// Check the content of the block device. The file "foo" should // Check the content of the block device. The file "foo" should
// contain "bar". // contain "bar".
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command("cat mount_image/foo") .ssh_command("cat mount_image/foo")
.unwrap_or_default() .unwrap_or_default()
@ -1326,12 +1325,11 @@ mod tests {
thread::sleep(std::time::Duration::new(10, 0)); thread::sleep(std::time::Duration::new(10, 0));
aver!(tb, guest.get_total_memory().unwrap_or_default() > 960_000); assert!(guest.get_total_memory().unwrap_or_default() > 960_000);
// Check again the content of the block device after the resize // Check again the content of the block device after the resize
// has been performed. // has been performed.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command("cat mount_image/foo") .ssh_command("cat mount_image/foo")
.unwrap_or_default() .unwrap_or_default()
@ -1341,20 +1339,20 @@ mod tests {
} }
// Unmount the device // Unmount the device
guest.ssh_command("sudo umount /dev/vdc")?; guest.ssh_command("sudo umount /dev/vdc").unwrap();
guest.ssh_command("rm -r mount_image")?; guest.ssh_command("rm -r mount_image").unwrap();
let _ = cloud_child.kill();
let _ = cloud_child.wait();
if let Some(mut daemon_child) = daemon_child {
thread::sleep(std::time::Duration::new(5, 0));
let _ = daemon_child.kill();
let _ = daemon_child.wait();
}
Ok(())
}); });
let _ = child.kill();
let output = child.wait_with_output().unwrap();
if let Some(mut daemon_child) = daemon_child {
thread::sleep(std::time::Duration::new(5, 0));
let _ = daemon_child.kill();
let _ = daemon_child.wait();
}
handle_child_output(r, &output);
} }
fn test_boot_from_vhost_user_blk( fn test_boot_from_vhost_user_blk(
@ -1364,75 +1362,72 @@ mod tests {
prepare_vhost_user_blk_daemon: Option<&PrepareBlkDaemon>, prepare_vhost_user_blk_daemon: Option<&PrepareBlkDaemon>,
self_spawned: bool, self_spawned: bool,
) { ) {
test_block!(tb, "", { let mut focal = UbuntuDiskConfig::new(FOCAL_IMAGE_NAME.to_string());
let mut focal = UbuntuDiskConfig::new(FOCAL_IMAGE_NAME.to_string()); let guest = Guest::new(&mut focal);
let guest = Guest::new(&mut focal);
let mut workload_path = dirs::home_dir().unwrap(); let mut workload_path = dirs::home_dir().unwrap();
workload_path.push("workloads"); workload_path.push("workloads");
let kernel_path = direct_kernel_boot_path().unwrap(); let kernel_path = direct_kernel_boot_path().unwrap();
let disk_path = guest let disk_path = guest
.disk_config .disk_config
.disk(DiskType::RawOperatingSystem) .disk(DiskType::RawOperatingSystem)
.unwrap(); .unwrap();
let (blk_boot_params, daemon_child) = if self_spawned { let (blk_boot_params, daemon_child) = if self_spawned {
( (
format!( format!(
"vhost_user=true,path={},num_queues={},queue_size=128", "vhost_user=true,path={},num_queues={},queue_size=128",
disk_path, num_queues, disk_path, num_queues,
), ),
None, None,
) )
} else { } else {
let prepare_daemon = prepare_vhost_user_blk_daemon.unwrap(); let prepare_daemon = prepare_vhost_user_blk_daemon.unwrap();
// Start the daemon // Start the daemon
let (daemon_child, vubd_socket_path) = prepare_daemon( let (daemon_child, vubd_socket_path) = prepare_daemon(
&guest.tmp_dir, &guest.tmp_dir,
disk_path.as_str(), disk_path.as_str(),
num_queues, num_queues,
readonly, readonly,
direct, direct,
);
(
format!(
"vhost_user=true,socket={},num_queues={},queue_size=128",
vubd_socket_path, num_queues,
),
Some(daemon_child),
)
};
let mut cloud_child = GuestCommand::new(&guest)
.args(&["--cpus", format!("boot={}", num_queues).as_str()])
.args(&["--memory", "size=512M,shared=on"])
.args(&["--kernel", kernel_path.to_str().unwrap()])
.args(&["--cmdline", DIRECT_KERNEL_BOOT_CMDLINE])
.args(&[
"--disk",
blk_boot_params.as_str(),
format!(
"path={}",
guest.disk_config.disk(DiskType::CloudInit).unwrap()
)
.as_str(),
])
.default_net()
.spawn()
.unwrap();
thread::sleep(std::time::Duration::new(40, 0));
// Just check the VM booted correctly.
aver_eq!(
tb,
guest.get_cpu_count().unwrap_or_default(),
num_queues as u32
); );
aver!(tb, guest.get_total_memory().unwrap_or_default() > 480_000);
(
format!(
"vhost_user=true,socket={},num_queues={},queue_size=128",
vubd_socket_path, num_queues,
),
Some(daemon_child),
)
};
let mut child = GuestCommand::new(&guest)
.args(&["--cpus", format!("boot={}", num_queues).as_str()])
.args(&["--memory", "size=512M,shared=on"])
.args(&["--kernel", kernel_path.to_str().unwrap()])
.args(&["--cmdline", DIRECT_KERNEL_BOOT_CMDLINE])
.args(&[
"--disk",
blk_boot_params.as_str(),
format!(
"path={}",
guest.disk_config.disk(DiskType::CloudInit).unwrap()
)
.as_str(),
])
.default_net()
.capture_output()
.spawn()
.unwrap();
thread::sleep(std::time::Duration::new(40, 0));
let r = std::panic::catch_unwind(|| {
// Just check the VM booted correctly.
assert_eq!(guest.get_cpu_count().unwrap_or_default(), num_queues as u32);
assert!(guest.get_total_memory().unwrap_or_default() > 480_000);
if self_spawned { if self_spawned {
// The reboot is not supported with mmio, so no reason to test it. // The reboot is not supported with mmio, so no reason to test it.
@ -1445,7 +1440,7 @@ mod tests {
.parse::<u32>() .parse::<u32>()
.unwrap_or(1); .unwrap_or(1);
aver_eq!(tb, reboot_count, 0); assert_eq!(reboot_count, 0);
guest.ssh_command("sudo reboot").unwrap_or_default(); guest.ssh_command("sudo reboot").unwrap_or_default();
thread::sleep(std::time::Duration::new(20, 0)); thread::sleep(std::time::Duration::new(20, 0));
@ -1455,21 +1450,20 @@ mod tests {
.trim() .trim()
.parse::<u32>() .parse::<u32>()
.unwrap_or_default(); .unwrap_or_default();
aver_eq!(tb, reboot_count, 1); assert_eq!(reboot_count, 1);
} }
} }
let _ = cloud_child.kill();
let _ = cloud_child.wait();
if let Some(mut daemon_child) = daemon_child {
thread::sleep(std::time::Duration::new(5, 0));
let _ = daemon_child.kill();
let _ = daemon_child.wait();
}
Ok(())
}); });
let _ = child.kill();
let output = child.wait_with_output().unwrap();
if let Some(mut daemon_child) = daemon_child {
thread::sleep(std::time::Duration::new(5, 0));
let _ = daemon_child.kill();
let _ = daemon_child.wait();
}
handle_child_output(r, &output);
} }
fn test_virtio_fs( fn test_virtio_fs(