Skip to content

Commit 4168d1b

Browse files
committed
Update how to create beacon processor channels according to the latest update
1 parent 2e26f77 commit 4168d1b

File tree

1 file changed

+7
-14
lines changed

1 file changed

+7
-14
lines changed

beacon_node/network/src/service/tests.rs

Lines changed: 7 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,20 @@
1-
#[cfg(not(debug_assertions))]
1+
// #[cfg(not(debug_assertions))]
22
#[cfg(test)]
33
mod tests {
44
use crate::persisted_dht::load_dht;
55
use crate::{NetworkConfig, NetworkService};
66
use beacon_chain::test_utils::BeaconChainHarness;
7-
use beacon_processor::{
8-
BeaconProcessorChannels, BeaconProcessorSend, MAX_SCHEDULED_WORK_QUEUE_LEN,
9-
MAX_WORK_EVENT_QUEUE_LEN,
10-
};
7+
use beacon_processor::{BeaconProcessorChannels, BeaconProcessorConfig};
118
use futures::StreamExt;
129
use lighthouse_network::types::{GossipEncoding, GossipKind};
1310
use lighthouse_network::{Enr, GossipTopic};
1411
use slog::{o, Drain, Level, Logger};
1512
use sloggers::{null::NullLoggerBuilder, Build};
1613
use std::str::FromStr;
1714
use std::sync::Arc;
18-
use tokio::{runtime::Runtime, sync::mpsc};
15+
use tokio::runtime::Runtime;
1916
use types::{Epoch, EthSpec, ForkName, MinimalEthSpec, SubnetId};
2017

21-
2218
fn get_logger(actual_log: bool) -> Logger {
2319
if actual_log {
2420
let drain = {
@@ -147,18 +143,15 @@ mod tests {
147143
config.discv5_config.table_filter = |_| true; // Do not ignore local IPs
148144
config.upnp_enabled = false;
149145

150-
let (beacon_processor_send, _beacon_processor_receive) =
151-
mpsc::channel(MAX_WORK_EVENT_QUEUE_LEN);
152-
let (beacon_processor_reprocess_tx, _beacon_processor_reprocess_rx) =
153-
mpsc::channel(MAX_SCHEDULED_WORK_QUEUE_LEN);
154-
146+
let beacon_processor_channels =
147+
BeaconProcessorChannels::new(&BeaconProcessorConfig::default());
155148
NetworkService::build(
156149
beacon_chain.clone(),
157150
&config,
158151
executor.clone(),
159152
None,
160-
BeaconProcessorSend(beacon_processor_send),
161-
beacon_processor_reprocess_tx,
153+
beacon_processor_channels.beacon_processor_tx,
154+
beacon_processor_channels.work_reprocessing_tx,
162155
)
163156
.await
164157
.unwrap()

0 commit comments

Comments
 (0)