@@ -442,6 +442,7 @@ struct CommandResponseResult {
442
442
443
443
#[ cfg( test) ]
444
444
mod tests {
445
+ use std:: env;
445
446
use std:: sync:: { Arc , Mutex } ;
446
447
use config:: Config ;
447
448
use crate :: { CommandAccessor ,
@@ -525,17 +526,14 @@ mod tests {
525
526
526
527
impl InboundChannel for TokioInboundChannel {
527
528
fn consume ( & mut self ) -> Option < Vec < u8 > > {
528
- let runtime = tokio:: runtime:: Runtime :: new ( ) . unwrap ( ) ;
529
- return runtime. block_on ( async {
530
- if let Some ( receiver) = self . receiver . take ( ) {
531
- match receiver. await {
532
- Ok ( k) => Some ( k) ,
533
- Err ( _) => None
534
- }
535
- } else {
536
- None
529
+ if let Some ( receiver) = self . receiver . take ( ) {
530
+ match receiver. blocking_recv ( ) {
531
+ Ok ( k) => Some ( k) ,
532
+ Err ( _) => None
537
533
}
538
- } )
534
+ } else {
535
+ None
536
+ }
539
537
}
540
538
}
541
539
@@ -594,13 +592,18 @@ mod tests {
594
592
595
593
596
594
#[ tokio:: test]
597
- async fn test_serialize_command_response ( ) {
595
+ async fn xtest_serialize_command_response ( ) {
598
596
599
597
let command = TestCreateUserCommand {
600
598
user_id : String :: from ( "user_id" ) ,
601
599
name : String :: from ( "user_name" )
602
600
} ;
603
601
602
+ if env:: var ( "RUST_LOG" ) . is_err ( ) {
603
+ env:: set_var ( "RUST_LOG" , "debug" )
604
+ }
605
+
606
+ env_logger:: init ( ) ;
604
607
605
608
let ( tx_command, rx_command) : ( Sender < Vec < u8 > > , Receiver < Vec < u8 > > ) = oneshot:: channel ( ) ;
606
609
let ( tx_command_response, rx_command_response) : ( Sender < Vec < u8 > > , Receiver < Vec < u8 > > ) = oneshot:: channel ( ) ;
@@ -616,7 +619,9 @@ mod tests {
616
619
617
620
match rx_command. await {
618
621
Ok ( mut k) => {
619
- let mut channel1 = TokioOutboundChannel :: new ( tx_command_response) ;
622
+ let mut channel1 = TokioOutboundChannel {
623
+ sender : Some ( tx_command_response)
624
+ } ;
620
625
command_service_server. handle_message ( & mut k, & mut channel1)
621
626
}
622
627
Err ( _) => {
0 commit comments