use super::*; macro_rules! probe_node_buddy { ($layer:ident<$rt: ident> ($kind:literal, $expr: expr)) => { paste::paste! { #[test] fn [< test_ $rt:snake _ $kind:snake _probe_node_buddy >]() { [< $rt:snake _run >](async move { let mut t1_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_1".into(), $expr); t1_opts.add_bind_address(next_socket_addr_v4(0)); let t1 = NetTransport::<_, SocketAddrResolver<[< $rt:camel Runtime >]>, _, Lpe<_, _>, [< $rt:camel Runtime >]>::new(t1_opts).await.unwrap(); let t1_opts = Options::lan(); let mut t2_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_2".into(), $expr); t2_opts.add_bind_address(next_socket_addr_v4(0)); let t2 = NetTransport::new(t2_opts).await.unwrap(); probe_node_buddy(t1, t1_opts, t2).await; }); } #[cfg(feature = "compression")] #[test] fn [< test_ $rt:snake _ $kind:snake _probe_node_buddy_with_compression >]() { [< $rt:snake _run >](async move { let mut t1_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_1".into(), $expr).with_compressor(Some(Default::default())).with_offload_size(10); t1_opts.add_bind_address(next_socket_addr_v4(0)); let t1 = NetTransport::<_, SocketAddrResolver<[< $rt:camel Runtime >]>, _, Lpe<_, _>, [< $rt:camel Runtime >]>::new(t1_opts).await.unwrap(); let t1_opts = Options::lan(); let mut t2_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_2".into(), $expr).with_compressor(Some(Default::default())).with_offload_size(10); t2_opts.add_bind_address(next_socket_addr_v4(0)); let t2 = NetTransport::new(t2_opts).await.unwrap(); probe_node_buddy(t1, t1_opts, t2).await; }); } #[cfg(feature = "encryption")] #[test] fn [< test_ $rt:snake _ $kind:snake _probe_node_buddy_with_encryption >]() { [< $rt:snake _run >](async move { let mut t1_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_1".into(), $expr).with_primary_key(Some(TEST_KEYS[0])).with_offload_size(10); t1_opts.add_bind_address(next_socket_addr_v4(0)); let t1 = NetTransport::<_, SocketAddrResolver<[< $rt:camel Runtime >]>, _, Lpe<_, _>, [< $rt:camel Runtime >]>::new(t1_opts).await.unwrap(); let t1_opts = Options::lan(); let mut t2_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_2".into(), $expr).with_primary_key(Some(TEST_KEYS[1])); t2_opts.add_bind_address(next_socket_addr_v4(0)); let t2 = NetTransport::new(t2_opts).await.unwrap(); probe_node_buddy(t1, t1_opts, t2).await; }); } #[cfg(all(feature = "encryption", feature = "compression"))] #[test] fn [< test_ $rt:snake _ $kind:snake _probe_node_buddy_with_encryption_and_compression >]() { [< $rt:snake _run >](async move { let mut t1_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_1".into(), $expr).with_primary_key(Some(TEST_KEYS[0])).with_offload_size(10).with_compressor(Some(Default::default())); t1_opts.add_bind_address(next_socket_addr_v4(0)); let t1 = NetTransport::<_, SocketAddrResolver<[< $rt:camel Runtime >]>, _, Lpe<_, _>, [< $rt:camel Runtime >]>::new(t1_opts).await.unwrap(); let t1_opts = Options::lan(); let mut t2_opts = NetTransportOptions::]>>::with_stream_layer_options("probe_node_buddy_2".into(), $expr).with_primary_key(Some(TEST_KEYS[1])).with_compressor(Some(Default::default())); t2_opts.add_bind_address(next_socket_addr_v4(0)); let t2 = NetTransport::new(t2_opts).await.unwrap(); probe_node_buddy(t1, t1_opts, t2).await; }); } } }; } test_mods!(probe_node_buddy);