tests: Port test_vfio to new test methodology

Now the testing aspects are run inside a panic handler block rather than
inside a credibility TestBlock. If the test fails then the output from
the cloud-hypervisor binary is then presented.

Signed-off-by: Rob Bradford <robert.bradford@intel.com>
This commit is contained in:
Rob Bradford 2020-08-12 10:18:31 +01:00 committed by Sebastien Boeuf
parent 4ef702ea76
commit d749aa2e6b

View File

@ -3237,7 +3237,6 @@ mod tests {
// The third device is added to validate that hotplug works correctly since // The third device is added to validate that hotplug works correctly since
// it is being added to the L2 VM through hotplugging mechanism. // it is being added to the L2 VM through hotplugging mechanism.
fn test_vfio() { fn test_vfio() {
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_from_ip_range(&mut focal, "172.17", 0); let guest = Guest::new_from_ip_range(&mut focal, "172.17", 0);
@ -3327,12 +3326,14 @@ mod tests {
) )
.as_str(), .as_str(),
]) ])
.capture_output()
.spawn() .spawn()
.unwrap(); .unwrap();
thread::sleep(std::time::Duration::new(30, 0)); thread::sleep(std::time::Duration::new(30, 0));
guest.ssh_command_l1("sudo systemctl start vfio")?; let r = std::panic::catch_unwind(|| {
guest.ssh_command_l1("sudo systemctl start vfio").unwrap();
thread::sleep(std::time::Duration::new(120, 0)); thread::sleep(std::time::Duration::new(120, 0));
// We booted our cloud hypervisor L2 guest with a "VFIOTAG" tag // We booted our cloud hypervisor L2 guest with a "VFIOTAG" tag
@ -3340,8 +3341,7 @@ mod tests {
// Let's ssh into it and verify that it's there. If it is it means // Let's ssh into it and verify that it's there. If it is it means
// we're in the right guest (The L2 one) because the QEMU L1 guest // we're in the right guest (The L2 one) because the QEMU L1 guest
// does not have this command line tag. // does not have this command line tag.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command_l2_1("grep -c VFIOTAG /proc/cmdline") .ssh_command_l2_1("grep -c VFIOTAG /proc/cmdline")
.unwrap_or_default() .unwrap_or_default()
@ -3353,8 +3353,7 @@ mod tests {
// Let's also verify from the second virtio-net device passed to // Let's also verify from the second virtio-net device passed to
// the L2 VM. // the L2 VM.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command_l2_2("grep -c VFIOTAG /proc/cmdline") .ssh_command_l2_2("grep -c VFIOTAG /proc/cmdline")
.unwrap_or_default() .unwrap_or_default()
@ -3365,8 +3364,7 @@ mod tests {
); );
// Check the amount of PCI devices appearing in L2 VM. // Check the amount of PCI devices appearing in L2 VM.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command_l2_1("ls /sys/bus/pci/devices | wc -l") .ssh_command_l2_1("ls /sys/bus/pci/devices | wc -l")
.unwrap_or_default() .unwrap_or_default()
@ -3379,17 +3377,20 @@ mod tests {
// Hotplug an extra virtio-net device through L2 VM. // Hotplug an extra virtio-net device through L2 VM.
guest.ssh_command_l1( guest.ssh_command_l1(
"echo 0000:00:08.0 | sudo tee /sys/bus/pci/devices/0000:00:08.0/driver/unbind", "echo 0000:00:08.0 | sudo tee /sys/bus/pci/devices/0000:00:08.0/driver/unbind",
)?; ).unwrap();
guest.ssh_command_l1( guest
.ssh_command_l1(
"echo 1af4 1041 | sudo tee /sys/bus/pci/drivers/vfio-pci/new_id", "echo 1af4 1041 | sudo tee /sys/bus/pci/drivers/vfio-pci/new_id",
)?; )
let vfio_hotplug_output = guest.ssh_command_l1( .unwrap();
let vfio_hotplug_output = guest
.ssh_command_l1(
"sudo /mnt/ch-remote \ "sudo /mnt/ch-remote \
--api-socket=/tmp/ch_api.sock \ --api-socket=/tmp/ch_api.sock \
add-device path=/sys/bus/pci/devices/0000:00:08.0,id=vfio123", add-device path=/sys/bus/pci/devices/0000:00:08.0,id=vfio123",
)?; )
aver!( .unwrap();
tb, assert!(
vfio_hotplug_output.contains("{\"id\":\"vfio123\",\"bdf\":\"0000:00:07.0\"}") vfio_hotplug_output.contains("{\"id\":\"vfio123\",\"bdf\":\"0000:00:07.0\"}")
); );
@ -3398,8 +3399,7 @@ mod tests {
// Let's also verify from the third virtio-net device passed to // Let's also verify from the third virtio-net device passed to
// the L2 VM. This third device has been hotplugged through the L2 // the L2 VM. This third device has been hotplugged through the L2
// VM, so this is our way to validate hotplug works for VFIO PCI. // VM, so this is our way to validate hotplug works for VFIO PCI.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command_l2_3("grep -c VFIOTAG /proc/cmdline") .ssh_command_l2_3("grep -c VFIOTAG /proc/cmdline")
.unwrap_or_default() .unwrap_or_default()
@ -3412,8 +3412,7 @@ mod tests {
// Check the amount of PCI devices appearing in L2 VM. // Check the amount of PCI devices appearing in L2 VM.
// There should be one more device than before, raising the count // There should be one more device than before, raising the count
// up to 8 PCI devices. // up to 8 PCI devices.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command_l2_1("ls /sys/bus/pci/devices | wc -l") .ssh_command_l2_1("ls /sys/bus/pci/devices | wc -l")
.unwrap_or_default() .unwrap_or_default()
@ -3426,17 +3425,18 @@ mod tests {
// Let's now verify that we can correctly remove the virtio-net // Let's now verify that we can correctly remove the virtio-net
// device through the "remove-device" command responsible for // device through the "remove-device" command responsible for
// unplugging VFIO devices. // unplugging VFIO devices.
guest.ssh_command_l1( guest
.ssh_command_l1(
"sudo /mnt/ch-remote \ "sudo /mnt/ch-remote \
--api-socket=/tmp/ch_api.sock \ --api-socket=/tmp/ch_api.sock \
remove-device vfio123", remove-device vfio123",
)?; )
.unwrap();
thread::sleep(std::time::Duration::new(10, 0)); thread::sleep(std::time::Duration::new(10, 0));
// Check the amount of PCI devices appearing in L2 VM is back down // Check the amount of PCI devices appearing in L2 VM is back down
// to 7 devices. // to 7 devices.
aver_eq!( assert_eq!(
tb,
guest guest
.ssh_command_l2_1("ls /sys/bus/pci/devices | wc -l") .ssh_command_l2_1("ls /sys/bus/pci/devices | wc -l")
.unwrap_or_default() .unwrap_or_default()
@ -3450,28 +3450,24 @@ mod tests {
// up as expected. In order to check, we will use the virtio-net // up as expected. In order to check, we will use the virtio-net
// device already passed through L2 as a VFIO device, this will // device already passed through L2 as a VFIO device, this will
// verify that VFIO devices are functional with memory hotplug. // verify that VFIO devices are functional with memory hotplug.
aver!( assert!(guest.get_total_memory_l2().unwrap_or_default() > 480_000);
tb,
guest.get_total_memory_l2().unwrap_or_default() > 480_000
);
guest.ssh_command_l2_1( guest.ssh_command_l2_1(
"sudo bash -c 'echo online > /sys/devices/system/memory/auto_online_blocks'", "sudo bash -c 'echo online > /sys/devices/system/memory/auto_online_blocks'",
)?; ).unwrap();
guest.ssh_command_l1( guest
.ssh_command_l1(
"sudo /mnt/ch-remote \ "sudo /mnt/ch-remote \
--api-socket=/tmp/ch_api.sock \ --api-socket=/tmp/ch_api.sock \
resize --memory=1073741824", resize --memory=1073741824",
)?; )
aver!( .unwrap();
tb, assert!(guest.get_total_memory_l2().unwrap_or_default() > 960_000);
guest.get_total_memory_l2().unwrap_or_default() > 960_000 });
);
let _ = child.kill(); let _ = child.kill();
let _ = child.wait(); let output = child.wait_with_output().unwrap();
Ok(()) handle_child_output(r, &output);
});
} }
#[cfg_attr(feature = "mmio", test)] #[cfg_attr(feature = "mmio", test)]