mirror of
https://github.com/telemt/telemt.git
synced 2026-06-19 02:00:08 +07:00
David Osipov
5933b5e821
- Renamed test functions in `client_tls_clienthello_truncation_adversarial_tests.rs` to remove "but_leaks" suffix for clarity. - Added new tests in `direct_relay_business_logic_tests.rs` to validate business logic for data center resolution and scope hints. - Introduced tests in `direct_relay_common_mistakes_tests.rs` to cover common mistakes in direct relay configurations. - Added security tests in `direct_relay_security_tests.rs` to ensure proper handling of symlink and parent swap scenarios. - Created `direct_relay_subtle_adversarial_tests.rs` to stress test concurrent logging and validate scope hint behavior. - Implemented `relay_quota_lock_pressure_adversarial_tests.rs` to test quota lock behavior under high contention and stress. - Updated `relay_security_tests.rs` to include quota lock contention tests ensuring proper behavior under concurrent access. - Introduced `ip_tracker_hotpath_adversarial_tests.rs` to validate the performance and correctness of the IP tracking logic under various scenarios.
52 lines
1.9 KiB
Rust
52 lines
1.9 KiB
Rust
use super::*;
|
|
use crate::protocol::constants::{TG_DATACENTER_PORT, TG_DATACENTERS_V4, TG_DATACENTERS_V6};
|
|
use std::net::SocketAddr;
|
|
|
|
#[test]
|
|
fn business_scope_hint_accepts_exact_boundary_length() {
|
|
let value = format!("scope_{}", "a".repeat(MAX_SCOPE_HINT_LEN));
|
|
assert_eq!(validated_scope_hint(&value), Some("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"));
|
|
}
|
|
|
|
#[test]
|
|
fn business_scope_hint_rejects_missing_prefix_even_when_charset_is_valid() {
|
|
assert_eq!(validated_scope_hint("alpha-01"), None);
|
|
}
|
|
|
|
#[test]
|
|
fn business_known_dc_uses_ipv4_table_by_default() {
|
|
let cfg = ProxyConfig::default();
|
|
let resolved = get_dc_addr_static(2, &cfg).expect("known dc must resolve");
|
|
let expected = SocketAddr::new(TG_DATACENTERS_V4[1], TG_DATACENTER_PORT);
|
|
assert_eq!(resolved, expected);
|
|
}
|
|
|
|
#[test]
|
|
fn business_negative_dc_maps_by_absolute_value() {
|
|
let cfg = ProxyConfig::default();
|
|
let resolved = get_dc_addr_static(-3, &cfg).expect("negative dc index must map by absolute value");
|
|
let expected = SocketAddr::new(TG_DATACENTERS_V4[2], TG_DATACENTER_PORT);
|
|
assert_eq!(resolved, expected);
|
|
}
|
|
|
|
#[test]
|
|
fn business_known_dc_uses_ipv6_table_when_preferred_and_enabled() {
|
|
let mut cfg = ProxyConfig::default();
|
|
cfg.network.prefer = 6;
|
|
cfg.network.ipv6 = Some(true);
|
|
|
|
let resolved = get_dc_addr_static(1, &cfg).expect("known dc must resolve on ipv6 path");
|
|
let expected = SocketAddr::new(TG_DATACENTERS_V6[0], TG_DATACENTER_PORT);
|
|
assert_eq!(resolved, expected);
|
|
}
|
|
|
|
#[test]
|
|
fn business_unknown_dc_uses_configured_default_dc_when_in_range() {
|
|
let mut cfg = ProxyConfig::default();
|
|
cfg.default_dc = Some(4);
|
|
|
|
let resolved = get_dc_addr_static(29_999, &cfg).expect("unknown dc must resolve to configured default");
|
|
let expected = SocketAddr::new(TG_DATACENTERS_V4[3], TG_DATACENTER_PORT);
|
|
assert_eq!(resolved, expected);
|
|
}
|