63 lines
2.2 KiB
Rust
63 lines
2.2 KiB
Rust
use tokio::task;
|
|
|
|
use crate::tcp_relay_client::client_config::ClientConfig;
|
|
use crate::tcp_relay_server::server_config::ServerConfig;
|
|
use crate::test::dummy_tcp_sockets::{
|
|
dummy_tcp_client_square_root_requests, wait_for_port, DummyTCPServer,
|
|
};
|
|
use crate::test::pki::Pki;
|
|
use crate::test::{get_port_number, PortsAllocation, LOCALHOST_IP};
|
|
|
|
const VALID_TOKEN: &str = "AvalidTOKEN";
|
|
|
|
fn port(index: u16) -> u16 {
|
|
get_port_number(PortsAllocation::ValidWithTokenAuthAndServerTLS, index)
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 5)]
|
|
async fn test() {
|
|
let _ = env_logger::builder().is_test(true).try_init();
|
|
|
|
// Start internal service
|
|
let local_server = DummyTCPServer::start(port(1)).await;
|
|
tokio::spawn(async move {
|
|
local_server.loop_conn_square_operations().await;
|
|
});
|
|
|
|
let pki = Pki::load();
|
|
|
|
let local_set = task::LocalSet::new();
|
|
local_set
|
|
.run_until(async move {
|
|
wait_for_port(port(1)).await;
|
|
|
|
// Start server relay
|
|
task::spawn_local(crate::tcp_relay_server::run_app(ServerConfig {
|
|
tokens: vec![VALID_TOKEN.to_string()],
|
|
tokens_file: None,
|
|
ports: vec![port(1)],
|
|
upstream_server: "127.0.0.1".to_string(),
|
|
listen_address: format!("127.0.0.1:{}", port(0)),
|
|
increment_ports: 1,
|
|
tls_cert: Some(pki.localhost_crt.file_path()),
|
|
tls_key: Some(pki.localhost_key.file_path()),
|
|
tls_client_auth_root_cert: None,
|
|
tls_revocation_list: None,
|
|
}));
|
|
wait_for_port(port(0)).await;
|
|
|
|
// Start client relay
|
|
task::spawn(crate::tcp_relay_client::run_app(ClientConfig {
|
|
token: Some(VALID_TOKEN.to_string()),
|
|
relay_url: format!("https://localhost:{}", port(0)),
|
|
listen_address: LOCALHOST_IP.to_string(),
|
|
root_certificate: Some(pki.root_ca_crt.file_path()),
|
|
..Default::default()
|
|
}));
|
|
wait_for_port(port(2)).await;
|
|
|
|
dummy_tcp_client_square_root_requests(port(2), 10).await;
|
|
})
|
|
.await;
|
|
}
|