test(efiboot): add tests for efiboot boot enable and `efiboot boot …

…disable`
iTrooz · Oct 19, 2023 · 0a54f7c · 0a54f7c
1 parent 98fd9d6
commit 0a54f7c
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/efiboot/src/cli/boot/tests/enable_disable.rs b/efiboot/src/cli/boot/tests/enable_disable.rs
@@ -31,6 +31,27 @@ fn enable() {
     assert_eq!(orig_entry, new_entry);
 }
 
+#[test]
+fn enable_enabled() {
+    let manager = &mut MemoryStore::new();
+
+    let orig_entry = add_entry(manager, 0x0001, true);
+
+    assert_eq!(
+        ExitCode::FAILURE,
+        crate::run(
+            Command::parse_from(["efiboot", "boot", "enable", "0001",]),
+            manager,
+        )
+    );
+
+    // verify variable did not change
+    assert_eq!(
+        orig_entry,
+        BootEntry::read(manager, &Variable::new("Boot0001")).unwrap()
+    );
+}
+
 #[test]
 fn disable() {
     let manager = &mut MemoryStore::new();
@@ -51,3 +72,23 @@ fn disable() {
 
     assert_eq!(orig_entry, new_entry);
 }
+
+#[test]
+fn disable_disabled() {
+    let manager = &mut MemoryStore::new();
+
+    let orig_entry = add_entry(manager, 0x0001, false);
+
+    assert_eq!(
+        ExitCode::FAILURE,
+        crate::run(
+            Command::parse_from(["efiboot", "boot", "disable", "0001",]),
+            manager,
+        )
+    );
+
+    assert_eq!(
+        orig_entry,
+        BootEntry::read(manager, &Variable::new("Boot0001")).unwrap()
+    );
+}