use std::io;
use bytes::Bytes;
use futures::SinkExt;
use ntex::codec::{BytesCodec, Framed};
use ntex::connect::{Connect, ResolverConfig, ResolverOpts};
use ntex::rt::net::TcpStream;
use ntex::server::test_server;
use ntex::service::{fn_service, Service, ServiceFactory};
#[cfg(feature = "openssl")]
#[ntex::test]
async fn test_string() {
let srv = test_server(|| {
fn_service(|io: TcpStream| async {
let mut framed = Framed::new(io, BytesCodec);
framed.send(Bytes::from_static(b"test")).await.unwrap();
Ok::<_, io::Error>(())
})
});
let conn = ntex::connect::Connector::default();
let addr = format!("localhost:{}", srv.addr().port());
let con = conn.call(addr.into()).await.unwrap();
assert_eq!(con.peer_addr().unwrap(), srv.addr());
}
#[cfg(feature = "rustls")]
#[ntex::test]
async fn test_rustls_string() {
let srv = test_server(|| {
fn_service(|io: TcpStream| async {
let mut framed = Framed::new(io, BytesCodec);
framed.send(Bytes::from_static(b"test")).await.unwrap();
Ok::<_, io::Error>(())
})
});
let conn = ntex::connect::Connector::default();
let addr = format!("localhost:{}", srv.addr().port());
let con = conn.call(addr.into()).await.unwrap();
assert_eq!(con.peer_addr().unwrap(), srv.addr());
}
#[ntex::test]
async fn test_static_str() {
let srv = test_server(|| {
fn_service(|io: TcpStream| async {
let mut framed = Framed::new(io, BytesCodec);
framed.send(Bytes::from_static(b"test")).await.unwrap();
Ok::<_, io::Error>(())
})
});
let resolver = ntex::connect::default_resolver();
let conn = ntex::connect::Connector::new(resolver.clone());
let con = conn.call(Connect::with("10", srv.addr())).await.unwrap();
assert_eq!(con.peer_addr().unwrap(), srv.addr());
let connect = Connect::new("127.0.0.1".to_owned());
let conn = ntex::connect::Connector::new(resolver);
let con = conn.call(connect).await;
assert!(con.is_err());
}
#[ntex::test]
async fn test_new_service() {
let srv = test_server(|| {
fn_service(|io: TcpStream| async {
let mut framed = Framed::new(io, BytesCodec);
framed.send(Bytes::from_static(b"test")).await.unwrap();
Ok::<_, io::Error>(())
})
});
let resolver = ntex::connect::start_resolver(
ResolverConfig::default(),
ResolverOpts::default(),
);
let factory = ntex::connect::Connector::new(resolver);
let conn = factory.new_service(()).await.unwrap();
let con = conn.call(Connect::with("10", srv.addr())).await.unwrap();
assert_eq!(con.peer_addr().unwrap(), srv.addr());
}
#[cfg(feature = "openssl")]
#[ntex::test]
async fn test_uri() {
use std::convert::TryFrom;
let srv = test_server(|| {
fn_service(|io: TcpStream| async {
let mut framed = Framed::new(io, BytesCodec);
framed.send(Bytes::from_static(b"test")).await.unwrap();
Ok::<_, io::Error>(())
})
});
let conn = ntex::connect::Connector::default();
let addr =
ntex::http::Uri::try_from(format!("https://localhost:{}", srv.addr().port()))
.unwrap();
let con = conn.call(addr.into()).await.unwrap();
assert_eq!(con.peer_addr().unwrap(), srv.addr());
}
#[cfg(feature = "rustls")]
#[ntex::test]
async fn test_rustls_uri() {
use std::convert::TryFrom;
let srv = test_server(|| {
fn_service(|io: TcpStream| async {
let mut framed = Framed::new(io, BytesCodec);
framed.send(Bytes::from_static(b"test")).await.unwrap();
Ok::<_, io::Error>(())
})
});
let conn = ntex::connect::Connector::default();
let addr =
ntex::http::Uri::try_from(format!("https://localhost:{}", srv.addr().port()))
.unwrap();
let con = conn.call(addr.into()).await.unwrap();
assert_eq!(con.peer_addr().unwrap(), srv.addr());
}