11pub ( crate ) mod admin_request;
2+ pub mod error;
23pub ( crate ) mod event;
34mod info;
45mod session_handle;
@@ -26,15 +27,17 @@ use tracing::{debug, enabled, error, info, warn};
2627
2728use crate :: Application ;
2829use crate :: application:: { InboundDecision , OutboundDecision } ;
29- use crate :: error:: { CompIdType , MessageVerificationError } ;
3030use crate :: message:: logout:: Logout ;
3131use crate :: message:: reject:: Reject ;
3232use crate :: message:: resend_request:: ResendRequest ;
3333use crate :: message:: sequence_reset:: SequenceReset ;
3434use crate :: message:: test_request:: TestRequest ;
3535use crate :: message:: verification:: verify_message;
36+ use crate :: message:: verification_error:: { CompIdType , MessageVerificationError } ;
3637use crate :: message_utils:: { is_admin, prepare_message_for_resend} ;
3738use crate :: session:: admin_request:: AdminRequest ;
39+ pub use crate :: session:: info:: { SessionInfo , Status } ;
40+ pub use crate :: session:: session_handle:: SessionHandle ;
3841#[ cfg( not( feature = "test-utils" ) ) ]
3942pub ( crate ) use crate :: session:: session_ref:: InternalSessionRef ;
4043#[ cfg( feature = "test-utils" ) ]
@@ -49,9 +52,6 @@ use hotfix_message::session_fields::{
4952 SessionRejectReason , TEST_REQ_ID ,
5053} ;
5154
52- pub use crate :: session:: info:: { SessionInfo , Status } ;
53- pub use crate :: session:: session_handle:: SessionHandle ;
54-
5555const SCHEDULE_CHECK_INTERVAL : u64 = 1 ;
5656
5757struct Session < A , I , O , S > {
@@ -365,7 +365,8 @@ where
365365 }
366366 }
367367
368- self . store . increment_target_seq_number ( ) . await
368+ self . store . increment_target_seq_number ( ) . await ?;
369+ Ok ( ( ) )
369370 }
370371
371372 async fn on_heartbeat ( & mut self , message : & Message ) -> Result < ( ) > {
@@ -378,7 +379,8 @@ where
378379 self . reset_peer_timer ( None ) ;
379380 }
380381
381- self . store . increment_target_seq_number ( ) . await
382+ self . store . increment_target_seq_number ( ) . await ?;
383+ Ok ( ( ) )
382384 }
383385
384386 async fn on_test_request ( & mut self , message : & Message ) -> Result < ( ) > {
@@ -515,7 +517,8 @@ where
515517 return Ok ( ( ) ) ;
516518 }
517519
518- self . store . set_target_seq_number ( end - 1 ) . await
520+ self . store . set_target_seq_number ( end - 1 ) . await ?;
521+ Ok ( ( ) )
519522 }
520523
521524 async fn handle_verification_error ( & mut self , error : MessageVerificationError ) -> Result < ( ) > {
@@ -1129,6 +1132,7 @@ mod tests {
11291132 use super :: * ;
11301133 use crate :: application:: { InboundDecision , OutboundDecision } ;
11311134 use crate :: message:: { InboundMessage , OutboundMessage } ;
1135+ use crate :: store:: { Result as StoreResult , StoreError } ;
11321136 use chrono:: { DateTime , Datelike , NaiveDate , NaiveTime , TimeDelta , Timelike } ;
11331137 use chrono_tz:: Tz ;
11341138 use hotfix_message:: message:: Message ;
@@ -1168,11 +1172,11 @@ mod tests {
11681172
11691173 #[ async_trait:: async_trait]
11701174 impl MessageStore for TestStore {
1171- async fn add ( & mut self , _sequence_number : u64 , _message : & [ u8 ] ) -> Result < ( ) > {
1175+ async fn add ( & mut self , _sequence_number : u64 , _message : & [ u8 ] ) -> StoreResult < ( ) > {
11721176 Ok ( ( ) )
11731177 }
11741178
1175- async fn get_slice ( & self , _begin : usize , _end : usize ) -> Result < Vec < Vec < u8 > > > {
1179+ async fn get_slice ( & self , _begin : usize , _end : usize ) -> StoreResult < Vec < Vec < u8 > > > {
11761180 Ok ( vec ! [ ] )
11771181 }
11781182
@@ -1184,25 +1188,25 @@ mod tests {
11841188 self . target_seq
11851189 }
11861190
1187- async fn increment_sender_seq_number ( & mut self ) -> Result < ( ) > {
1191+ async fn increment_sender_seq_number ( & mut self ) -> StoreResult < ( ) > {
11881192 self . sender_seq += 1 ;
11891193 Ok ( ( ) )
11901194 }
11911195
1192- async fn increment_target_seq_number ( & mut self ) -> Result < ( ) > {
1196+ async fn increment_target_seq_number ( & mut self ) -> StoreResult < ( ) > {
11931197 self . target_seq += 1 ;
11941198 Ok ( ( ) )
11951199 }
11961200
1197- async fn set_target_seq_number ( & mut self , seq_number : u64 ) -> Result < ( ) > {
1201+ async fn set_target_seq_number ( & mut self , seq_number : u64 ) -> StoreResult < ( ) > {
11981202 self . target_seq = seq_number;
11991203 Ok ( ( ) )
12001204 }
12011205
1202- async fn reset ( & mut self ) -> Result < ( ) > {
1206+ async fn reset ( & mut self ) -> StoreResult < ( ) > {
12031207 self . reset_called . store ( true , Ordering :: SeqCst ) ;
12041208 if self . fail_reset . load ( Ordering :: SeqCst ) {
1205- bail ! ( "simulated reset failure" )
1209+ return Err ( StoreError :: Reset ( "simulated reset failure" . into ( ) ) ) ;
12061210 }
12071211 self . creation_time = Utc :: now ( ) ;
12081212 Ok ( ( ) )
0 commit comments