Add test for collect_apex_names.
Bug: 211458160
Test: virtualizationservice_device_test
Change-Id: Id531da9f99c4110b41481a1bea2b8d3187e134e5
diff --git a/virtualizationservice/src/payload.rs b/virtualizationservice/src/payload.rs
index a8c22cd..84d3b2f 100644
--- a/virtualizationservice/src/payload.rs
+++ b/virtualizationservice/src/payload.rs
@@ -374,6 +374,7 @@
#[cfg(test)]
mod tests {
use super::*;
+
#[test]
fn test_find_apex_names_in_classpath() {
let vars = r#"
@@ -387,4 +388,35 @@
assert_eq!(find_apex_names_in_classpath(vars).unwrap(), expected);
}
+
+ #[test]
+ fn test_collect_apex_names() {
+ let apex_list = ApexInfoList {
+ list: vec![
+ ApexInfo {
+ name: "hasnt_classpath".to_string(),
+ path: PathBuf::from("path0"),
+ has_classpath_jar: false,
+ },
+ ApexInfo {
+ name: "has_classpath".to_string(),
+ path: PathBuf::from("path1"),
+ has_classpath_jar: true,
+ },
+ ],
+ };
+ let apexes = vec![
+ ApexConfig { name: "config_name".to_string() },
+ ApexConfig { name: "{CLASSPATH}".to_string() },
+ ];
+ assert_eq!(
+ collect_apex_names(&apex_list, &apexes, DebugLevel::FULL),
+ vec![
+ "com.android.adbd".to_string(),
+ "com.android.os.statsd".to_string(),
+ "config_name".to_string(),
+ "has_classpath".to_string(),
+ ]
+ );
+ }
}