From b01e619f8d3a36091e9f54005040a688a9f6a0fb Mon Sep 17 00:00:00 2001
From: "error.d" <error.d@gmail.com>
Date: Tue, 28 Apr 2020 18:33:33 +0800
Subject: [PATCH] update timeout to examples

---
 examples/autobahn-client.rs | 5 +++--
 examples/client.rs          | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/examples/autobahn-client.rs b/examples/autobahn-client.rs
index 31aedb2..b0b3bb4 100644
--- a/examples/autobahn-client.rs
+++ b/examples/autobahn-client.rs
@@ -6,7 +6,7 @@ use tungstenite::{connect, Error, Message, Result};
 const AGENT: &str = "Tungstenite";
 
 fn get_case_count() -> Result<u32> {
-    let (mut socket, _) = connect(Url::parse("ws://localhost:9001/getCaseCount").unwrap())?;
+    let (mut socket, _) = connect(Url::parse("ws://localhost:9001/getCaseCount").unwrap(), None)?;
     let msg = socket.read_message()?;
     socket.close(None)?;
     Ok(msg.into_text()?.parse::<u32>().unwrap())
@@ -19,6 +19,7 @@ fn update_reports() -> Result<()> {
             AGENT
         ))
         .unwrap(),
+        None
     )?;
     socket.close(None)?;
     Ok(())
@@ -31,7 +32,7 @@ fn run_test(case: u32) -> Result<()> {
         case, AGENT
     ))
     .unwrap();
-    let (mut socket, _) = connect(case_url)?;
+    let (mut socket, _) = connect(case_url, None)?;
     loop {
         match socket.read_message()? {
             msg @ Message::Text(_) | msg @ Message::Binary(_) => {
diff --git a/examples/client.rs b/examples/client.rs
index 7938cfb..132a579 100644
--- a/examples/client.rs
+++ b/examples/client.rs
@@ -5,7 +5,7 @@ fn main() {
     env_logger::init();
 
     let (mut socket, response) =
-        connect(Url::parse("ws://localhost:3012/socket").unwrap()).expect("Can't connect");
+        connect(Url::parse("ws://localhost:3012/socket").unwrap(), None).expect("Can't connect");
 
     println!("Connected to the server");
     println!("Response HTTP code: {}", response.status());