From 7d11aa866837eea50a6f1e0ef7e24846a653cbf1 Mon Sep 17 00:00:00 2001
From: Zeki Sherif <9832640+zekisherif@users.noreply.github.com>
Date: Tue, 17 Nov 2020 11:58:00 -0600
Subject: net: add SO_LINGER get/set to TcpStream (#3143)

---
 tokio/tests/tcp_stream.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'tokio/tests')

diff --git a/tokio/tests/tcp_stream.rs b/tokio/tests/tcp_stream.rs
index 84d58dc5..58b06ee3 100644
--- a/tokio/tests/tcp_stream.rs
+++ b/tokio/tests/tcp_stream.rs
@@ -9,9 +9,25 @@ use tokio_test::{assert_ok, assert_pending, assert_ready_ok};
 
 use std::io;
 use std::task::Poll;
+use std::time::Duration;
 
 use futures::future::poll_fn;
 
+#[tokio::test]
+async fn set_linger() {
+    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
+
+    let stream = TcpStream::connect(listener.local_addr().unwrap())
+        .await
+        .unwrap();
+
+    assert_ok!(stream.set_linger(Some(Duration::from_secs(1))));
+    assert_eq!(stream.linger().unwrap().unwrap().as_secs(), 1);
+
+    assert_ok!(stream.set_linger(None));
+    assert!(stream.linger().unwrap().is_none());
+}
+
 #[tokio::test]
 async fn try_read_write() {
     const DATA: &[u8] = b"this is some data to write to the socket";
-- 
cgit v1.2.3