1 files changed, 74 insertions, 0 deletions
diff --git a/third_party/rust/tokio/tests/tcp_socket.rs b/third_party/rust/tokio/tests/tcp_socket.rs
new file mode 100644
index 0000000000..3030416502
--- /dev/null
+++ b/third_party/rust/tokio/tests/tcp_socket.rs
@@ -0,0 +1,74 @@
+#![warn(rust_2018_idioms)]
+#![cfg(feature = "full")]
+
+use std::time::Duration;
+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);
+}
+
+#[tokio::test]
+async fn basic_linger() {
+    // Create server
+    let addr = assert_ok!("127.0.0.1:0".parse());
+    let srv = assert_ok!(TcpSocket::new_v4());
+    assert_ok!(srv.bind(addr));
+
+    assert!(srv.linger().unwrap().is_none());
+
+    srv.set_linger(Some(Duration::new(0, 0))).unwrap();
+    assert_eq!(srv.linger().unwrap(), Some(Duration::new(0, 0)));
+}