test_parachain/
main.rs

1// Copyright (C) Parity Technologies (UK) Ltd.
2// This file is part of Cumulus.
3
4// Cumulus is free software: you can redistribute it and/or modify
5// it under the terms of the GNU General Public License as published by
6// the Free Software Foundation, either version 3 of the License, or
7// (at your option) any later version.
8
9// Cumulus is distributed in the hope that it will be useful,
10// but WITHOUT ANY WARRANTY; without even the implied warranty of
11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12// GNU General Public License for more details.
13
14// You should have received a copy of the GNU General Public License
15// along with Cumulus.  If not, see <http://www.gnu.org/licenses/>.
16
17mod cli;
18
19use std::sync::Arc;
20
21use cli::{AuthoringPolicy, RelayChainCli, Subcommand, TestCollatorCli};
22use cumulus_primitives_core::relay_chain::CollatorPair;
23use cumulus_test_service::{new_partial, AnnounceBlockFn};
24use sc_cli::{CliConfiguration, SubstrateCli};
25use sp_core::Pair;
26
27pub fn wrap_announce_block() -> Box<dyn FnOnce(AnnounceBlockFn) -> AnnounceBlockFn> {
28	tracing::info!("Block announcements disabled.");
29	Box::new(|_| {
30		// Never announce any block
31		Arc::new(|_, _| {})
32	})
33}
34
35fn main() -> Result<(), sc_cli::Error> {
36	let cli = TestCollatorCli::from_args();
37
38	match &cli.subcommand {
39		#[allow(deprecated)]
40		Some(Subcommand::BuildSpec(cmd)) => {
41			let runner = cli.create_runner(cmd)?;
42			runner.sync_run(|config| cmd.run(config.chain_spec, config.network))
43		},
44
45		Some(Subcommand::ExportChainSpec(cmd)) => {
46			let chain_spec = cli.load_spec(&cmd.chain)?;
47			cmd.run(chain_spec)
48		},
49
50		Some(Subcommand::ExportGenesisHead(cmd)) => {
51			let runner = cli.create_runner(cmd)?;
52			runner.sync_run(|mut config| {
53				let partial = new_partial(&mut config, false)?;
54				cmd.run(partial.client)
55			})
56		},
57		Some(Subcommand::ExportGenesisWasm(cmd)) => {
58			let runner = cli.create_runner(cmd)?;
59			runner.sync_run(|config| cmd.run(&*config.chain_spec))
60		},
61		None => {
62			let log_filters = cli.run.normalize().log_filters();
63			let mut builder = sc_cli::LoggerBuilder::new(log_filters.unwrap_or_default());
64			builder.with_colors(false);
65			let _ = builder.init();
66
67			let collator_options = cli.run.collator_options();
68			let tokio_runtime = sc_cli::build_runtime()?;
69			let tokio_handle = tokio_runtime.handle();
70			let parachain_config = cli
71				.run
72				.normalize()
73				.create_configuration(&cli, tokio_handle.clone())
74				.expect("Should be able to generate config");
75
76			let relay_chain_cli = RelayChainCli::new(
77				&parachain_config,
78				[RelayChainCli::executable_name()].iter().chain(cli.relaychain_args.iter()),
79			);
80			let tokio_handle = parachain_config.tokio_handle.clone();
81			let mut relay_chain_config = SubstrateCli::create_configuration(
82				&relay_chain_cli,
83				&relay_chain_cli,
84				tokio_handle,
85			)
86			.map_err(|err| format!("Relay chain argument error: {}", err))?;
87
88			// The relay chain side of a collator doesn't listen on WebRTC by default;
89			// `--force-enable-webrtc` or explicit `--listen-addr` take precedence.
90			let relay_network_params = &relay_chain_cli.base.base.network_params;
91			if parachain_config.role.is_authority() &&
92				!relay_network_params.force_enable_webrtc &&
93				relay_network_params.listen_addr.is_empty()
94			{
95				relay_chain_config.network.remove_webrtc_addresses();
96			}
97
98			tracing::info!(
99				"Is collating: {}",
100				if parachain_config.role.is_authority() { "yes" } else { "no" }
101			);
102			if cli.fail_pov_recovery {
103				tracing::info!("PoV recovery failure enabled");
104			}
105
106			let collator_key =
107				parachain_config.role.is_authority().then(|| CollatorPair::generate().0);
108
109			let use_slot_based_collator = cli.authoring == AuthoringPolicy::SlotBased;
110			let collator_reserved_slots = cli.collator_reserved_slots;
111			let (mut task_manager, _, _, _, _, _) = tokio_runtime
112				.block_on(async move {
113					match relay_chain_config.network.network_backend {
114						sc_network::config::NetworkBackendType::Libp2p => {
115							cumulus_test_service::start_node_impl::<
116								_,
117								sc_network::NetworkWorker<_, _>,
118							>(
119								parachain_config,
120								collator_key,
121								relay_chain_config,
122								cli.disable_block_announcements.then(wrap_announce_block),
123								cli.fail_pov_recovery,
124								|_| Ok(jsonrpsee::RpcModule::new(())),
125								collator_options,
126								true,
127								use_slot_based_collator,
128								collator_reserved_slots,
129							)
130							.await
131						},
132						sc_network::config::NetworkBackendType::Litep2p => {
133							cumulus_test_service::start_node_impl::<
134								_,
135								sc_network::Litep2pNetworkBackend,
136							>(
137								parachain_config,
138								collator_key,
139								relay_chain_config,
140								cli.disable_block_announcements.then(wrap_announce_block),
141								cli.fail_pov_recovery,
142								|_| Ok(jsonrpsee::RpcModule::new(())),
143								collator_options,
144								true,
145								use_slot_based_collator,
146								collator_reserved_slots,
147							)
148							.await
149						},
150					}
151				})
152				.expect("could not create Cumulus test service");
153
154			tokio_runtime
155				.block_on(task_manager.future())
156				.expect("Could not run service to completion");
157			Ok(())
158		},
159	}
160}
test_parachain/main.rs

test_parachain/
main.rs
