test: Fix timeout assertions

Author: shahnami

tests/integration/blockchain/transports/midnight/ws.rs

@@ -330,7 +330,24 @@ async fn test_websocket_connection_timeout() {
 	let invalid_url = "ws://invalid-domain-that-does-not-exist:12345";
 
 	let network = create_test_network_with_urls(vec![invalid_url]);
+
+	// Measure time to ensure it fails quickly due to timeout
+	let start = std::time::Instant::now();
 	let result = MidnightWsTransportClient::new(&network, Some(config)).await;
-	assert!(result.is_err());
-	assert!(result.unwrap_err().to_string().contains("timeout"));
+	let elapsed = start.elapsed();
+
+	assert!(result.is_err(), "Should fail with invalid URL");
+	// Verify it failed quickly (within 100ms) which indicates the timeout worked
+	assert!(
+		elapsed < Duration::from_millis(10),
+		"Should fail quickly due to timeout, took {:?}",
+		elapsed
+	);
+	// The error should indicate connection failure
+	let error_msg = result.unwrap_err().to_string();
+	assert!(
+		error_msg.contains("Failed to connect"),
+		"Error should indicate connection failure, got: {}",
+		error_msg
+	);
 }

tests/integration/blockchain/transports/ws/transport.rs

@@ -363,7 +363,24 @@ async fn test_websocket_connection_timeout() {
 	let invalid_url = "ws://invalid-domain-that-does-not-exist:12345";
 
 	let network = create_test_network_with_urls(vec![invalid_url]);
+
+	// Measure time to ensure it fails quickly due to timeout
+	let start = std::time::Instant::now();
 	let result = WsTransportClient::new(&network, Some(config)).await;
-	assert!(result.is_err());
-	assert!(result.unwrap_err().to_string().contains("timeout"));
+	let elapsed = start.elapsed();
+
+	assert!(result.is_err(), "Should fail with invalid URL");
+	// Verify it failed quickly (within 100ms) which indicates the timeout worked
+	assert!(
+		elapsed < Duration::from_millis(10),
+		"Should fail quickly due to timeout, took {:?}",
+		elapsed
+	);
+	// The error should indicate connection failure
+	let error_msg = result.unwrap_err().to_string();
+	assert!(
+		error_msg.contains("Failed to connect"),
+		"Error should indicate connection failure, got: {}",
+		error_msg
+	);
 }