From ea58d2f68aea55a5b79e1fe887e7a0bb50aaad33 Mon Sep 17 00:00:00 2001
From: Rob Bradford <robert.bradford@intel.com>
Date: Mon, 8 Aug 2022 14:58:22 +0100
Subject: [PATCH] vmm: config: Enhance test_cpu_parsing to add "affinity"
 parameter

Signed-off-by: Rob Bradford <robert.bradford@intel.com>
---
 vmm/src/config.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/vmm/src/config.rs b/vmm/src/config.rs
index 97ba606f0..8a74cb339 100644
--- a/vmm/src/config.rs
+++ b/vmm/src/config.rs
@@ -2738,6 +2738,25 @@ mod tests {
                 ..Default::default()
             }
         );
+        assert_eq!(
+            CpusConfig::parse("boot=2,affinity=[0@[0,2],1@[1,3]]")?,
+            CpusConfig {
+                boot_vcpus: 2,
+                max_vcpus: 2,
+                affinity: Some(vec![
+                    CpuAffinity {
+                        vcpu: 0,
+                        host_cpus: vec![0, 2],
+                    },
+                    CpuAffinity {
+                        vcpu: 1,
+                        host_cpus: vec![1, 3],
+                    }
+                ]),
+                ..Default::default()
+            },
+        );
+
         Ok(())
     }