// file: crates/ksp-onchain-transport-lib/unit_tests/rpc_canary.rs // version: 2 fn pool_for_url(url: &str) -> crate::HttpTransportPool { let role = crate::HttpEndpointRoleSettings::new( crate::HttpRoleName::new("default"), true, std::vec![crate::HttpRequestKind::wildcard()], 10, crate::HttpRoleLimits::new(std::option::Option::None, std::option::Option::None, std::option::Option::None, std::option::Option::None), ); let endpoint = crate::HttpEndpointSettings::new( "fixture", true, crate::HttpProviderName::new("fixture"), crate::HttpClusterName::new("local"), crate::HttpEndpointUrl::parse(url).expect("fixture URL must parse"), std::time::Duration::from_secs(1), std::time::Duration::from_secs(1), std::option::Option::Some(1), std::vec![role], ); let settings = crate::HttpTransportSettings::new( std::vec![endpoint], crate::HttpRetrySettings::new(0, std::time::Duration::from_millis(1), std::time::Duration::from_millis(1)), ); return crate::HttpTransportPool::new(settings).expect("fixture pool must build"); } fn serve_once(body: &'static str) -> (std::string::String, std::thread::JoinHandle) { let listener = std::net::TcpListener::bind("127.0.0.1:0").expect("fixture listener must bind"); let address = listener.local_addr().expect("fixture listener address must resolve"); let handle = std::thread::spawn(move || { let (mut stream, _) = listener.accept().expect("fixture server must accept one request"); let request = read_request(&mut stream); let response = format!("HTTP/1.1 200 OK\r\nContent-Type: application/json\r\nContent-Length: {}\r\nConnection: close\r\n\r\n{}", body.len(), body); std::io::Write::write_all(&mut stream, response.as_bytes()).expect("fixture response must write"); return request; }); return (format!("http://{address}"), handle); } fn read_request(stream: &mut std::net::TcpStream) -> std::string::String { let mut bytes = std::vec::Vec::new(); let mut buffer = [0_u8; 1024]; loop { let count = std::io::Read::read(stream, &mut buffer).expect("fixture request must read"); if count == 0 { break; } bytes.extend_from_slice(&buffer[..count]); if request_complete(bytes.as_slice()) { break; } } return std::string::String::from_utf8(bytes).expect("fixture request must be UTF-8"); } fn request_complete(bytes: &[u8]) -> bool { let text = match std::str::from_utf8(bytes) { std::result::Result::Ok(text) => text, std::result::Result::Err(_) => return false, }; let header_end = match text.find("\r\n\r\n") { std::option::Option::Some(value) => value, std::option::Option::None => return false, }; let mut content_length = 0_usize; for line in text[..header_end].lines() { let (name, value) = match line.split_once(':') { std::option::Option::Some(parts) => parts, std::option::Option::None => continue, }; if name.eq_ignore_ascii_case("content-length") { content_length = value.trim().parse::().expect("content length must parse"); } } return bytes.len() >= header_end.saturating_add(4).saturating_add(content_length); } fn request_body(request: &str) -> serde_json::Value { let body = request.split("\r\n\r\n").nth(1).expect("fixture request body must exist"); return serde_json::from_str(body).expect("fixture request body must be JSON"); } #[tokio::test(flavor = "current_thread")] async fn typed_get_health_executes_real_http_fixture() { let (url, handle) = serve_once(include_str!("../fixtures/http/get_health.success.json")); let pool = pool_for_url(url.as_str()); let result = pool.get_health(&crate::HttpRoleName::new("default")).await.expect("getHealth fixture must succeed"); assert_eq!(result, crate::SolanaNodeHealth::Healthy); let request = handle.join().expect("fixture server must join"); let body = request_body(request.as_str()); assert_eq!(body["method"], serde_json::json!("getHealth")); } #[tokio::test(flavor = "current_thread")] async fn typed_get_genesis_hash_executes_real_http_fixture() { let (url, handle) = serve_once(include_str!("../fixtures/http/get_genesis_hash.success.json")); let pool = pool_for_url(url.as_str()); let result = pool.get_genesis_hash(&crate::HttpRoleName::new("default")).await.expect("getGenesisHash fixture must succeed"); assert_eq!(result.as_str(), "GH7ome3EiwEr7tu9JuTh2dpYWBJK3z69Xm1ZE3MEE6JC"); let request = handle.join().expect("fixture server must join"); assert_eq!(request_body(request.as_str())["method"], serde_json::json!("getGenesisHash")); } #[tokio::test(flavor = "current_thread")] async fn typed_get_version_executes_real_http_fixture() { let (url, handle) = serve_once(include_str!("../fixtures/http/get_version.success.json")); let pool = pool_for_url(url.as_str()); let result = pool.get_version(&crate::HttpRoleName::new("default")).await.expect("getVersion fixture must succeed"); assert_eq!(result.solana_core(), "3.1.8"); assert_eq!(result.feature_set(), std::option::Option::Some(2_891_131_721)); let request = handle.join().expect("fixture server must join"); assert_eq!(request_body(request.as_str())["method"], serde_json::json!("getVersion")); } #[tokio::test(flavor = "current_thread")] async fn typed_get_balance_executes_real_http_fixture_and_encodes_config() { let (url, handle) = serve_once(include_str!("../fixtures/http/get_balance.success.json")); let pool = pool_for_url(url.as_str()); let account = "11111111111111111111111111111111".parse::().expect("system address must parse"); let config = crate::GetBalanceConfig::new(std::option::Option::Some(crate::SolanaCommitment::Finalized), std::option::Option::Some(123)); let result = pool .get_balance(&crate::HttpRoleName::new("default"), &account, std::option::Option::Some(&config)) .await .expect("getBalance fixture must succeed"); assert_eq!(result.value(), 424_242); assert_eq!(result.context().slot(), 123_456_789); assert_eq!(result.context().api_version(), std::option::Option::Some("3.1.8")); let request = handle.join().expect("fixture server must join"); let body = request_body(request.as_str()); assert_eq!(body["method"], serde_json::json!("getBalance")); assert_eq!(body["params"][0], serde_json::json!("11111111111111111111111111111111")); assert_eq!(body["params"][1]["commitment"], serde_json::json!("finalized")); assert_eq!(body["params"][1]["minContextSlot"], serde_json::json!(123)); }