Repository: paritytech/substrate
File: utils/frame/try-runtime/cli/tests/create_snapshot.rs (Rust)

Code before:
// This file is part of Substrate.

// Copyright (C) Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

#![cfg(unix)]
#[cfg(feature = "try-runtime")]
mod tests {
	use assert_cmd::cargo::cargo_bin;
	use node_primitives::Hash;
	use regex::Regex;
	use std::{
		path::{Path, PathBuf},
		process,
		time::Duration,
	};
	use substrate_cli_test_utils as common;
	use tokio::process::{Child, Command};

	#[tokio::test]
	async fn create_snapshot_works() {
		// Build substrate so binaries used in the test use the latest code.
		common::build_substrate(&["--features=try-runtime"]);

		let temp_dir = tempfile::Builder::new()
			.prefix("try-runtime-cli-test-dir")
			.tempdir()
			.expect("Failed to create a tempdir");
		let snap_file_path = temp_dir.path().join("snapshot.snap");

		common::run_with_timeout(Duration::from_secs(60), async move {
			fn create_snapshot(ws_url: &str, snap_file: &PathBuf, at: Hash) -> Child {
				Command::new(cargo_bin("substrate"))

Review comment (suggestion):
```suggestion
#![cfg(feature = "try-runtime")]
```

Code after:
// This file is part of Substrate.

// Copyright (C) Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

#![cfg(unix)]
#![cfg(feature = "try-runtime")]

use assert_cmd::cargo::cargo_bin;
use node_primitives::Hash;
use regex::Regex;
use remote_externalities::{Builder, Mode, OfflineConfig, SnapshotConfig};
use sp_runtime::testing::{Block as RawBlock, ExtrinsicWrapper};
use std::{
	path::{Path, PathBuf},
	process,
	time::Duration,
};
use substrate_cli_test_utils as common;
use tokio::process::{Child, Command};

type Block = RawBlock<ExtrinsicWrapper<Hash>>;

#[tokio::test]
async fn create_snapshot_works() {
	// Build substrate so binaries used in the test use the latest code.
	common::build_substrate(&["--features=try-runtime"]);

	let temp_dir = tempfile::Builder::new()
		.prefix("try-runtime-cli-test-dir")
		.tempdir()
		.expect("Failed to create a tempdir");
	let snap_file_path = temp_dir.path().join("snapshot.snap");