diff options
Diffstat (limited to 'src/tests/cases/test_utils.rs')
-rw-r--r-- | src/tests/cases/test_utils.rs | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/src/tests/cases/test_utils.rs b/src/tests/cases/test_utils.rs index 22e6c63..18440cc 100644 --- a/src/tests/cases/test_utils.rs +++ b/src/tests/cases/test_utils.rs @@ -24,9 +24,9 @@ pub fn os_input_output( ) -> OsInputOutput { os_input_output_factory( network_frames, - sleep_num, None, create_fake_dns_client(HashMap::new()), + sleep_and_quit_events(sleep_num), ) } pub fn os_input_output_stdout( @@ -36,9 +36,9 @@ pub fn os_input_output_stdout( ) -> OsInputOutput { os_input_output_factory( network_frames, - sleep_num, stdout, create_fake_dns_client(HashMap::new()), + sleep_and_quit_events(sleep_num), ) } @@ -48,14 +48,19 @@ pub fn os_input_output_dns( stdout: Option<Arc<Mutex<Vec<u8>>>>, dns_client: Option<Client>, ) -> OsInputOutput { - os_input_output_factory(network_frames, sleep_num, stdout, dns_client) + os_input_output_factory( + network_frames, + stdout, + dns_client, + sleep_and_quit_events(sleep_num), + ) } -fn os_input_output_factory( +pub fn os_input_output_factory( network_frames: Vec<Box<dyn DataLinkReceiver>>, - sleep_num: usize, stdout: Option<Arc<Mutex<Vec<u8>>>>, dns_client: Option<Client>, + keyboard_events: Box<dyn Iterator<Item = Event> + Send>, ) -> OsInputOutput { let on_winch = create_fake_on_winch(false); let cleanup = Box::new(|| {}); @@ -74,7 +79,7 @@ fn os_input_output_factory( network_interfaces: get_interfaces(), network_frames, get_open_sockets, - keyboard_events: sleep_and_quit_events(sleep_num), + keyboard_events, dns_client, on_winch, cleanup, |