@@ -4,36 +4,26 @@ use tokio_tungstenite::{
44 connect_async,
55 tungstenite:: { Error , Result } ,
66} ;
7- use url:: Url ;
87
98const AGENT : & str = "Tungstenite" ;
109
1110async fn get_case_count ( ) -> Result < u32 > {
12- let ( mut socket, _) = connect_async (
13- Url :: parse ( "ws://localhost:9001/getCaseCount" ) . expect ( "Can't connect to case count URL" ) ,
14- )
15- . await ?;
11+ let ( mut socket, _) = connect_async ( "ws://localhost:9001/getCaseCount" ) . await ?;
1612 let msg = socket. next ( ) . await . expect ( "Can't fetch case count" ) ?;
1713 socket. close ( None ) . await ?;
1814 Ok ( msg. into_text ( ) ?. parse :: < u32 > ( ) . expect ( "Can't parse case count" ) )
1915}
2016
2117async fn update_reports ( ) -> Result < ( ) > {
22- let ( mut socket, _) = connect_async (
23- Url :: parse ( & format ! ( "ws://localhost:9001/updateReports?agent={}" , AGENT ) )
24- . expect ( "Can't update reports" ) ,
25- )
26- . await ?;
18+ let ( mut socket, _) =
19+ connect_async ( & format ! ( "ws://localhost:9001/updateReports?agent={}" , AGENT ) ) . await ?;
2720 socket. close ( None ) . await ?;
2821 Ok ( ( ) )
2922}
3023
3124async fn run_test ( case : u32 ) -> Result < ( ) > {
3225 info ! ( "Running test case {}" , case) ;
33- let case_url =
34- Url :: parse ( & format ! ( "ws://localhost:9001/runCase?case={}&agent={}" , case, AGENT ) )
35- . expect ( "Bad testcase URL" ) ;
36-
26+ let case_url = & format ! ( "ws://localhost:9001/runCase?case={}&agent={}" , case, AGENT ) ;
3727 let ( mut ws_stream, _) = connect_async ( case_url) . await ?;
3828 while let Some ( msg) = ws_stream. next ( ) . await {
3929 let msg = msg?;
0 commit comments