From 592e5508c478423add3aa5223362b3d85a3ffdb7 Mon Sep 17 00:00:00 2001
From: Ben Stoltz <ben.stoltz@oxidecomputer.com>
Date: Tue, 5 Mar 2024 18:24:03 -0800
Subject: [PATCH] Add test for SpStateV3

---
 gateway-sp-comms/src/sp_response_expect.rs | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/gateway-sp-comms/src/sp_response_expect.rs b/gateway-sp-comms/src/sp_response_expect.rs
index 84f95ae..94bb865 100644
--- a/gateway-sp-comms/src/sp_response_expect.rs
+++ b/gateway-sp-comms/src/sp_response_expect.rs
@@ -200,7 +200,7 @@ pub(crate) fn expect_sp_state(
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::{SpStateV1, SpStateV2, VersionedSpState};
+    use crate::{SpStateV1, SpStateV2, SpStateV3, VersionedSpState};
     use std::net::Ipv6Addr;
 
     fn dummy_addr() -> SocketAddrV6 {
@@ -351,6 +351,24 @@ mod tests {
             panic!("mismatched value {v:?}");
         };
         assert_eq!(r, VersionedSpState::V2(state));
+
+        let state = SpStateV3 {
+            hubris_archive_id: [1, 2, 3, 4, 5, 6, 7, 8],
+            serial_number: [0; 32],
+            model: [0; 32],
+            revision: 123,
+            base_mac_address: [0; 6],
+            power_state: gateway_messages::PowerState::A0,
+        };
+        let v = expect_sp_state((
+            dummy_addr(),
+            SpResponse::SpStateV3(state),
+            vec![],
+        ));
+        let Ok(r) = v else {
+            panic!("mismatched value {v:?}");
+        };
+        assert_eq!(r, VersionedSpState::V3(state));
     }
 
     #[test]