Skip to content

Commit 15750c9

Browse files
authored
simulators/lean: adjust gossip mock (#1474)
1 parent 6eb48f0 commit 15750c9

2 files changed

Lines changed: 13 additions & 1 deletion

File tree

simulators/lean/src/scenarios/gossip.rs

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@ use std::time::Duration;
1616
use tokio::time::sleep;
1717

1818
const GOSSIPSUB_TIMEOUT_SECS: u64 = 30;
19+
const NODE_ID_ENVIRONMENT_VARIABLE: &str = "HIVE_NODE_ID";
20+
const CLIENT_PRIVATE_KEY_ENVIRONMENT_VARIABLE: &str = "HIVE_CLIENT_PRIVATE_KEY";
1921

2022
// Suite: gossip
2123
// Tests gossipsub protocol behavior using a mock node.
@@ -104,6 +106,8 @@ async fn setup_mock_bootnode(clients: Vec<Client>) -> (MockNode, Client, String)
104106
let client = expect_single_client(clients);
105107
let client_type = client.kind.clone();
106108
let test = client.test.clone();
109+
let client_kind = crate::utils::util::lean_client_kind(&client_type)
110+
.unwrap_or_else(|err| panic!("Unable to derive client kind for gossip test: {err}"));
107111

108112
let mut mock = MockNode::new().expect("failed to create mock node");
109113
let listen_addr = mock
@@ -131,6 +135,14 @@ async fn setup_mock_bootnode(clients: Vec<Client>) -> (MockNode, Client, String)
131135

132136
let mut environment = lean_environment();
133137
environment.insert("HIVE_BOOTNODES".to_string(), mock_enr);
138+
environment.insert(
139+
NODE_ID_ENVIRONMENT_VARIABLE.to_string(),
140+
format!("{client_kind}_mock"),
141+
);
142+
environment.insert(
143+
CLIENT_PRIVATE_KEY_ENVIRONMENT_VARIABLE.to_string(),
144+
format!("{:064x}", 0xfeed_u64),
145+
);
134146
let files = prepare_client_runtime_files(&client_type, &environment)
135147
.unwrap_or_else(|e| panic!("failed to prepare client files: {e}"));
136148
let client = test

simulators/lean/src/utils/libp2p_mock.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -505,7 +505,7 @@ impl MockNode {
505505
let cfg = request_response::Config::default();
506506
let gossipsub_config = GossipsubConfig::default();
507507
let gossipsub = GossipsubBehaviour::new(
508-
MessageAuthenticity::Signed(keypair.clone()),
508+
MessageAuthenticity::Anonymous,
509509
gossipsub_config,
510510
)
511511
.expect("Failed to create gossipsub behaviour");

0 commit comments

Comments
 (0)