TBS: device_manager

Signed-off-by: Babis Chalios <[email protected]>

Author: bchalios

src/vmm/src/arch/aarch64/fdt.rs

@@ -516,19 +516,21 @@ mod tests {
 
     #[test]
     fn test_create_fdt_with_vmgenid() {
-        let mut vmm = default_vmm();
-        let mem = vmm.vm.guest_memory().clone();
+        let mem = arch_mem(layout::FDT_MAX_SIZE + 0x1000);
+        let mut device_manager = default_device_manager();
+        let kvm = Kvm::new().unwrap();
+        let vm = kvm.create_vm().unwrap();
+        let gic = create_gic(&vm, 1, None).unwrap();
+        let mut cmdline = kernel_cmdline::Cmdline::new(4096).unwrap();
+        cmdline.insert("console", "/dev/tty0").unwrap();
 
-        let gic = create_gic(vmm.vm.fd(), 1, None).unwrap();
-        vmm.device_manager
-            .attach_vmgenid_device(&mem, vmm.vm.fd())
-            .unwrap();
+        device_manager.attach_vmgenid_device(&mem, &vm).unwrap();
 
         create_fdt(
             &mem,
             vec![0],
             CString::new("console=tty0").unwrap(),
-            &vmm.device_manager,
+            &device_manager,
             &gic,
             &None,
         )