mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-04-26 19:17:58 +00:00
Liam Aharon
0b3444c51d
* new test for try-runtime tuple stuff * fix * remove development comment * formatting * remove todo comment * follow-chain working test * refactor common cli testing utils * fix comment * revert Cargo.lock changes * update Cargo.lock * improve doc comment * fix error typo * update Cargo.lock * feature gate try-runtime test * build_substrate cli test util * feature gate follow_chain tests * move fn start_node to test-utils * improve test pkg name * use tokio Child and Command * remove redundant import * fix ci * fix ci * don't leave hanging processes * improved child process cleanup * use existing KillChildOnDrop * remove redundant comment * Update test-utils/cli/src/lib.rs Co-authored-by: Koute <koute@users.noreply.github.com> --------- Co-authored-by: kianenigma <kian@parity.io> Co-authored-by: Koute <koute@users.noreply.github.com>
92 lines
2.7 KiB
Rust
92 lines
2.7 KiB
Rust
// 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/>.
|
|
|
|
use assert_cmd::cargo::cargo_bin;
|
|
use std::{process, time::Duration};
|
|
|
|
use crate::common::KillChildOnDrop;
|
|
|
|
use substrate_cli_test_utils as common;
|
|
pub mod websocket_server;
|
|
|
|
#[tokio::test]
|
|
async fn telemetry_works() {
|
|
common::run_with_timeout(Duration::from_secs(60 * 10), async move {
|
|
let config = websocket_server::Config {
|
|
capacity: 1,
|
|
max_frame_size: 1048 * 1024,
|
|
send_buffer_len: 32,
|
|
bind_address: "127.0.0.1:0".parse().unwrap(),
|
|
};
|
|
let mut server = websocket_server::WsServer::new(config).await.unwrap();
|
|
|
|
let addr = server.local_addr().unwrap();
|
|
|
|
let server_task = tokio::spawn(async move {
|
|
loop {
|
|
use websocket_server::Event;
|
|
match server.next_event().await {
|
|
// New connection on the listener.
|
|
Event::ConnectionOpen { address } => {
|
|
println!("New connection from {:?}", address);
|
|
server.accept();
|
|
},
|
|
|
|
// Received a message from a connection.
|
|
Event::BinaryFrame { message, .. } => {
|
|
let json: serde_json::Value = serde_json::from_slice(&message).unwrap();
|
|
let object =
|
|
json.as_object().unwrap().get("payload").unwrap().as_object().unwrap();
|
|
if matches!(object.get("best"), Some(serde_json::Value::String(_))) {
|
|
break
|
|
}
|
|
},
|
|
|
|
Event::TextFrame { .. } =>
|
|
panic!("Got a TextFrame over the socket, this is a bug"),
|
|
|
|
// Connection has been closed.
|
|
Event::ConnectionError { .. } => {},
|
|
}
|
|
}
|
|
});
|
|
|
|
let mut substrate = process::Command::new(cargo_bin("substrate"));
|
|
|
|
let mut substrate = KillChildOnDrop(
|
|
substrate
|
|
.args(&["--dev", "--tmp", "--telemetry-url"])
|
|
.arg(format!("ws://{} 10", addr))
|
|
.arg("--no-hardware-benchmarks")
|
|
.stdout(process::Stdio::piped())
|
|
.stderr(process::Stdio::piped())
|
|
.stdin(process::Stdio::null())
|
|
.spawn()
|
|
.unwrap(),
|
|
);
|
|
|
|
server_task.await.expect("server task panicked");
|
|
|
|
substrate.assert_still_running();
|
|
|
|
// Stop the process
|
|
substrate.stop();
|
|
})
|
|
.await;
|
|
}
|