benedettadavico
93 lines
2.9 KiB
Rust
93 lines
2.9 KiB
Rust
use reqwest::Client;
|
|
use serde_json::Value;
|
|
use reqwest::Response;
|
|
|
|
pub fn test_client() -> Client {
|
|
Client::new()
|
|
}
|
|
|
|
pub fn base_url() -> String {
|
|
std::env::var("API_BASE_URL")
|
|
.unwrap_or_else(|_| "https://sandbox-nym-api1.nymtech.net/api".into())
|
|
}
|
|
|
|
pub async fn validate_json_response(res: Response) -> Value {
|
|
assert!(
|
|
res.status().is_success(),
|
|
"Expected 2xx but got {}",
|
|
res.status()
|
|
);
|
|
let json: Value = res
|
|
.json()
|
|
.await
|
|
.unwrap_or_else(|err| panic!("Invalid JSON response: {}", err));
|
|
json
|
|
}
|
|
|
|
pub async fn get_any_node_id() -> String {
|
|
let url = format!("{}/v1/nym-nodes/bonded", base_url());
|
|
let res = test_client().get(&url).send().await.unwrap();
|
|
let json: Value = res.json().await.unwrap();
|
|
|
|
json.get("data")
|
|
.and_then(|list| list.as_array())
|
|
.and_then(|arr| arr.first())
|
|
.and_then(|node| node.get("bond_information"))
|
|
.and_then(|n| n.get("node_id"))
|
|
.and_then(|v| v.as_u64())
|
|
.map(|id| id.to_string())
|
|
.unwrap_or_else(|| "INVALID_ID".into())
|
|
.to_string()
|
|
}
|
|
|
|
pub async fn get_mixnode_node_id() -> u64 {
|
|
let url = format!("{}/v1/nym-nodes/described", base_url());
|
|
let res = test_client().get(&url).send().await.unwrap();
|
|
let json: Value = res.json().await.unwrap();
|
|
|
|
json.get("data")
|
|
.and_then(|v| v.as_array())
|
|
.expect("Expected 'data' to be an array")
|
|
.iter()
|
|
.find(|entry| {
|
|
entry
|
|
.get("description")
|
|
.and_then(|d| d.get("declared_role"))
|
|
.and_then(|r| r.get("mixnode"))
|
|
.and_then(|m| m.as_bool())
|
|
.map(|is_mixnode| is_mixnode)
|
|
.unwrap_or(true)
|
|
})
|
|
.and_then(|node| node.get("node_id").and_then(|v| v.as_u64()))
|
|
.expect("Unable to find mixnode node id")
|
|
}
|
|
|
|
pub async fn get_gateway_identity_key() -> String {
|
|
let url = format!("{}/v1/nym-nodes/described", base_url());
|
|
let res = test_client().get(&url).send().await.unwrap();
|
|
let json: Value = res.json().await.unwrap();
|
|
|
|
json.get("data")
|
|
.and_then(|v| v.as_array())
|
|
.expect("Expected 'data' to be an array")
|
|
.iter()
|
|
.find(|entry| {
|
|
entry
|
|
.get("description")
|
|
.and_then(|d| d.get("declared_role"))
|
|
.and_then(|r| r.get("mixnode"))
|
|
.and_then(|m| m.as_bool())
|
|
.map(|is_mixnode| !is_mixnode) // we want gateways, not mixnodes
|
|
.unwrap_or(false)
|
|
})
|
|
.and_then(|node| {
|
|
node.get("description")
|
|
.and_then(|d| d.get("host_information"))
|
|
.and_then(|h| h.get("keys"))
|
|
.and_then(|k| k.get("ed25519"))
|
|
.and_then(|v| v.as_str())
|
|
})
|
|
.expect("Unable to find gateway identity key with mixnode = false")
|
|
.to_string()
|
|
}
|