Skip to content

Commit f82e831

Browse files
committed
fix clippy + some tests
1 parent 3b7f7fe commit f82e831

9 files changed

Lines changed: 35 additions & 20 deletions

File tree

benches/pbs/src/main.rs

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
use std::time::{Duration, Instant};
22

3-
use alloy::{hex, primitives::B256};
3+
use alloy::primitives::B256;
44
use cb_common::{
55
config::RelayConfig,
66
pbs::{BlsPublicKey, BlsSecretKey, GetHeaderResponse, RelayClient, RelayEntry},
77
types::Chain,
8+
utils::TestRandomSeed,
89
};
910
use cb_tests::mock_relay::{start_mock_relay_service, MockRelayState};
1011
use comfy_table::Table;
@@ -16,9 +17,6 @@ mod config;
1617
fn get_random_hash() -> B256 {
1718
B256::from(rand::random::<[u8; 32]>())
1819
}
19-
fn get_random_pubkey() -> BlsPublicKey {
20-
BlsPublicKey::deserialize(&hex!("0x000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000")).unwrap()
21-
}
2220

2321
#[tokio::main]
2422
async fn main() {
@@ -44,7 +42,7 @@ async fn main() {
4442
// bench
4543
for slot in 0..config.benchmark.n_slots {
4644
let parent_hash = get_random_hash();
47-
let validator_pubkey = get_random_pubkey();
45+
let validator_pubkey = BlsPublicKey::test_random();
4846
let url = mock_validator.get_header_url(slot, &parent_hash, &validator_pubkey).unwrap();
4947

5048
for _ in 0..config.benchmark.headers_per_slot {
@@ -150,7 +148,7 @@ async fn start_mock_relay(chain: Chain, relay_config: RelayConfig) {
150148
}
151149

152150
fn get_mock_validator(bench: BenchConfig) -> RelayClient {
153-
let entry = RelayEntry { id: bench.id, pubkey: get_random_pubkey(), url: bench.url };
151+
let entry = RelayEntry { id: bench.id, pubkey: BlsPublicKey::test_random(), url: bench.url };
154152
let config = RelayConfig {
155153
entry,
156154
id: None,

crates/common/src/commit/request.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ mod tests {
261261

262262
#[test]
263263
fn test_decode_response_signature() {
264-
let data = r#""0xa3ffa9241f78279f1af04644cb8c79c2d8f02bcf0e28e2f186f6dcccac0a869c2be441fda50f0dea895cfce2e53f0989a3ffa9241f78279f1af04644cb8c79c2d8f02bcf0e28e2f186f6dcccac0a869c2be441fda50f0dea895cfce2e53f0989""#;
264+
let data = r#""0xc00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000""#;
265265
let _: BlsSignature = serde_json::from_str(data).unwrap();
266266

267267
let data = r#""0x985b495f49d1b96db3bba3f6c5dd1810950317c10d4c2042bd316f338cdbe74359072e209b85e56ac492092d7860063dd096ca31b4e164ef27e3f8d508e656801c""#;
@@ -294,7 +294,7 @@ mod tests {
294294
"delegator": "0xa3366b54f28e4bf1461926a3c70cdb0ec432b5c92554ecaae3742d33fb33873990cbed1761c68020e6d3c14d30a22050",
295295
"proxy": "0xa3366b54f28e4bf1461926a3c70cdb0ec432b5c92554ecaae3742d33fb33873990cbed1761c68020e6d3c14d30a22050"
296296
},
297-
"signature": "0xa3ffa9241f78279f1af04644cb8c79c2d8f02bcf0e28e2f186f6dcccac0a869c2be441fda50f0dea895cfce2e53f0989a3ffa9241f78279f1af04644cb8c79c2d8f02bcf0e28e2f186f6dcccac0a869c2be441fda50f0dea895cfce2e53f0989"
297+
"signature": "0xc00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000"
298298
}"#;
299299

300300
let _: SignedProxyDelegationBls = serde_json::from_str(data).unwrap();
@@ -304,7 +304,7 @@ mod tests {
304304
"delegator": "0xa3366b54f28e4bf1461926a3c70cdb0ec432b5c92554ecaae3742d33fb33873990cbed1761c68020e6d3c14d30a22050",
305305
"proxy": "0x4ca9939a8311a7cab3dde201b70157285fa81a9d"
306306
},
307-
"signature": "0xa3ffa9241f78279f1af04644cb8c79c2d8f02bcf0e28e2f186f6dcccac0a869c2be441fda50f0dea895cfce2e53f0989a3ffa9241f78279f1af04644cb8c79c2d8f02bcf0e28e2f186f6dcccac0a869c2be441fda50f0dea895cfce2e53f0989"
307+
"signature": "0xc00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000"
308308
}"#;
309309

310310
let _: SignedProxyDelegationEcdsa = serde_json::from_str(data).unwrap();

crates/common/src/config/utils.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ fn decode_string_to_map(raw: &str) -> Result<HashMap<ModuleId, String>> {
6161
#[cfg(test)]
6262
mod tests {
6363
use super::*;
64+
use crate::utils::TestRandomSeed;
6465

6566
#[test]
6667
fn test_decode_string_to_map() {
@@ -74,8 +75,8 @@ mod tests {
7475

7576
#[test]
7677
fn test_remove_duplicate_keys() {
77-
let key1 = BlsPublicKey::deserialize(&[1; 48]).unwrap();
78-
let key2 = BlsPublicKey::deserialize(&[2; 48]).unwrap();
78+
let key1 = BlsPublicKey::test_random();
79+
let key2 = BlsPublicKey::test_random();
7980
let keys = vec![key1.clone(), key2.clone(), key1.clone()];
8081

8182
let unique_keys = remove_duplicate_keys(keys);

crates/common/src/pbs/error.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ pub enum ValidationError {
5555
EmptyBlockhash,
5656

5757
#[error("pubkey mismatch: expected {expected} got {got}")]
58-
PubkeyMismatch { expected: BlsPublicKey, got: BlsPublicKey },
58+
PubkeyMismatch { expected: Box<BlsPublicKey>, got: Box<BlsPublicKey> },
5959

6060
#[error("parent hash mismatch: expected {expected} got {got}")]
6161
ParentHashMismatch { expected: B256, got: B256 },

crates/common/src/signer/schemes/ecdsa.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ impl EcdsaSigner {
9191
match self {
9292
EcdsaSigner::Local(sk) => {
9393
let domain = compute_domain(chain, COMMIT_BOOST_DOMAIN);
94-
let signing_root = compute_signing_root(object_root, domain).into();
94+
let signing_root = compute_signing_root(object_root, domain);
9595
sk.sign_hash_sync(&signing_root).map(EcdsaSignature::from)
9696
}
9797
}

crates/common/src/utils.rs

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ use std::{
55

66
use alloy::primitives::U256;
77
use axum::http::HeaderValue;
8+
use lh_types::test_utils::{SeedableRng, TestRandom, XorShiftRng};
89
use rand::{distr::Alphanumeric, Rng};
910
use reqwest::header::HeaderMap;
1011
use serde::{de::DeserializeOwned, Serialize};
@@ -347,6 +348,18 @@ pub async fn wait_for_signal() -> eyre::Result<()> {
347348
Ok(())
348349
}
349350

351+
pub trait TestRandomSeed: TestRandom {
352+
fn test_random() -> Self
353+
where
354+
Self: Sized,
355+
{
356+
let mut rng = XorShiftRng::from_seed([42; 16]);
357+
Self::random_for_test(&mut rng)
358+
}
359+
}
360+
361+
impl<T: TestRandom> TestRandomSeed for T {}
362+
350363
#[cfg(test)]
351364
mod test {
352365
use super::{create_jwt, decode_jwt, validate_jwt};

crates/pbs/src/mev_boost/get_header.rs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -460,14 +460,14 @@ fn validate_signature<T: TreeHash>(
460460
) -> Result<(), ValidationError> {
461461
if expected_relay_pubkey != received_relay_pubkey {
462462
return Err(ValidationError::PubkeyMismatch {
463-
expected: expected_relay_pubkey.clone(),
464-
got: received_relay_pubkey.clone(),
463+
expected: Box::new(expected_relay_pubkey.clone()),
464+
got: Box::new(received_relay_pubkey.clone()),
465465
});
466466
}
467467

468468
if !verify_signed_message(
469469
chain,
470-
&received_relay_pubkey,
470+
received_relay_pubkey,
471471
&message,
472472
signature,
473473
APPLICATION_BUILDER_DOMAIN,
@@ -595,7 +595,10 @@ mod tests {
595595

596596
assert_eq!(
597597
validate_signature(Chain::Holesky, &wrong_pubkey, &pubkey, &message, &wrong_signature),
598-
Err(ValidationError::PubkeyMismatch { expected: wrong_pubkey, got: pubkey.clone() })
598+
Err(ValidationError::PubkeyMismatch {
599+
expected: Box::new(wrong_pubkey),
600+
got: Box::new(pubkey.clone())
601+
})
599602
);
600603

601604
assert!(matches!(

examples/da_commit/src/main.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ impl DaCommitService {
6969
let mut data = 0;
7070

7171
loop {
72-
self.send_request(data, pubkey.clone(), proxy_bls.clone(), proxy_ecdsa.clone()).await?;
72+
self.send_request(data, pubkey.clone(), proxy_bls.clone(), proxy_ecdsa).await?;
7373
sleep(Duration::from_secs(self.config.extra.sleep_secs)).await;
7474
data += 1;
7575
}

tests/src/mock_validator.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,13 +46,13 @@ impl MockValidator {
4646
let url = self.comm_boost.submit_block_url().unwrap();
4747

4848
let signed_blinded_block =
49-
signed_blinded_block.unwrap_or_else(|| load_test_signed_blinded_block());
49+
signed_blinded_block.unwrap_or_else(load_test_signed_blinded_block);
5050

5151
Ok(self.comm_boost.client.post(url).json(&signed_blinded_block).send().await?)
5252
}
5353
}
5454

5555
pub fn load_test_signed_blinded_block() -> SignedBlindedBeaconBlock {
5656
let data_json = include_str!("signed-blinded-beacon-block-electra-2.json");
57-
serde_json::from_str(&data_json).unwrap()
57+
serde_json::from_str(data_json).unwrap()
5858
}

0 commit comments

Comments
 (0)