1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-10-21 17:33:59 +00:00

feature: allow TestServer to open a websocket on any URL

* added `TestServer::ws_at(uri_str)`
* modified `TestServer::ws()` to call `self.ws_at("/")` to preserve
behavior

Closes #432
This commit is contained in:
jrconlin 2018-07-31 16:21:18 -07:00
parent 2071ea0532
commit aa1e75f071
No known key found for this signature in database
GPG key ID: 6E7B344D108DBD3D

View file

@ -207,13 +207,21 @@ impl TestServer {
self.rt.block_on(fut) self.rt.block_on(fut)
} }
/// Connect to websocket server /// Connect to websocket server at a given path
pub fn ws_at(
&mut self,
path: &str,
) -> Result<(ws::ClientReader, ws::ClientWriter), ws::ClientError> {
let url = self.url(path);
self.rt
.block_on(ws::Client::with_connector(url, self.conn.clone()).connect())
}
/// Connect to a websocket server
pub fn ws( pub fn ws(
&mut self, &mut self,
) -> Result<(ws::ClientReader, ws::ClientWriter), ws::ClientError> { ) -> Result<(ws::ClientReader, ws::ClientWriter), ws::ClientError> {
let url = self.url("/"); self.ws_at("/")
self.rt
.block_on(ws::Client::with_connector(url, self.conn.clone()).connect())
} }
/// Create `GET` request /// Create `GET` request