mirror of
https://github.com/fafhrd91/actix-net
synced 2025-06-27 01:37:42 +02:00
actix-tls release candidate prep (#422)
This commit is contained in:
@ -10,7 +10,9 @@ use actix_server::TestServer;
|
||||
use actix_service::{fn_service, Service, ServiceFactory};
|
||||
use futures_core::future::LocalBoxFuture;
|
||||
|
||||
use actix_tls::connect::{new_connector_factory, Connect, Resolve, Resolver};
|
||||
use actix_tls::connect::{
|
||||
ConnectError, ConnectInfo, Connection, Connector, Host, Resolve, Resolver,
|
||||
};
|
||||
|
||||
#[actix_rt::test]
|
||||
async fn custom_resolver() {
|
||||
@ -36,6 +38,18 @@ async fn custom_resolver() {
|
||||
|
||||
#[actix_rt::test]
|
||||
async fn custom_resolver_connect() {
|
||||
pub fn connector_factory<T: Host + 'static>(
|
||||
resolver: Resolver,
|
||||
) -> impl ServiceFactory<
|
||||
ConnectInfo<T>,
|
||||
Config = (),
|
||||
Response = Connection<T, TcpStream>,
|
||||
Error = ConnectError,
|
||||
InitError = (),
|
||||
> {
|
||||
Connector::new(resolver)
|
||||
}
|
||||
|
||||
use trust_dns_resolver::TokioAsyncResolver;
|
||||
|
||||
let srv =
|
||||
@ -68,12 +82,11 @@ async fn custom_resolver_connect() {
|
||||
trust_dns: TokioAsyncResolver::tokio_from_system_conf().unwrap(),
|
||||
};
|
||||
|
||||
let resolver = Resolver::new_custom(resolver);
|
||||
let factory = new_connector_factory(resolver);
|
||||
let factory = connector_factory(Resolver::custom(resolver));
|
||||
|
||||
let conn = factory.new_service(()).await.unwrap();
|
||||
let con = conn
|
||||
.call(Connect::with_addr("example.com", srv.addr()))
|
||||
.call(ConnectInfo::with_addr("example.com", srv.addr()))
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(con.peer_addr().unwrap(), srv.addr());
|
||||
|
Reference in New Issue
Block a user