Add test helper fn

crates/testing-utils/src/helpers.rs

@@ -0,0 +1,54 @@
+// Copyright (C) 2023 Entropy Cryptography Inc.
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero 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 Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+use crate::{
+    chain_api::{get_api, get_rpc, EntropyConfig},
+    spawn_testing_validators,
+    substrate_context::{test_context_stationary, test_node_process_testing_state},
+    ChainSpecType,
+};
+use entropy_protocol::PartyId;
+use subxt::{backend::legacy::LegacyRpcMethods, OnlineClient};
+
+/// A helper for setting up tests which starts both a set of TS servers and a chain node and returns
+/// the chain API as well as IP addresses and PartyId of the started validators
+pub async fn spawn_tss_nodes_and_start_chain(
+    add_parent_key: bool,
+    chain_spec_type: ChainSpecType,
+) -> (OnlineClient<EntropyConfig>, LegacyRpcMethods<EntropyConfig>, Vec<String>, Vec<PartyId>) {
+    let (validator_ips, validator_ids) =
+        spawn_testing_validators(add_parent_key, chain_spec_type.clone()).await;
+
+    let (api, rpc) = match chain_spec_type {
+        ChainSpecType::Development => {
+            let substrate_context = test_context_stationary().await;
+            (
+                get_api(&substrate_context.node_proc.ws_url).await.unwrap(),
+                get_rpc(&substrate_context.node_proc.ws_url).await.unwrap(),
+            )
+        },
+        ChainSpecType::Integration => {
+            // Here we need to use `--chain=integration-tests` and force authoring otherwise we won't be
+            // able to get our chain in the right state to be jump started.
+            let force_authoring = true;
+            let substrate_context = test_node_process_testing_state(force_authoring).await;
+            (
+                get_api(&substrate_context.ws_url).await.unwrap(),
+                get_rpc(&substrate_context.ws_url).await.unwrap(),
+            )
+        },
+    };
+    (api, rpc, validator_ips, validator_ids)
+}

crates/testing-utils/src/lib.rs

@@ -19,6 +19,7 @@ extern crate lazy_static;
 pub use entropy_tss::chain_api;
 pub mod constants;
 pub mod create_test_keyshares;
+pub mod helpers;
 mod node_proc;
 pub mod substrate_context;
 pub use entropy_tss::helpers::tests::{

crates/threshold-signature-server/src/helpers/tests.rs

@@ -42,7 +42,6 @@ use entropy_kvdb::{encrypted_sled::PasswordMethod, get_db_path, kv_manager::KvMa
 use entropy_protocol::PartyId;
 use entropy_shared::{DAVE_VERIFYING_KEY, EVE_VERIFYING_KEY, NETWORK_PARENT_KEY};
 use std::time::Duration;
-
 use subxt::{
     backend::legacy::LegacyRpcMethods, ext::sp_core::sr25519, tx::PairSigner,
     utils::AccountId32 as SubxtAccountId32, Config, OnlineClient,
@@ -121,7 +120,7 @@ pub async fn create_clients(
 
 /// A way to specify whether the test environment uses the Development chainspec, which has 3 TSS
 /// nodes, or the Integration test chainspec which has 4 TSS nodes
-#[derive(PartialEq)]
+#[derive(Clone, PartialEq)]
 pub enum ChainSpecType {
     Development,
     Integration,