test(vmm): Add test for consistency of static CPU templates

Adds a test that verify consistency of static CPU templates between
hardcoded ones and JSON ones in order not to forget to update one or the
other.

Signed-off-by: Takahiro Itazuri <[email protected]>

Author: zulinx86

src/vmm/src/cpu_config/aarch64/static_cpu_templates/mod.rs

@@ -40,3 +40,19 @@ impl std::fmt::Display for StaticCpuTemplate {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::cpu_config::test_utils::get_json_template;
+
+    #[test]
+    fn verify_consistency_with_json_templates() {
+        let static_templates = [(v1n1::v1n1(), "aarch64_v1n1.json")];
+
+        for (hardcoded_template, filename) in static_templates {
+            let json_template = get_json_template(filename);
+            assert_eq!(hardcoded_template, json_template);
+        }
+    }
+}

src/vmm/src/cpu_config/mod.rs

@@ -13,3 +13,6 @@ pub mod x86_64;
 /// Module containing type implementations needed for aarch64 (ARM) CPU configuration
 #[cfg(target_arch = "aarch64")]
 pub mod aarch64;
+
+#[cfg(test)]
+pub mod test_utils;

src/vmm/src/cpu_config/test_utils.rs

@@ -0,0 +1,19 @@
+// Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+use std::path::PathBuf;
+
+use crate::cpu_config::templates::CustomCpuTemplate;
+
+// Get a static CPU template stored as a JSON file.
+pub fn get_json_template(filename: &str) -> CustomCpuTemplate {
+    let json_path = [
+        env!("CARGO_MANIFEST_DIR"),
+        "../../resources/tests/static_cpu_templates",
+        filename,
+    ]
+    .iter()
+    .collect::<PathBuf>();
+
+    serde_json::from_str(&std::fs::read_to_string(json_path).unwrap()).unwrap()
+}

src/vmm/src/cpu_config/x86_64/static_cpu_templates/mod.rs

@@ -50,3 +50,25 @@ impl StaticCpuTemplate {
         self == &StaticCpuTemplate::None
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::cpu_config::test_utils::get_json_template;
+
+    #[test]
+    fn verify_consistency_with_json_templates() {
+        let static_templates = [
+            (c3::c3(), "c3.json"),
+            (t2::t2(), "t2.json"),
+            (t2s::t2s(), "t2s.json"),
+            (t2cl::t2cl(), "t2cl.json"),
+            (t2a::t2a(), "t2a.json"),
+        ];
+
+        for (hardcoded_template, filename) in static_templates {
+            let json_template = get_json_template(filename);
+            assert_eq!(hardcoded_template, json_template);
+        }
+    }
+}