with externalities

paritytech · Apr 6, 2020 · f7bf658 · f7bf658
1 parent b041046
commit f7bf658
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 16 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/validation/Cargo.toml b/validation/Cargo.toml
@@ -34,6 +34,7 @@ bitvec = { version = "0.17.4", default-features = false, features = ["alloc"] }
 runtime_babe = { package = "pallet-babe", git = "https://github.com/paritytech/substrate", branch = "master" }
 babe-primitives = { package = "sp-consensus-babe", git = "https://github.com/paritytech/substrate", branch = "master" }
 keystore = { package = "sc-keystore", git = "https://github.com/paritytech/substrate", branch = "master" }
+sp-io = { git = "https://github.com/paritytech/substrate", branch = "master" }
 
 [dev-dependencies]
 sp-keyring = { git = "https://github.com/paritytech/substrate", branch = "master" }
diff --git a/validation/src/lib.rs b/validation/src/lib.rs
@@ -216,21 +216,23 @@ mod tests {
 
 	#[test]
 	fn sign_and_check_statement() {
-		let statement: Statement = GenericStatement::Valid([1; 32].into());
-		let parent_hash = [2; 32].into();
-		let signing_context = SigningContext {
-			session_index: Default::default(),
-			parent_hash,
-		};
-
-		let sig = sign_table_statement(&statement, &Sr25519Keyring::Alice.pair().into(), &signing_context);
-
-		let wrong_signing_context = SigningContext {
-			session_index: Default::default(),
-			parent_hash: [0xff; 32].into(),
-		};
-		assert!(check_statement(&statement, &sig, Sr25519Keyring::Alice.public().into(), &signing_context));
-		assert!(!check_statement(&statement, &sig, Sr25519Keyring::Alice.public().into(), &wrong_signing_context));
-		assert!(!check_statement(&statement, &sig, Sr25519Keyring::Bob.public().into(), &signing_context));
+		sp_io::TestExternalities::new_empty().execute_with(|| {
+			let statement: Statement = GenericStatement::Valid([1; 32].into());
+			let parent_hash = [2; 32].into();
+			let signing_context = SigningContext {
+				session_index: Default::default(),
+				parent_hash,
+			};
+
+			let sig = sign_table_statement(&statement, &Sr25519Keyring::Alice.pair().into(), &signing_context);
+
+			let wrong_signing_context = SigningContext {
+				session_index: Default::default(),
+				parent_hash: [0xff; 32].into(),
+			};
+			assert!(check_statement(&statement, &sig, Sr25519Keyring::Alice.public().into(), &signing_context));
+			assert!(!check_statement(&statement, &sig, Sr25519Keyring::Alice.public().into(), &wrong_signing_context));
+			assert!(!check_statement(&statement, &sig, Sr25519Keyring::Bob.public().into(), &signing_context));
+		});
 	}
 }