Add memory option to test framework VM config

Author: dlon

test/test-manager/src/config/vm.rs

@@ -65,6 +65,11 @@ pub struct VmConfig {
     #[serde(default)]
     #[arg(long)]
     pub vcpus: Option<usize>,
+
+    /// Amount of memory, in MBs
+    #[serde(default)]
+    #[arg(long)]
+    pub memory: Option<usize>,
 }
 
 impl VmConfig {

test/test-manager/src/vm/qemu.rs

@@ -28,6 +28,9 @@ const OBTAIN_IP_TIMEOUT: Duration = Duration::from_secs(180);
 /// Default number of VCPU cores (passed to -smp)
 const DEFAULT_NUM_VCPUS: usize = 2;
 
+/// Default amount of memory, in MBs (passed to -m)
+const DEFAULT_AMOUNT_MEMORY: usize = 2048;
+
 #[derive(thiserror::Error, Debug)]
 pub enum Error {
     #[error("Failed to set up network")]
@@ -87,13 +90,18 @@ pub async fn run(config: &Config, vm_config: &VmConfig) -> Result<QemuInstance>
 
     let mut qemu_cmd = Command::new("qemu-system-x86_64");
     let vcpus = vm_config.vcpus.unwrap_or(DEFAULT_NUM_VCPUS);
+    let memory = vm_config.memory.unwrap_or(DEFAULT_AMOUNT_MEMORY);
+
+    log::debug!("CPU count: {vcpus}");
+    log::debug!("Memory: {memory}M");
+
     qemu_cmd.args([
         "-cpu",
         "host",
         "-accel",
         "kvm",
         "-m",
-        "4096",
+        &memory.to_string(),
         "-smp",
         &vcpus.to_string(),
         "-drive",

test/test-manager/src/vm/tart.rs

@@ -49,16 +49,17 @@ pub async fn run(config: &Config, vm_config: &VmConfig) -> Result<TartInstance>
         MachineCopy::clone_vm(&vm_config.image_path).await?
     };
 
+    if let Err(err) = machine_copy.configure(&vm_config).await {
+        log::error!("Failed to configure tart vm: {err}");
+    }
+
     // Start VM
     let mut tart_cmd = Command::new("tart");
     tart_cmd.args(["run", &machine_copy.name, "--serial"]);
 
     if !vm_config.disks.is_empty() {
         log::warn!("Mounting disks is not yet supported")
     }
-    if let Some(cpu) = vm_config.vcpus {
-        tart_cmd.args(["--cpu", &cpu.to_string()]);
-    }
 
     match config.runtime_opts.display {
         config::Display::None => {
@@ -170,6 +171,28 @@ impl MachineCopy {
         })
     }
 
+    pub async fn configure(&self, vm_config: &VmConfig) -> Result<()> {
+        let mut args = vec![];
+        if let Some(cpu) = vm_config.vcpus {
+            args.extend(["--cpu".to_owned(), cpu.to_string()]);
+            log::info!("vCPUs: {cpu}");
+        }
+        if let Some(mem) = vm_config.memory {
+            args.extend(["--memory".to_owned(), mem.to_string()]);
+            log::info!("Memory: {mem} MB");
+        }
+        if !args.is_empty() {
+            let mut tart_cmd = Command::new("tart");
+            tart_cmd.args(["set", &self.name]);
+            tart_cmd.args(args);
+            tart_cmd
+                .status()
+                .await
+                .context("failed to update tart config")?;
+        }
+        Ok(())
+    }
+
     pub async fn cleanup(mut self) {
         let _ = tokio::task::spawn_blocking(move || self.try_destroy()).await;
     }