11use crate :: {
22 network:: { Network , NetworkError , Node } ,
3- types:: NodeEvent ,
3+ types:: { Event , NodeEvent } ,
44} ;
55use crossbeam_channel:: Sender ;
6- use std:: {
7- any:: Any ,
8- collections:: { HashMap , HashSet } ,
9- } ;
6+ use std:: collections:: { HashMap , HashSet } ;
107use wg_internal:: {
118 network:: { NodeId , SourceRoutingHeader } ,
129 packet:: { Ack , FloodRequest , FloodResponse , Fragment , Nack , NackType , NodeType , Packet } ,
@@ -85,7 +82,7 @@ pub struct RoutingHandler {
8582 flood_seen : HashSet < ( u64 , NodeId ) > ,
8683 session_counter : u64 ,
8784 flood_counter : u64 ,
88- controller_send : Sender < Box < dyn Any > > ,
85+ controller_send : Sender < Box < dyn Event > > ,
8986 buffer : Buffer ,
9087}
9188
@@ -95,7 +92,7 @@ impl RoutingHandler {
9592 id : NodeId ,
9693 node_type : NodeType ,
9794 neighbors : HashMap < NodeId , Sender < Packet > > ,
98- controller_send : Sender < Box < dyn Any > > ,
95+ controller_send : Sender < Box < dyn Event > > ,
9996 ) -> Self {
10097 Self {
10198 id,
@@ -506,6 +503,7 @@ mod routing_handler_tests {
506503 handler. start_flood ( ) . unwrap ( ) ;
507504
508505 let packet = receiver. try_recv ( ) . unwrap ( ) ;
506+ let packet = packet. into_any ( ) ;
509507 if let Ok ( cmd) = packet. downcast :: < NodeEvent > ( ) {
510508 assert ! ( matches!( * cmd, NodeEvent :: FloodStarted ( _, _) ) ) ;
511509 }
@@ -566,7 +564,7 @@ mod routing_handler_tests {
566564 handler. handle_ack ( & ack, 1 , 2 ) ;
567565 }
568566
569- fn create_test_routing_handler ( ) -> ( RoutingHandler , Receiver < Box < dyn Any > > ) {
567+ fn create_test_routing_handler ( ) -> ( RoutingHandler , Receiver < Box < dyn Event > > ) {
570568 let ( controller_send, controller_recv) = unbounded ( ) ;
571569 let ( neighbor_send, _) = unbounded ( ) ;
572570 let mut neighbors = HashMap :: new ( ) ;
0 commit comments