mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-25 00:00:18 +02:00
Carl Lerche
066965cd59
Uses the infrastructure added by #2828 to enable switching `TcpListener::accept` to use `&self`. This also switches `poll_accept` to use `&self`. While doing introduces a hazard, `poll_*` style functions are considered low-level. Most users will use the `async fn` variants which are more misuse-resistant. TcpListener::incoming() is temporarily removed as it has the same problem as `TcpSocket::by_ref()` and will be implemented later.
61 lines
1.5 KiB
Rust
61 lines
1.5 KiB
Rust
#![warn(rust_2018_idioms)]
|
|
#![cfg(feature = "full")]
|
|
|
|
use tokio::net::TcpSocket;
|
|
use tokio_test::assert_ok;
|
|
|
|
#[tokio::test]
|
|
async fn basic_usage_v4() {
|
|
// Create server
|
|
let addr = assert_ok!("127.0.0.1:0".parse());
|
|
let srv = assert_ok!(TcpSocket::new_v4());
|
|
assert_ok!(srv.bind(addr));
|
|
|
|
let srv = assert_ok!(srv.listen(128));
|
|
|
|
// Create client & connect
|
|
let addr = srv.local_addr().unwrap();
|
|
let cli = assert_ok!(TcpSocket::new_v4());
|
|
let _cli = assert_ok!(cli.connect(addr).await);
|
|
|
|
// Accept
|
|
let _ = assert_ok!(srv.accept().await);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn basic_usage_v6() {
|
|
// Create server
|
|
let addr = assert_ok!("[::1]:0".parse());
|
|
let srv = assert_ok!(TcpSocket::new_v6());
|
|
assert_ok!(srv.bind(addr));
|
|
|
|
let srv = assert_ok!(srv.listen(128));
|
|
|
|
// Create client & connect
|
|
let addr = srv.local_addr().unwrap();
|
|
let cli = assert_ok!(TcpSocket::new_v6());
|
|
let _cli = assert_ok!(cli.connect(addr).await);
|
|
|
|
// Accept
|
|
let _ = assert_ok!(srv.accept().await);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn bind_before_connect() {
|
|
// Create server
|
|
let any_addr = assert_ok!("127.0.0.1:0".parse());
|
|
let srv = assert_ok!(TcpSocket::new_v4());
|
|
assert_ok!(srv.bind(any_addr));
|
|
|
|
let srv = assert_ok!(srv.listen(128));
|
|
|
|
// Create client & connect
|
|
let addr = srv.local_addr().unwrap();
|
|
let cli = assert_ok!(TcpSocket::new_v4());
|
|
assert_ok!(cli.bind(any_addr));
|
|
let _cli = assert_ok!(cli.connect(addr).await);
|
|
|
|
// Accept
|
|
let _ = assert_ok!(srv.accept().await);
|
|
}
|