fixup! [wlan] Add connect-to-wpa3 test

Change-Id: Ieb80a90fb3b161057c8c726e768c7ae0707ccefa
diff --git a/src/connectivity/wlan/lib/rsn/src/lib.rs b/src/connectivity/wlan/lib/rsn/src/lib.rs
index 0aa0a87..dd95005 100644
--- a/src/connectivity/wlan/lib/rsn/src/lib.rs
+++ b/src/connectivity/wlan/lib/rsn/src/lib.rs
@@ -363,6 +363,8 @@
     UnknownKeyExchange,
     #[error("cannot initiate Fourway Handshake as Supplicant")]
     UnexpectedInitiationRequest,
+    #[error("cannot initiate Supplicant in current EssSa state")]
+    UnexpectedEsssaInitiation,
     #[error("unsupported Key Descriptor Type: {:?}", _0)]
     UnsupportedKeyDescriptor(eapol::KeyDescriptor),
     #[error("unexpected Key Descriptor Type {:?}; expected {:?}", _0, _1)]
diff --git a/src/connectivity/wlan/lib/rsn/src/rsna/esssa.rs b/src/connectivity/wlan/lib/rsn/src/rsna/esssa.rs
index ab6512a..a8b46d9 100644
--- a/src/connectivity/wlan/lib/rsn/src/rsna/esssa.rs
+++ b/src/connectivity/wlan/lib/rsn/src/rsna/esssa.rs
@@ -198,7 +198,11 @@
         // This function will not succeed unless called on a new Esssa or one that was reset.
         match (self.ptksa.as_ref(), self.gtksa.as_ref(), self.igtksa.as_ref()) {
             (Ptksa::Uninitialized { .. }, Gtksa::Uninitialized { .. }, Igtksa::Uninitialized) => (),
-            _ => return Err(Error::UnexpectedInitiationRequest),
+            // TODO (69388): The Ptksa can be in the Initialized state
+            // if the auth method was SAE and the Pmksa key was
+            // confirmed prior to initiate()'ing this Esssa.
+            (Ptksa::Initialized { .. }, Gtksa::Uninitialized { .. }, Igtksa::Uninitialized) => (),
+            _ => return Err(Error::UnexpectedEsssaInitiation),
         };
         info!("establishing ESSSA...");