Skip to content

Commit 7d33e89

Browse files
committed
refactor(aggregator-client): rename query/certificate mod to get
Since there will be arround ~20 requests maximum, making a module by subject will result on a lots of modules with one or two files, not that helpful. + rename `CertificateDetailsQuery` to `GetCertificateQuery` to have an uniform convention: `{Verb}{Subject}Query`.
1 parent 0625621 commit 7d33e89

File tree

6 files changed

+35
-32
lines changed

6 files changed

+35
-32
lines changed

internal/mithril-aggregator-client/src/query/certificate/mod.rs

Lines changed: 0 additions & 3 deletions
This file was deleted.

internal/mithril-aggregator-client/src/query/get/get_aggregator_features.rs

Whitespace-only changes.
Lines changed: 16 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,12 @@ use crate::AggregatorClientResult;
99
use crate::error::AggregatorClientError;
1010
use crate::query::{AggregatorQuery, QueryContext, QueryMethod};
1111

12-
/// Get the details of a certificate
13-
pub struct CertificateDetailsQuery {
12+
/// Query to get the details of a certificate
13+
pub struct GetCertificateQuery {
1414
hash: String,
1515
}
1616

17-
impl CertificateDetailsQuery {
17+
impl GetCertificateQuery {
1818
/// Instantiate a query to get a certificate by hash
1919
pub fn by_hash<H: Into<String>>(hash: H) -> Self {
2020
Self { hash: hash.into() }
@@ -30,7 +30,7 @@ impl CertificateDetailsQuery {
3030

3131
#[cfg_attr(target_family = "wasm", async_trait(?Send))]
3232
#[cfg_attr(not(target_family = "wasm"), async_trait)]
33-
impl AggregatorQuery for CertificateDetailsQuery {
33+
impl AggregatorQuery for GetCertificateQuery {
3434
type Response = Option<CertificateMessage>;
3535
type Body = ();
3636

@@ -65,7 +65,7 @@ mod tests {
6565

6666
use mithril_common::test::double::Dummy;
6767

68-
use crate::test::setup_server_and_client;
68+
use crate::test::{assert_error_matches, setup_server_and_client};
6969

7070
use super::*;
7171

@@ -79,7 +79,7 @@ mod tests {
7979
});
8080

8181
let fetched_message = client
82-
.send(CertificateDetailsQuery::by_hash(&expected_message.hash))
82+
.send(GetCertificateQuery::by_hash(&expected_message.hash))
8383
.await
8484
.unwrap();
8585

@@ -94,10 +94,7 @@ mod tests {
9494
then.status(404);
9595
});
9696

97-
let fetched_message = client
98-
.send(CertificateDetailsQuery::by_hash("whatever"))
99-
.await
100-
.unwrap();
97+
let fetched_message = client.send(GetCertificateQuery::by_hash("whatever")).await.unwrap();
10198

10299
assert_eq!(None, fetched_message);
103100
}
@@ -110,14 +107,12 @@ mod tests {
110107
then.status(500).body("an error occurred");
111108
});
112109

113-
match client
114-
.send(CertificateDetailsQuery::by_hash("whatever"))
110+
let error = client
111+
.send(GetCertificateQuery::by_hash("whatever"))
115112
.await
116-
.unwrap_err()
117-
{
118-
AggregatorClientError::RemoteServerTechnical(_) => (),
119-
e => panic!("Expected Aggregator::RemoteServerTechnical error, got '{e:?}'."),
120-
};
113+
.unwrap_err();
114+
115+
assert_error_matches!(error, AggregatorClientError::RemoteServerTechnical(_));
121116
}
122117

123118
#[tokio::test]
@@ -129,7 +124,7 @@ mod tests {
129124
then.status(200).body(json!(genesis_message).to_string());
130125
});
131126

132-
let fetched = client.send(CertificateDetailsQuery::latest_genesis()).await.unwrap();
127+
let fetched = client.send(GetCertificateQuery::latest_genesis()).await.unwrap();
133128

134129
assert_eq!(Some(genesis_message), fetched);
135130
}
@@ -142,7 +137,7 @@ mod tests {
142137
then.status(404);
143138
});
144139

145-
let fetched = client.send(CertificateDetailsQuery::latest_genesis()).await.unwrap();
140+
let fetched = client.send(GetCertificateQuery::latest_genesis()).await.unwrap();
146141

147142
assert_eq!(None, fetched);
148143
}
@@ -155,14 +150,8 @@ mod tests {
155150
then.status(500).body("an error occurred");
156151
});
157152

158-
let error = client
159-
.send(CertificateDetailsQuery::latest_genesis())
160-
.await
161-
.unwrap_err();
153+
let error = client.send(GetCertificateQuery::latest_genesis()).await.unwrap_err();
162154

163-
assert!(
164-
matches!(error, AggregatorClientError::RemoteServerTechnical(_)),
165-
"Expected Aggregator::RemoteServerTechnical error, got {error:?}"
166-
);
155+
assert_error_matches!(error, AggregatorClientError::RemoteServerTechnical(_));
167156
}
168157
}
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
mod get_certificate;
2+
3+
pub use get_certificate::*;

internal/mithril-aggregator-client/src/query/mod.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
//! - Certificate: Get by hash, get latest genesis certificate
55
//!
66
mod api;
7-
mod certificate;
7+
mod get;
88

99
pub(crate) use api::*;
10-
pub use certificate::*;
10+
pub use get::*;

internal/mithril-aggregator-client/src/test/mod.rs

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,3 +15,17 @@ pub(crate) fn setup_server_and_client() -> (MockServer, AggregatorClient) {
1515

1616
(server, client)
1717
}
18+
19+
#[cfg(test)]
20+
macro_rules! assert_error_matches {
21+
($error:expr, $error_type:pat) => {
22+
assert!(
23+
matches!($error, $error_type),
24+
"Expected {} error, got '{:?}'.",
25+
stringify!($error_type),
26+
$error
27+
);
28+
};
29+
}
30+
#[cfg(test)]
31+
pub(crate) use assert_error_matches;

0 commit comments

Comments
 (0)