Cover await_in_schedule InSchedule/Shutdown/SessionGone paths

Author: davidsteiner

crates/hotfix/src/session.rs

@@ -9,7 +9,7 @@ mod session_handle;
 pub mod session_ref;
 mod state;
 #[cfg(test)]
-mod test_utils;
+pub(crate) mod test_utils;
 
 use chrono::Utc;
 use hotfix_message::dict::Dictionary;

crates/hotfix/src/session/session_ref.rs

@@ -111,45 +111,18 @@ impl From<oneshot::error::RecvError> for SessionGone {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::message::Message;
-    use crate::message::OutboundMessage;
-    use crate::session::admin_request::AdminRequest;
-
-    #[derive(Clone, Debug, PartialEq)]
-    struct TestMessage;
-
-    impl OutboundMessage for TestMessage {
-        fn write(&self, _msg: &mut Message) {}
-        fn message_type(&self) -> &str {
-            "TEST"
-        }
-    }
-
-    fn create_test_session_ref() -> (
-        InternalSessionRef<TestMessage>,
-        mpsc::Receiver<SessionEvent>,
-    ) {
-        let (event_sender, event_receiver) = mpsc::channel::<SessionEvent>(100);
-        let (outbound_message_sender, _outbound_receiver) =
-            mpsc::channel::<OutboundRequest<TestMessage>>(10);
-        let (admin_request_sender, _admin_receiver) = mpsc::channel::<AdminRequest>(10);
-
-        let session_ref = InternalSessionRef {
-            event_sender,
-            outbound_message_sender,
-            admin_request_sender,
-        };
-
-        (session_ref, event_receiver)
-    }
+    use crate::session::test_utils::create_test_session_ref;
 
     #[tokio::test]
     async fn await_in_schedule_returns_in_schedule_when_session_responds_in_schedule() {
         let (session_ref, mut event_receiver) = create_test_session_ref();
 
         tokio::spawn(async move {
-            if let Some(SessionEvent::AwaitSchedule(responder)) = event_receiver.recv().await {
-                let _ = responder.send(ScheduleResponse::InSchedule);
+            match event_receiver.recv().await {
+                Some(SessionEvent::AwaitSchedule(responder)) => {
+                    let _ = responder.send(ScheduleResponse::InSchedule);
+                }
+                other => panic!("unexpected event: {other:?}"),
             }
         });
 
@@ -162,8 +135,11 @@ mod tests {
         let (session_ref, mut event_receiver) = create_test_session_ref();
 
         tokio::spawn(async move {
-            if let Some(SessionEvent::AwaitSchedule(responder)) = event_receiver.recv().await {
-                let _ = responder.send(ScheduleResponse::Shutdown);
+            match event_receiver.recv().await {
+                Some(SessionEvent::AwaitSchedule(responder)) => {
+                    let _ = responder.send(ScheduleResponse::Shutdown);
+                }
+                other => panic!("unexpected event: {other:?}"),
             }
         });
 
@@ -177,6 +153,6 @@ mod tests {
         drop(event_receiver);
 
         let result = session_ref.await_in_schedule().await;
-        assert!(result.is_err());
+        assert!(matches!(result, Err(SessionGone(_))));
     }
 }

crates/hotfix/src/session/test_utils.rs

@@ -1,5 +1,9 @@
 use crate::config::SessionConfig;
+use crate::message::{Message, OutboundMessage};
+use crate::session::admin_request::AdminRequest;
 use crate::session::ctx::SessionCtx;
+use crate::session::event::SessionEvent;
+use crate::session::session_ref::{InternalSessionRef, OutboundRequest};
 use crate::store::{MessageStore, Result as StoreResult};
 use crate::transport::writer::{WriterMessage, WriterRef};
 use chrono::{DateTime, Utc};
@@ -117,3 +121,31 @@ pub(crate) fn extract_field(raw: &[u8], tag: u32) -> Option<String> {
     }
     None
 }
+
+#[derive(Clone)]
+pub(crate) struct TestMessage;
+
+impl OutboundMessage for TestMessage {
+    fn write(&self, _msg: &mut Message) {}
+    fn message_type(&self) -> &str {
+        "TEST"
+    }
+}
+
+pub(crate) fn create_test_session_ref() -> (
+    InternalSessionRef<TestMessage>,
+    mpsc::Receiver<SessionEvent>,
+) {
+    let (event_sender, event_receiver) = mpsc::channel::<SessionEvent>(100);
+    let (outbound_message_sender, _outbound_receiver) =
+        mpsc::channel::<OutboundRequest<TestMessage>>(10);
+    let (admin_request_sender, _admin_receiver) = mpsc::channel::<AdminRequest>(10);
+
+    let session_ref = InternalSessionRef {
+        event_sender,
+        outbound_message_sender,
+        admin_request_sender,
+    };
+
+    (session_ref, event_receiver)
+}

crates/hotfix/src/transport/socket/socket_reader.rs

@@ -84,42 +84,9 @@ where
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::message::Message;
-    use crate::session::admin_request::AdminRequest;
     use crate::session::event::SessionEvent;
-    use crate::session::session_ref::OutboundRequest;
+    use crate::session::test_utils::create_test_session_ref;
     use tokio::io::{AsyncWriteExt, duplex};
-    use tokio::sync::mpsc;
-
-    #[derive(Clone, Debug, PartialEq)]
-    struct TestMessage;
-
-    impl OutboundMessage for TestMessage {
-        fn write(&self, _msg: &mut Message) {}
-
-        fn message_type(&self) -> &str {
-            "TEST"
-        }
-    }
-
-    /// Creates a test InternalSessionRef that captures events for verification
-    fn create_test_session_ref() -> (
-        InternalSessionRef<TestMessage>,
-        mpsc::Receiver<SessionEvent>,
-    ) {
-        let (event_sender, event_receiver) = mpsc::channel::<SessionEvent>(100);
-        let (outbound_message_sender, _outbound_receiver) =
-            mpsc::channel::<OutboundRequest<TestMessage>>(10);
-        let (admin_request_sender, _admin_receiver) = mpsc::channel::<AdminRequest>(10);
-
-        let session_ref = InternalSessionRef {
-            event_sender,
-            outbound_message_sender,
-            admin_request_sender,
-        };
-
-        (session_ref, event_receiver)
-    }
 
     /// Test that the reader correctly parses a valid FIX message and sends it to the session
     /// for processing.