mirror of
https://github.com/fafhrd91/actix-net
synced 2024-12-19 15:32:38 +01:00
155 lines
4.5 KiB
Rust
Executable File
155 lines
4.5 KiB
Rust
Executable File
#![cfg(feature = "connect")]
|
|
|
|
use std::{
|
|
io,
|
|
net::{IpAddr, Ipv4Addr},
|
|
};
|
|
|
|
use actix_codec::{BytesCodec, Framed};
|
|
use actix_rt::net::TcpStream;
|
|
use actix_server::TestServer;
|
|
use actix_service::{fn_service, Service, ServiceFactory};
|
|
use bytes::Bytes;
|
|
use futures_util::sink::SinkExt;
|
|
|
|
use actix_tls::connect::{self as actix_connect, Connect};
|
|
|
|
#[cfg(feature = "openssl")]
|
|
#[actix_rt::test]
|
|
async fn test_string() {
|
|
let srv = TestServer::with(|| {
|
|
fn_service(|io: TcpStream| async {
|
|
let mut framed = Framed::new(io, BytesCodec);
|
|
framed.send(Bytes::from_static(b"test")).await?;
|
|
Ok::<_, io::Error>(())
|
|
})
|
|
});
|
|
|
|
let conn = actix_connect::default_connector();
|
|
let addr = format!("localhost:{}", srv.port());
|
|
let con = conn.call(addr.into()).await.unwrap();
|
|
assert_eq!(con.peer_addr().unwrap(), srv.addr());
|
|
}
|
|
|
|
#[cfg(feature = "rustls")]
|
|
#[actix_rt::test]
|
|
async fn test_rustls_string() {
|
|
let srv = TestServer::with(|| {
|
|
fn_service(|io: TcpStream| async {
|
|
let mut framed = Framed::new(io, BytesCodec);
|
|
framed.send(Bytes::from_static(b"test")).await?;
|
|
Ok::<_, io::Error>(())
|
|
})
|
|
});
|
|
|
|
let conn = actix_connect::default_connector();
|
|
let addr = format!("localhost:{}", srv.port());
|
|
let con = conn.call(addr.into()).await.unwrap();
|
|
assert_eq!(con.peer_addr().unwrap(), srv.addr());
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_static_str() {
|
|
let srv = TestServer::with(|| {
|
|
fn_service(|io: TcpStream| async {
|
|
let mut framed = Framed::new(io, BytesCodec);
|
|
framed.send(Bytes::from_static(b"test")).await?;
|
|
Ok::<_, io::Error>(())
|
|
})
|
|
});
|
|
|
|
let conn = actix_connect::default_connector();
|
|
|
|
let con = conn
|
|
.call(Connect::with_addr("10", srv.addr()))
|
|
.await
|
|
.unwrap();
|
|
assert_eq!(con.peer_addr().unwrap(), srv.addr());
|
|
|
|
let connect = Connect::new(srv.host().to_owned());
|
|
|
|
let conn = actix_connect::default_connector();
|
|
let con = conn.call(connect).await;
|
|
assert!(con.is_err());
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_new_service() {
|
|
let srv = TestServer::with(|| {
|
|
fn_service(|io: TcpStream| async {
|
|
let mut framed = Framed::new(io, BytesCodec);
|
|
framed.send(Bytes::from_static(b"test")).await?;
|
|
Ok::<_, io::Error>(())
|
|
})
|
|
});
|
|
|
|
let factory = actix_connect::default_connector_factory();
|
|
|
|
let conn = factory.new_service(()).await.unwrap();
|
|
let con = conn
|
|
.call(Connect::with_addr("10", srv.addr()))
|
|
.await
|
|
.unwrap();
|
|
assert_eq!(con.peer_addr().unwrap(), srv.addr());
|
|
}
|
|
|
|
#[cfg(all(feature = "openssl", feature = "uri"))]
|
|
#[actix_rt::test]
|
|
async fn test_openssl_uri() {
|
|
use std::convert::TryFrom;
|
|
|
|
let srv = TestServer::with(|| {
|
|
fn_service(|io: TcpStream| async {
|
|
let mut framed = Framed::new(io, BytesCodec);
|
|
framed.send(Bytes::from_static(b"test")).await?;
|
|
Ok::<_, io::Error>(())
|
|
})
|
|
});
|
|
|
|
let conn = actix_connect::default_connector();
|
|
let addr = http::Uri::try_from(format!("https://localhost:{}", srv.port())).unwrap();
|
|
let con = conn.call(addr.into()).await.unwrap();
|
|
assert_eq!(con.peer_addr().unwrap(), srv.addr());
|
|
}
|
|
|
|
#[cfg(all(feature = "rustls", feature = "uri"))]
|
|
#[actix_rt::test]
|
|
async fn test_rustls_uri() {
|
|
use std::convert::TryFrom;
|
|
|
|
let srv = TestServer::with(|| {
|
|
fn_service(|io: TcpStream| async {
|
|
let mut framed = Framed::new(io, BytesCodec);
|
|
framed.send(Bytes::from_static(b"test")).await?;
|
|
Ok::<_, io::Error>(())
|
|
})
|
|
});
|
|
|
|
let conn = actix_connect::default_connector();
|
|
let addr = http::Uri::try_from(format!("https://localhost:{}", srv.port())).unwrap();
|
|
let con = conn.call(addr.into()).await.unwrap();
|
|
assert_eq!(con.peer_addr().unwrap(), srv.addr());
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_local_addr() {
|
|
let srv = TestServer::with(|| {
|
|
fn_service(|io: TcpStream| async {
|
|
let mut framed = Framed::new(io, BytesCodec);
|
|
framed.send(Bytes::from_static(b"test")).await?;
|
|
Ok::<_, io::Error>(())
|
|
})
|
|
});
|
|
|
|
let conn = actix_connect::default_connector();
|
|
let local = IpAddr::V4(Ipv4Addr::new(127, 0, 0, 3));
|
|
|
|
let (con, _) = conn
|
|
.call(Connect::with_addr("10", srv.addr()).set_local_addr(local))
|
|
.await
|
|
.unwrap()
|
|
.into_parts();
|
|
|
|
assert_eq!(con.local_addr().unwrap().ip(), local)
|
|
}
|