Skip to content

Commit d4394dd

Browse files
committed
refactor(test): use vm.create_vcpus() instead of open coding it
Instead of doing the irqchip dance inside unit testing code, just call vm.create_vcpus, which does it for us. Signed-off-by: Patrick Roy <[email protected]>
1 parent 9aa8357 commit d4394dd

File tree

2 files changed

+7
-12
lines changed

2 files changed

+7
-12
lines changed

src/vmm/src/device_manager/mmio.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -660,6 +660,7 @@ mod tests {
660660
}
661661

662662
#[test]
663+
#[cfg_attr(target_arch = "x86_64", allow(unused_mut))]
663664
fn test_register_virtio_device() {
664665
let start_addr1 = GuestAddress(0x0);
665666
let start_addr2 = GuestAddress(0x1000);
@@ -690,6 +691,7 @@ mod tests {
690691
}
691692

692693
#[test]
694+
#[cfg_attr(target_arch = "x86_64", allow(unused_mut))]
693695
fn test_register_too_many_devices() {
694696
let start_addr1 = GuestAddress(0x0);
695697
let start_addr2 = GuestAddress(0x1000);
@@ -745,6 +747,7 @@ mod tests {
745747
}
746748

747749
#[test]
750+
#[cfg_attr(target_arch = "x86_64", allow(unused_mut))]
748751
fn test_device_info() {
749752
let start_addr1 = GuestAddress(0x0);
750753
let start_addr2 = GuestAddress(0x1000);

src/vmm/src/vstate/vcpu/mod.rs

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -932,20 +932,12 @@ pub(crate) mod tests {
932932
pub(crate) fn setup_vcpu(mem_size: usize) -> (Kvm, Vm, Vcpu, GuestMemoryMmap) {
933933
let (kvm, mut vm, gm) = setup_vm_with_memory(mem_size);
934934

935-
let exit_evt = EventFd::new(libc::EFD_NONBLOCK).unwrap();
935+
let (mut vcpus, _) = vm.create_vcpus(1).unwrap();
936+
let mut vcpu = vcpus.remove(0);
936937

937938
#[cfg(target_arch = "aarch64")]
938-
let vcpu = {
939-
let mut vcpu = Vcpu::new(1, &vm, exit_evt).unwrap();
940-
vcpu.kvm_vcpu.init(&[]).unwrap();
941-
vm.setup_irqchip(1).unwrap();
942-
vcpu
943-
};
944-
#[cfg(target_arch = "x86_64")]
945-
let vcpu = {
946-
vm.setup_irqchip().unwrap();
947-
Vcpu::new(1, &vm, exit_evt).unwrap()
948-
};
939+
vcpu.kvm_vcpu.init(&[]).unwrap();
940+
949941
(kvm, vm, vcpu, gm)
950942
}
951943

0 commit comments

Comments
 (0)