diff --git a/beacon_node/tests/test.rs b/beacon_node/tests/test.rs index d1e66674cc..71edef8469 100644 --- a/beacon_node/tests/test.rs +++ b/beacon_node/tests/test.rs @@ -1,12 +1,25 @@ #![cfg(test)] -use node_test_rig::{environment::EnvironmentBuilder, testing_client_config, LocalBeaconNode}; -use types::{MinimalEthSpec, Slot}; +use node_test_rig::{ + environment::{Environment, EnvironmentBuilder}, + testing_client_config, LocalBeaconNode, +}; +use types::{EthSpec, MinimalEthSpec, Slot}; fn env_builder() -> EnvironmentBuilder { EnvironmentBuilder::minimal() } +fn build_node(env: &mut Environment) -> LocalBeaconNode { + let context = env.core_context(); + env.runtime() + .block_on(LocalBeaconNode::production( + context, + testing_client_config(), + )) + .expect("should block until node created") +} + #[test] fn http_server_genesis_state() { let mut env = env_builder() @@ -17,7 +30,7 @@ fn http_server_genesis_state() { .build() .expect("environment should build"); - let node = LocalBeaconNode::production(env.core_context(), testing_client_config()); + let node = build_node(&mut env); let remote_node = node.remote_node().expect("should produce remote node"); let (api_state, _root) = env