Use async test helper to simplify node testing (#1578)

This commit is contained in:
Cecile Tonglet
2020-08-13 19:01:37 +02:00
committed by GitHub
parent 2a96c19370
commit 3a7d9753f6
4 changed files with 58 additions and 90 deletions
@@ -14,60 +14,31 @@
// You should have received a copy of the GNU General Public License
// along with Polkadot. If not, see <http://www.gnu.org/licenses/>.
use tokio::{time::delay_for as sleep, task::spawn};
use futures::{pin_mut, select, FutureExt as _};
use polkadot_test_service::*;
use service::TaskExecutor;
use sp_keyring::Sr25519Keyring::{Alice, Bob};
use std::time::Duration;
static INTEGRATION_TEST_ALLOWED_TIME: Option<&str> = option_env!("INTEGRATION_TEST_ALLOWED_TIME");
#[substrate_test_utils::test]
async fn call_function_actually_work(task_executor: TaskExecutor) {
let alice = run_test_node(task_executor, Alice, || {}, Vec::new());
#[tokio::test]
async fn call_function_actually_work() {
let mut alice = run_test_node(
(move |fut, _| {
spawn(fut).map(|_| ())
})
.into(),
Alice,
|| {},
Vec::new(),
let function = polkadot_test_runtime::Call::Balances(pallet_balances::Call::transfer(
Default::default(),
1,
));
let output = alice.call_function(function, Bob).await.unwrap();
let res = output.result.expect("return value expected");
let json = serde_json::from_str::<serde_json::Value>(res.as_str()).expect("valid JSON");
let object = json.as_object().expect("JSON is an object");
assert!(object.contains_key("jsonrpc"), "key jsonrpc exists");
let result = object.get("result");
let result = result.expect("key result exists");
assert_eq!(
result.as_str().map(|x| x.starts_with("0x")),
Some(true),
"result starts with 0x"
);
let t1 = sleep(Duration::from_secs(
INTEGRATION_TEST_ALLOWED_TIME
.and_then(|x| x.parse().ok())
.unwrap_or(600),
))
.fuse();
let t2 = async {
let function = polkadot_test_runtime::Call::Balances(pallet_balances::Call::transfer(
Default::default(),
1,
));
let output = alice.call_function(function, Bob).await.unwrap();
let res = output.result.expect("return value expected");
let json = serde_json::from_str::<serde_json::Value>(res.as_str()).expect("valid JSON");
let object = json.as_object().expect("JSON is an object");
assert!(object.contains_key("jsonrpc"), "key jsonrpc exists");
let result = object.get("result");
let result = result.expect("key result exists");
assert_eq!(
result.as_str().map(|x| x.starts_with("0x")),
Some(true),
"result starts with 0x"
);
alice.task_manager.terminate();
}
.fuse();
pin_mut!(t1, t2);
select! {
_ = t1 => {
panic!("the test took too long, maybe no blocks have been produced");
},
_ = t2 => {},
}
alice.task_manager.clean_shutdown().await;
}