mithril_aggregator_client/query/get/
get_epoch_settings.rs

1use async_trait::async_trait;
2use reqwest::StatusCode;
3
4use mithril_common::messages::EpochSettingsMessage;
5
6use crate::AggregatorHttpClientResult;
7use crate::query::{AggregatorQuery, QueryContext, QueryMethod, ResponseExt};
8
9/// Query to get the current epoch settings
10pub struct GetEpochSettingsQuery {}
11
12impl GetEpochSettingsQuery {
13    /// Instantiate a query to get the current epoch settings
14    pub fn current() -> Self {
15        Self {}
16    }
17}
18
19#[cfg_attr(target_family = "wasm", async_trait(?Send))]
20#[cfg_attr(not(target_family = "wasm"), async_trait)]
21impl AggregatorQuery for GetEpochSettingsQuery {
22    type Response = EpochSettingsMessage;
23    type Body = ();
24
25    fn method() -> QueryMethod {
26        QueryMethod::Get
27    }
28
29    fn route(&self) -> String {
30        "epoch-settings".to_string()
31    }
32
33    async fn handle_response(
34        &self,
35        context: QueryContext,
36    ) -> AggregatorHttpClientResult<Self::Response> {
37        match context.response.status() {
38            StatusCode::OK => context.response.parse_json().await,
39            _ => Err(context.unhandled_status_code().await),
40        }
41    }
42}
43
44#[cfg(test)]
45mod tests {
46    use serde_json::json;
47
48    use mithril_common::test::double::Dummy;
49
50    use crate::AggregatorHttpClientError;
51    use crate::test::{assert_error_matches, setup_server_and_client};
52
53    use super::*;
54
55    #[tokio::test]
56    async fn test_epoch_settings_ok_200() {
57        let (server, client) = setup_server_and_client();
58        let epoch_settings_expected = EpochSettingsMessage::dummy();
59        let _server_mock = server.mock(|when, then| {
60            when.path("/epoch-settings");
61            then.status(200).body(json!(epoch_settings_expected).to_string());
62        });
63
64        let epoch_settings = client.send(GetEpochSettingsQuery::current()).await.unwrap();
65        assert_eq!(epoch_settings_expected, epoch_settings);
66    }
67
68    #[tokio::test]
69    async fn test_epoch_settings_ko_500() {
70        let (server, client) = setup_server_and_client();
71        let _server_mock = server.mock(|when, then| {
72            when.any_request();
73            then.status(500).body("an error occurred");
74        });
75
76        let error = client.send(GetEpochSettingsQuery::current()).await.unwrap_err();
77
78        assert_error_matches!(error, AggregatorHttpClientError::RemoteServerTechnical(_));
79    }
80}
mithril_aggregator_client/query/get/get_epoch_settings.rs

mithril_aggregator_client/query/get/
get_epoch_settings.rs
