diff options
Diffstat (limited to 'src/tests/cases/ui.rs')
-rw-r--r-- | src/tests/cases/ui.rs | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/src/tests/cases/ui.rs b/src/tests/cases/ui.rs index 2209284..70d7e78 100644 --- a/src/tests/cases/ui.rs +++ b/src/tests/cases/ui.rs @@ -17,7 +17,8 @@ use crate::{ sleep_and_quit_events, sleep_resize_and_quit_events, test_backend_factory, }, fakes::{ - create_fake_dns_client, get_interfaces, get_open_sockets, NetworkFrames, TerminalEvents, + create_fake_dns_client, get_interfaces_with_frames, get_open_sockets, NetworkFrames, + TerminalEvents, }, }, Opt, OsInputOutput, @@ -640,6 +641,8 @@ fn sustained_traffic_from_multiple_processes_bi_directional_total( fn traffic_with_host_names(network_frames: Vec<Box<dyn DataLinkReceiver>>) { let (terminal_events, terminal_draw_events, backend) = test_backend_factory(190, 50); + let interfaces_with_frames = get_interfaces_with_frames(network_frames); + let mut ips_to_hostnames = HashMap::new(); ips_to_hostnames.insert( IpAddr::V4("1.1.1.1".parse().unwrap()), @@ -657,8 +660,7 @@ fn traffic_with_host_names(network_frames: Vec<Box<dyn DataLinkReceiver>>) { let write_to_stdout = Box::new(move |_output: String| {}); let os_input = OsInputOutput { - network_interfaces: get_interfaces(), - network_frames, + interfaces_with_frames, get_open_sockets, terminal_events: sleep_and_quit_events(3), dns_client, @@ -678,6 +680,8 @@ fn traffic_with_host_names(network_frames: Vec<Box<dyn DataLinkReceiver>>) { fn truncate_long_hostnames(network_frames: Vec<Box<dyn DataLinkReceiver>>) { let (terminal_events, terminal_draw_events, backend) = test_backend_factory(190, 50); + let interfaces_with_frames = get_interfaces_with_frames(network_frames); + let mut ips_to_hostnames = HashMap::new(); ips_to_hostnames.insert( IpAddr::V4("1.1.1.1".parse().unwrap()), @@ -695,8 +699,7 @@ fn truncate_long_hostnames(network_frames: Vec<Box<dyn DataLinkReceiver>>) { let write_to_stdout = Box::new(move |_output: String| {}); let os_input = OsInputOutput { - network_interfaces: get_interfaces(), - network_frames, + interfaces_with_frames, get_open_sockets, terminal_events: sleep_and_quit_events(3), dns_client, @@ -716,6 +719,8 @@ fn truncate_long_hostnames(network_frames: Vec<Box<dyn DataLinkReceiver>>) { fn no_resolve_mode(network_frames: Vec<Box<dyn DataLinkReceiver>>) { let (terminal_events, terminal_draw_events, backend) = test_backend_factory(190, 50); + let interfaces_with_frames = get_interfaces_with_frames(network_frames); + let mut ips_to_hostnames = HashMap::new(); ips_to_hostnames.insert( IpAddr::V4("1.1.1.1".parse().unwrap()), @@ -733,8 +738,7 @@ fn no_resolve_mode(network_frames: Vec<Box<dyn DataLinkReceiver>>) { let write_to_stdout = Box::new(move |_output: String| {}); let os_input = OsInputOutput { - network_interfaces: get_interfaces(), - network_frames, + interfaces_with_frames, get_open_sockets, terminal_events: sleep_and_quit_events(3), dns_client, @@ -759,6 +763,7 @@ fn traffic_with_winch_event() { 12345, b"I am a fake tcp packet", ))]) as Box<dyn DataLinkReceiver>]; + let interfaces_with_frames = get_interfaces_with_frames(network_frames); let (terminal_events, terminal_draw_events, backend) = test_backend_factory(190, 50); @@ -766,8 +771,7 @@ fn traffic_with_winch_event() { let write_to_stdout = Box::new(move |_output: String| {}); let os_input = OsInputOutput { - network_interfaces: get_interfaces(), - network_frames, + interfaces_with_frames, get_open_sockets, terminal_events: sleep_resize_and_quit_events(2), dns_client, |