From b082e7f8150ca10cf757c2c3ba71b2e6aeb78696 Mon Sep 17 00:00:00 2001
From: Joshy Orndorff <JoshOrndorff@users.noreply.github.com>
Date: Thu, 18 Jun 2020 03:54:47 -0400
Subject: [PATCH] Allow custom polkadot chainspec. (#122)

---
 test/parachain/src/command.rs | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/test/parachain/src/command.rs b/test/parachain/src/command.rs
index f27c0825fe..a24dd33520 100644
--- a/test/parachain/src/command.rs
+++ b/test/parachain/src/command.rs
@@ -102,11 +102,15 @@ impl SubstrateCli for PolkadotCli {
 		"cumulus-test-parachain-collator"
 	}
 
-	fn load_spec(&self, _id: &str) -> std::result::Result<Box<dyn sc_service::ChainSpec>, String> {
-		polkadot_service::PolkadotChainSpec::from_json_bytes(
-			&include_bytes!("../res/polkadot_chainspec.json")[..],
-		)
-		.map(|r| Box::new(r) as Box<_>)
+	fn load_spec(&self, id: &str) -> std::result::Result<Box<dyn sc_service::ChainSpec>, String> {
+		Ok(match id {
+			"" | "local" | "dev" => Box::new(polkadot_service::PolkadotChainSpec::from_json_bytes(
+				&include_bytes!("../res/polkadot_chainspec.json")[..],
+			)?),
+			path => Box::new(chain_spec::ChainSpec::from_json_file(
+				std::path::PathBuf::from(path),
+			)?),
+		})
 	}
 }