luckyimnotanigger/tun2proxy
1
0
Fork 0
mirror of https://github.com/tun2proxy/tun2proxy.git synced 2025-04-22 06:49:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add DNS tests

Browse source
This commit is contained in:
B. Blechschmidt 2023-03-23 14:51:56 +01:00
parent 32403c5423
commit d13c7ff61b
1 changed files with 39 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
tests/proxy.rs
View file

@ -116,9 +116,18 @@ mod tests {
} }
} }
fn run_test<F>(filter: F) fn request_ip_host_http() {
reqwest::blocking::get("http://1.1.1.1").expect("failed to issue HTTP request");
}
fn request_example_https() {
reqwest::blocking::get("https://example.org").expect("failed to issue HTTPs request");
}
fn run_test<F, T>(filter: F, test_function: T)
where where
F: Fn(&Test) -> bool, F: Fn(&Test) -> bool,
T: Fn(),
{ {
for potential_test in tests() { for potential_test in tests() {
match potential_test { match potential_test {
@ -129,8 +138,7 @@ mod tests {
match fork::fork() { match fork::fork() {
Ok(Fork::Parent(child)) => { Ok(Fork::Parent(child)) => {
reqwest::blocking::get("https://1.1.1.1") test_function();
.expect("failed to issue HTTP request");
signal::kill(Pid::from_raw(child), signal::SIGKILL) signal::kill(Pid::from_raw(child), signal::SIGKILL)
.expect("failed to kill child"); .expect("failed to kill child");
nix::sys::wait::waitpid(Pid::from_raw(child), None) nix::sys::wait::waitpid(Pid::from_raw(child), None)
@ -158,13 +166,39 @@ mod tests {
#[test] #[test]
fn test_socks5() { fn test_socks5() {
require_var("SOCKS5_SERVER"); require_var("SOCKS5_SERVER");
run_test(|test| test.proxy.proxy_type == ProxyType::Socks5) run_test(
|test| test.proxy.proxy_type == ProxyType::Socks5,
request_ip_host_http,
)
} }
#[serial] #[serial]
#[test] #[test]
fn test_http() { fn test_http() {
require_var("HTTP_SERVER"); require_var("HTTP_SERVER");
run_test(|test| test.proxy.proxy_type == ProxyType::Http) run_test(
|test| test.proxy.proxy_type == ProxyType::Http,
request_ip_host_http,
)
}
#[serial]
#[test]
fn test_socks5_dns() {
require_var("SOCKS5_SERVER");
run_test(
|test| test.proxy.proxy_type == ProxyType::Socks5,
request_example_https,
)
}
#[serial]
#[test]
fn test_http_dns() {
require_var("HTTP_SERVER");
run_test(
|test| test.proxy.proxy_type == ProxyType::Http,
request_example_https,
)
} }
} }