uefi-test-runner: SNP: improve idempotence of test

snp::test() can now be called multiple times in a row

Author: phip1611

uefi-test-runner/src/proto/network/snp.rs

@@ -6,6 +6,7 @@ use uefi::boot::ScopedProtocol;
 use uefi::proto::network::MacAddress;
 use uefi::proto::network::snp::{InterruptStatus, ReceiveFlags, SimpleNetwork};
 use uefi::{Status, boot};
+use uefi_raw::protocol::network::snp::NetworkState;
 
 /// The MAC address configured for the interface.
 const EXPECTED_MAC: [u8; 6] = [0x52, 0x54, 0, 0, 0, 0x1];
@@ -63,15 +64,12 @@ pub fn test() {
         EXPECTED_MAC[5]
     ));
 
-    // Check shutdown
-    let res = simple_network.shutdown();
-    assert!(res == Ok(()) || res == Err(Status::NOT_STARTED.into()));
-
-    // Check stop
-    let res = simple_network.stop();
-    assert!(res == Ok(()) || res == Err(Status::NOT_STARTED.into()));
+    assert_eq!(
+        simple_network.mode().state,
+        NetworkState::STOPPED,
+        "Should be in stopped state"
+    );
 
-    // Check start
     simple_network
         .start()
         .expect("Failed to start Simple Network");
@@ -181,4 +179,7 @@ pub fn test() {
             }
         }
     }
+
+    simple_network.stop().unwrap();
+    simple_network.shutdown().unwrap();
 }