Skip to content

Commit da4c22b

Browse files
committed
cargo fmt + fix clippy
1 parent 3574923 commit da4c22b

File tree

8 files changed

+55
-33
lines changed

8 files changed

+55
-33
lines changed

benches/pbs/src/main.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ use std::time::{Duration, Instant};
33
use alloy::{primitives::B256, rpc::types::beacon::BlsPublicKey};
44
use cb_common::{
55
config::RelayConfig,
6-
pbs::{GetHeaderResponse, RelayClient, RelayEntry},
6+
pbs::{DenebSpec, GetHeaderResponse, RelayClient, RelayEntry},
77
signer::BlsSecretKey,
88
types::Chain,
99
utils::blst_pubkey_to_alloy,
@@ -63,7 +63,7 @@ async fn main() {
6363
.expect("failed to decode response");
6464

6565
assert!(
66-
serde_json::from_slice::<GetHeaderResponse>(&res).is_ok(),
66+
serde_json::from_slice::<GetHeaderResponse<DenebSpec>>(&res).is_ok(),
6767
"invalid header returned"
6868
);
6969

@@ -146,7 +146,7 @@ async fn start_mock_relay(chain: Chain, relay_config: RelayConfig) {
146146
let relay_port = relay_config.entry.url.port().expect("missing port");
147147

148148
let mock_relay = MockRelayState::new(chain, signer);
149-
start_mock_relay_service(mock_relay.into(), relay_port)
149+
start_mock_relay_service::<DenebSpec>(mock_relay.into(), relay_port)
150150
.await
151151
.expect("failed to start mock relay");
152152
}

tests/src/mock_relay.rs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@ use axum::{
1616
};
1717
use cb_common::{
1818
pbs::{
19-
GetHeaderParams, GetHeaderResponse, SubmitBlindedBlockResponse, BUILDER_API_PATH,
20-
GET_HEADER_PATH, GET_STATUS_PATH, REGISTER_VALIDATOR_PATH, SUBMIT_BLOCK_PATH, EthSpec
19+
EthSpec, GetHeaderParams, GetHeaderResponse, SubmitBlindedBlockResponse, BUILDER_API_PATH,
20+
GET_HEADER_PATH, GET_STATUS_PATH, REGISTER_VALIDATOR_PATH, SUBMIT_BLOCK_PATH,
2121
},
2222
signature::sign_builder_root,
2323
signer::BlsSecretKey,
@@ -29,7 +29,10 @@ use tokio::net::TcpListener;
2929
use tracing::debug;
3030
use tree_hash::TreeHash;
3131

32-
pub async fn start_mock_relay_service<T: EthSpec>(state: Arc<MockRelayState>, port: u16) -> eyre::Result<()> {
32+
pub async fn start_mock_relay_service<T: EthSpec>(
33+
state: Arc<MockRelayState>,
34+
port: u16,
35+
) -> eyre::Result<()> {
3336
let app = mock_relay_app_router::<T>(state);
3437

3538
let socket = SocketAddr::new("0.0.0.0".parse()?, port);
@@ -136,4 +139,4 @@ async fn handle_submit_block<T: EthSpec>(State(state): State<Arc<MockRelayState>
136139
let response = SubmitBlindedBlockResponse::<T>::default();
137140
(StatusCode::OK, Json(response)).into_response()
138141
}
139-
}
142+
}

tests/src/mock_validator.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ use alloy::{
22
primitives::B256,
33
rpc::types::beacon::{relay::ValidatorRegistration, BlsPublicKey},
44
};
5-
use cb_common::pbs::{RelayClient, SignedBlindedBeaconBlock, EthSpec};
5+
use cb_common::pbs::{EthSpec, RelayClient, SignedBlindedBeaconBlock};
66
use reqwest::Response;
77

88
use crate::utils::generate_mock_relay;

tests/tests/payloads.rs

Lines changed: 22 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,31 +1,41 @@
11
use alloy::rpc::types::beacon::relay::ValidatorRegistration;
22
use cb_common::{
3-
pbs::{SignedBlindedBeaconBlock, SubmitBlindedBlockResponse},
4-
utils::test_encode_decode, pbs::{DenebSpec, ElectraSpec}
3+
pbs::{DenebSpec, ElectraSpec, SignedBlindedBeaconBlock, SubmitBlindedBlockResponse},
4+
utils::test_encode_decode,
55
};
6-
use serde_json::Value;
76
use cb_tests::utils::SpecVersion;
7+
use serde_json::Value;
88

99
// Happy path tests
1010
#[test]
1111
fn test_registrations() {
12-
test_encode_decode::<Vec<ValidatorRegistration>>(include_str!("../data/deneb/registration_holesky.json"));
12+
test_encode_decode::<Vec<ValidatorRegistration>>(include_str!(
13+
"../data/deneb/registration_holesky.json"
14+
));
1315
}
1416

1517
#[test]
1618
fn test_signed_blinded_block() {
1719
// Deneb
18-
test_encode_decode::<SignedBlindedBeaconBlock<DenebSpec>>(include_str!("../data/deneb/signed_blinded_block_holesky.json"));
20+
test_encode_decode::<SignedBlindedBeaconBlock<DenebSpec>>(include_str!(
21+
"../data/deneb/signed_blinded_block_holesky.json"
22+
));
1923
// Electra
20-
test_encode_decode::<SignedBlindedBeaconBlock<ElectraSpec>>(include_str!("../data/electra/signed_blinded_block_holesky.json"));
24+
test_encode_decode::<SignedBlindedBeaconBlock<ElectraSpec>>(include_str!(
25+
"../data/electra/signed_blinded_block_holesky.json"
26+
));
2127
}
2228

2329
#[test]
2430
fn test_submit_block_response() {
2531
// Deneb
26-
test_encode_decode::<SubmitBlindedBlockResponse<DenebSpec>>(include_str!("../data/deneb/submit_block_response_holesky.json"));
32+
test_encode_decode::<SubmitBlindedBlockResponse<DenebSpec>>(include_str!(
33+
"../data/deneb/submit_block_response_holesky.json"
34+
));
2735
// Electra
28-
test_encode_decode::<SubmitBlindedBlockResponse<ElectraSpec>>(include_str!("../data/electra/submit_block_response_holesky.json"));
36+
test_encode_decode::<SubmitBlindedBlockResponse<ElectraSpec>>(include_str!(
37+
"../data/electra/submit_block_response_holesky.json"
38+
));
2939
}
3040

3141
// Unhappy path tests
@@ -62,9 +72,9 @@ fn test_registration_missing_fields() {
6272
let error = test_missing_registration_field(version, field);
6373
assert!(
6474
error.contains(&format!("missing field `{}`", field)),
65-
"Expected error about missing {}, got: {}",
66-
field,
67-
error
75+
"Expected error about missing {}, got: {}",
76+
field,
77+
error
6878
);
6979
}
7080
}
@@ -158,4 +168,4 @@ fn test_submit_block_response_missing_fields() {
158168
);
159169
}
160170
}
161-
}
171+
}

tests/tests/pbs_get_header.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,13 @@ use std::{sync::Arc, time::Duration};
22

33
use alloy::primitives::{B256, U256};
44
use cb_common::{
5-
pbs::GetHeaderResponse,
5+
pbs::{DenebSpec, ElectraSpec, GetHeaderResponse},
66
signature::sign_builder_root,
77
signer::{random_secret, BlsPublicKey},
88
types::Chain,
99
utils::{blst_pubkey_to_alloy, timestamp_of_slot_start_sec},
1010
};
1111
use cb_pbs::{DefaultBuilderApi, PbsService, PbsState};
12-
use cb_common::pbs::{DenebSpec, ElectraSpec};
1312
use cb_tests::{
1413
mock_relay::{start_mock_relay_service, MockRelayState},
1514
mock_validator::MockValidator,

tests/tests/pbs_get_status.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
use std::{sync::Arc, time::Duration};
22

33
use cb_common::{
4+
pbs::{DenebSpec, ElectraSpec},
45
signer::{random_secret, BlsPublicKey},
56
types::Chain,
6-
pbs::{DenebSpec, ElectraSpec},
77
utils::blst_pubkey_to_alloy,
88
};
99
use cb_pbs::{DefaultBuilderApi, PbsService, PbsState};
@@ -152,4 +152,4 @@ async fn test_get_status_returns_502_if_relay_down_electra() -> Result<()> {
152152
// Expect no statuses since relay is down
153153
assert_eq!(mock_state.received_get_status(), 0);
154154
Ok(())
155-
}
155+
}

tests/tests/pbs_post_blinded_blocks.rs

Lines changed: 18 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -40,13 +40,19 @@ async fn test_submit_block_deneb() -> Result<()> {
4040

4141
let mock_validator = MockValidator::new(pbs_port)?;
4242
info!("Sending submit block");
43-
let res = mock_validator.do_submit_block::<DenebSpec>(Some(SignedBlindedBeaconBlock::default())).await?;
43+
let res = mock_validator
44+
.do_submit_block::<DenebSpec>(Some(SignedBlindedBeaconBlock::default()))
45+
.await?;
4446

4547
assert_eq!(res.status(), StatusCode::OK);
4648
assert_eq!(mock_state.received_submit_block(), 1);
4749

48-
let response_body = serde_json::from_slice::<SubmitBlindedBlockResponse<DenebSpec>>(&res.bytes().await?)?;
49-
assert_eq!(response_body.block_hash(), SubmitBlindedBlockResponse::<DenebSpec>::default().block_hash());
50+
let response_body =
51+
serde_json::from_slice::<SubmitBlindedBlockResponse<DenebSpec>>(&res.bytes().await?)?;
52+
assert_eq!(
53+
response_body.block_hash(),
54+
SubmitBlindedBlockResponse::<DenebSpec>::default().block_hash()
55+
);
5056
Ok(())
5157
}
5258

@@ -74,13 +80,19 @@ async fn test_submit_block_electra() -> Result<()> {
7480

7581
let mock_validator = MockValidator::new(pbs_port)?;
7682
info!("Sending submit block");
77-
let res = mock_validator.do_submit_block::<ElectraSpec>(Some(SignedBlindedBeaconBlock::default())).await?;
83+
let res = mock_validator
84+
.do_submit_block::<ElectraSpec>(Some(SignedBlindedBeaconBlock::default()))
85+
.await?;
7886

7987
assert_eq!(res.status(), StatusCode::OK);
8088
assert_eq!(mock_state.received_submit_block(), 1);
8189

82-
let response_body = serde_json::from_slice::<SubmitBlindedBlockResponse<ElectraSpec>>(&res.bytes().await?)?;
83-
assert_eq!(response_body.block_hash(), SubmitBlindedBlockResponse::<ElectraSpec>::default().block_hash());
90+
let response_body =
91+
serde_json::from_slice::<SubmitBlindedBlockResponse<ElectraSpec>>(&res.bytes().await?)?;
92+
assert_eq!(
93+
response_body.block_hash(),
94+
SubmitBlindedBlockResponse::<ElectraSpec>::default().block_hash()
95+
);
8496
Ok(())
8597
}
8698

@@ -114,7 +126,6 @@ async fn test_submit_block_too_large_deneb() -> Result<()> {
114126
Ok(())
115127
}
116128

117-
118129
#[tokio::test]
119130
async fn test_submit_block_too_large_electra() -> Result<()> {
120131
setup_test_env();

tests/tests/pbs_post_validators.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@ use std::{sync::Arc, time::Duration};
22

33
use alloy::rpc::types::beacon::relay::ValidatorRegistration;
44
use cb_common::{
5+
pbs::{DenebSpec, ElectraSpec},
56
signer::{random_secret, BlsPublicKey},
6-
pbs::{ElectraSpec, DenebSpec},
77
types::Chain,
88
utils::blst_pubkey_to_alloy,
99
};
@@ -109,7 +109,6 @@ async fn test_register_validators_electra() -> Result<()> {
109109
Ok(())
110110
}
111111

112-
113112
#[tokio::test]
114113
async fn test_register_validators_returns_422_if_request_is_malformed_deneb() -> Result<()> {
115114
setup_test_env();

0 commit comments

Comments
 (0)