Add test environment with four TS servers

crates/testing-utils/src/constants.rs

@@ -32,6 +32,8 @@ lazy_static! {
         hex!["2cbc68e8bf0fbc1c28c282d1263fc9d29267dc12a1044fb730e8b65abc37524c"].into(),
         // Charlie - from DEFAULT_CHARLIE_MNEMONIC in entropy_tss::helpers::launch
         hex!["946140d3d5ddb980c74ffa1bb64353b5523d2d77cdf3dc617fd63de9d3b66338"].into(),
+        // Dave - from DEFAULT_DAVE_MNEMONIC in entropy_tss::helpers::launch
+        hex!["0a9054ef6b6b8ad0dd2c89895b2515583f2fbf1edced68e7328ae456d86b9402"].into(),
     ];
 
     // See entropy_tss::helpers::validator::get_signer_and_x25519_secret for how these are derived
@@ -57,6 +59,13 @@ lazy_static! {
         ]
         .try_into()
         .unwrap(),
+        // Dave - from DEFAULT_DAVE_MNEMONIC in entropy_tss::helpers::launch
+        vec![
+            165, 202, 97, 104, 222, 190, 168, 183, 231, 63, 209, 233, 19, 185, 187, 200, 10, 29, 102,
+            240, 39, 50, 140, 15, 124, 112, 94, 121, 44, 182, 40, 71
+        ]
+        .try_into()
+        .unwrap()
     ];
 }
 

crates/testing-utils/src/lib.rs

@@ -22,7 +22,7 @@ pub mod create_test_keyshares;
 mod node_proc;
 pub mod substrate_context;
 pub use entropy_tss::helpers::tests::{
-    jump_start_network_with_signer as jump_start_network, spawn_testing_validators,
+    jump_start_network_with_signer as jump_start_network, spawn_testing_validators, ChainSpecType,
 };
 pub use node_proc::TestNodeProcess;
 pub use substrate_context::*;

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

@@ -16,7 +16,7 @@ use crate::{
     chain_api::{entropy, get_api, get_rpc},
     helpers::{
         substrate::query_chain,
-        tests::{initialize_test_logger, run_to_block, spawn_testing_validators},
+        tests::{initialize_test_logger, run_to_block, spawn_testing_validators, ChainSpecType},
     },
 };
 use entropy_kvdb::clean_tests;
@@ -32,7 +32,9 @@ async fn test_attest() {
     clean_tests();
 
     let cxt = test_node_process_stationary().await;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(false).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(false, ChainSpecType::Integration).await;
+
     let api = get_api(&cxt.ws_url).await.unwrap();
     let rpc = get_rpc(&cxt.ws_url).await.unwrap();
 

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

@@ -119,10 +119,28 @@ pub async fn create_clients(
     (app, kv_store)
 }
 
-/// Spawn 3 TSS nodes with pre-stored keyshares
-pub async fn spawn_testing_validators(add_parent_key: bool) -> (Vec<String>, Vec<PartyId>) {
+/// 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)]
+pub enum ChainSpecType {
+    Development,
+    Integration,
+}
+
+/// Spawn either 3 or 4 TSS nodes depending on chain configuration, adding pre-stored keyshares if
+/// desired
+pub async fn spawn_testing_validators(
+    add_parent_key: bool,
+    chain_spec_type: ChainSpecType,
+) -> (Vec<String>, Vec<PartyId>) {
+    let add_fourth_server = chain_spec_type == ChainSpecType::Integration;
+
     // spawn threshold servers
-    let ports = [3001i64, 3002, 3003];
+    let mut ports = vec![3001i64, 3002, 3003];
+
+    if add_fourth_server {
+        ports.push(3004);
+    }
 
     let (alice_axum, alice_kv) =
         create_clients("validator1".to_string(), vec![], vec![], &Some(ValidatorName::Alice)).await;
@@ -143,11 +161,12 @@ pub async fn spawn_testing_validators(add_parent_key: bool) -> (Vec<String>, Vec
         *get_signer(&charlie_kv).await.unwrap().account_id().clone().as_ref(),
     ));
 
-    let ids = vec![alice_id, bob_id, charlie_id];
+    let mut ids = vec![alice_id, bob_id, charlie_id];
 
     put_keyshares_in_db("alice", alice_kv, add_parent_key).await;
     put_keyshares_in_db("bob", bob_kv, add_parent_key).await;
     put_keyshares_in_db("charlie", charlie_kv, add_parent_key).await;
+    // Don't give dave keyshares as dave is not initially in the signing committee
 
     let listener_alice = tokio::net::TcpListener::bind(format!("0.0.0.0:{}", ports[0]))
         .await
@@ -170,6 +189,23 @@ pub async fn spawn_testing_validators(add_parent_key: bool) -> (Vec<String>, Vec
         axum::serve(listener_charlie, charlie_axum).await.unwrap();
     });
 
+    if add_fourth_server {
+        let (dave_axum, dave_kv) =
+            create_clients("validator4".to_string(), vec![], vec![], &Some(ValidatorName::Dave))
+                .await;
+
+        let listener_dave = tokio::net::TcpListener::bind(format!("0.0.0.0:{}", ports[3]))
+            .await
+            .expect("Unable to bind to given server address.");
+        tokio::spawn(async move {
+            axum::serve(listener_dave, dave_axum).await.unwrap();
+        });
+        let dave_id = PartyId::new(SubxtAccountId32(
+            *get_signer(&dave_kv).await.unwrap().account_id().clone().as_ref(),
+        ));
+        ids.push(dave_id);
+    }
+
     tokio::time::sleep(Duration::from_secs(1)).await;
 
     let ips = ports.iter().map(|port| format!("127.0.0.1:{port}")).collect();

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

@@ -20,7 +20,7 @@ use crate::{
         launch::LATEST_BLOCK_NUMBER_PROACTIVE_REFRESH,
         tests::{
             initialize_test_logger, run_to_block, setup_client, spawn_testing_validators,
-            unsafe_get,
+            unsafe_get, ChainSpecType,
         },
     },
 };
@@ -45,7 +45,8 @@ async fn test_proactive_refresh() {
     clean_tests();
     let _cxt = test_node_process_testing_state(false).await;
 
-    let (validator_ips, _ids) = spawn_testing_validators(false).await;
+    let (validator_ips, _ids) = spawn_testing_validators(false, ChainSpecType::Integration).await;
+    let signing_committee_ips = &validator_ips[..3].to_vec();
 
     let client = reqwest::Client::new();
 
@@ -78,14 +79,14 @@ async fn test_proactive_refresh() {
     };
 
     let test_fail_incorrect_data =
-        submit_transaction_requests(validator_ips.clone(), ocw_message.clone()).await;
+        submit_transaction_requests(signing_committee_ips.clone(), ocw_message.clone()).await;
 
     for res in test_fail_incorrect_data {
         assert_eq!(res.unwrap().text().await.unwrap(), "Proactive Refresh data incorrect");
     }
     ocw_message.validators_info[0].x25519_public_key = X25519_PUBLIC_KEYS[0];
     let test_user_res =
-        submit_transaction_requests(validator_ips.clone(), ocw_message.clone()).await;
+        submit_transaction_requests(signing_committee_ips.clone(), ocw_message.clone()).await;
 
     for res in test_user_res {
         assert_eq!(res.unwrap().text().await.unwrap(), "");
@@ -104,7 +105,7 @@ async fn test_proactive_refresh() {
     ocw_message.validators_info[2].tss_account = alice.public().encode();
 
     let test_user_res_not_in_group =
-        submit_transaction_requests(validator_ips.clone(), ocw_message.clone()).await;
+        submit_transaction_requests(signing_committee_ips.clone(), ocw_message.clone()).await;
     for res in test_user_res_not_in_group {
         assert_eq!(
             res.unwrap().text().await.unwrap(),

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

@@ -122,7 +122,7 @@ use crate::{
         substrate::{get_oracle_data, query_chain, submit_transaction},
         tests::{
             create_clients, initialize_test_logger, jump_start_network_with_signer, remove_program,
-            run_to_block, setup_client, spawn_testing_validators, unsafe_get,
+            run_to_block, setup_client, spawn_testing_validators, unsafe_get, ChainSpecType,
         },
         user::compute_hash,
         validator::get_signer_and_x25519_secret_from_mnemonic,
@@ -170,7 +170,8 @@ async fn test_signature_requests_fail_on_different_conditions() {
     let two = AccountKeyring::Two;
 
     let add_parent_key_to_kvdb = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key_to_kvdb).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key_to_kvdb, ChainSpecType::Integration).await;
 
     // 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.
@@ -346,7 +347,8 @@ async fn signature_request_with_derived_account_works() {
     let charlie = AccountKeyring::Charlie;
 
     let add_parent_key_to_kvdb = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key_to_kvdb).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key_to_kvdb, ChainSpecType::Integration).await;
 
     // 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.
@@ -419,7 +421,8 @@ async fn test_signing_fails_if_wrong_participants_are_used() {
 
     let one = AccountKeyring::Dave;
 
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(false).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(false, ChainSpecType::Development).await;
     let substrate_context = test_context_stationary().await;
     let entropy_api = get_api(&substrate_context.node_proc.ws_url).await.unwrap();
     let rpc = get_rpc(&substrate_context.node_proc.ws_url).await.unwrap();
@@ -489,7 +492,8 @@ async fn test_request_limit_are_updated_during_signing() {
     let two = AccountKeyring::Two;
 
     let add_parent_key = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key, ChainSpecType::Integration).await;
 
     let force_authoring = true;
     let substrate_context = test_node_process_testing_state(force_authoring).await;
@@ -615,7 +619,8 @@ async fn test_fails_to_sign_if_non_signing_group_participants_are_used() {
     let two = AccountKeyring::Two;
 
     let add_parent_key = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key, ChainSpecType::Integration).await;
 
     let force_authoring = true;
     let substrate_context = test_node_process_testing_state(force_authoring).await;
@@ -730,7 +735,8 @@ async fn test_program_with_config() {
     let two = AccountKeyring::Two;
 
     let add_parent_key = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key, ChainSpecType::Integration).await;
 
     let force_authoring = true;
     let substrate_context = test_node_process_testing_state(force_authoring).await;
@@ -820,7 +826,8 @@ async fn test_jumpstart_network() {
     let alice = AccountKeyring::Alice;
 
     let cxt = test_context_stationary().await;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(false).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(false, ChainSpecType::Development).await;
     let api = get_api(&cxt.node_proc.ws_url).await.unwrap();
     let rpc = get_rpc(&cxt.node_proc.ws_url).await.unwrap();
 
@@ -1047,7 +1054,8 @@ async fn test_fail_infinite_program() {
     let two = AccountKeyring::Two;
 
     let add_parent_key = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key, ChainSpecType::Integration).await;
 
     let force_authoring = true;
     let substrate_context = test_node_process_testing_state(force_authoring).await;
@@ -1133,7 +1141,8 @@ async fn test_device_key_proxy() {
     let two = AccountKeyring::Two;
 
     let add_parent_key_to_kvdb = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key_to_kvdb).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key_to_kvdb, ChainSpecType::Integration).await;
 
     // 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.
@@ -1275,7 +1284,8 @@ async fn test_faucet() {
     let two = AccountKeyring::Eve;
     let alice = AccountKeyring::Alice;
 
-    let (validator_ips, _validator_ids) = spawn_testing_validators(false).await;
+    let (validator_ips, _validator_ids) =
+        spawn_testing_validators(false, ChainSpecType::Development).await;
     let substrate_context = test_node_process_testing_state(true).await;
     let entropy_api = get_api(&substrate_context.ws_url).await.unwrap();
     let rpc = get_rpc(&substrate_context.ws_url).await.unwrap();
@@ -1447,7 +1457,8 @@ async fn test_new_registration_flow() {
     let charlie = AccountKeyring::Charlie;
 
     let add_parent_key_to_kvdb = true;
-    let (_validator_ips, _validator_ids) = spawn_testing_validators(add_parent_key_to_kvdb).await;
+    let (_validator_ips, _validator_ids) =
+        spawn_testing_validators(add_parent_key_to_kvdb, ChainSpecType::Integration).await;
 
     // 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.