Test JDC fallback feature

Add a test to validate the behavior of JDC upon receiving a
SubmitShareError message from the connected Pool. JDC is expected to
switch to a different pool/jds pair if specfied in the config, otherwise
it should do solo mining.

+ Remove jdc-upstream-change MG test

Author: jbesraa

roles/tests-integration/tests/jdc_fallback.rs

@@ -0,0 +1,71 @@
+use const_sv2::{MESSAGE_TYPE_SETUP_CONNECTION, MESSAGE_TYPE_SUBMIT_SHARES_SUCCESS};
+use integration_tests_sv2::*;
+use roles_logic_sv2::{
+    mining_sv2::SubmitSharesError,
+    parsers::{AnyMessage, CommonMessages, Mining},
+};
+use sniffer::{MessageDirection, ReplaceMessage};
+use std::convert::TryInto;
+
+// Tests whether JDC will switch to a new pool after receiving a `SubmitSharesError` message from
+// the currently connected pool.
+#[tokio::test]
+async fn test_jdc_pool_fallback_after_submit_rejection() {
+    start_tracing();
+    let (tp, tp_addr) = start_template_provider(None);
+    let (_pool_1, pool_addr_1) = start_pool(Some(tp_addr)).await;
+    // Sniffer between JDC and first pool
+    let (sniffer_1, sniffer_addr_1) = start_sniffer(
+        "0".to_string(),
+        pool_addr_1,
+        false,
+        Some(
+            // Should trigger Fallback process in JDC
+            ReplaceMessage::new(
+                MessageDirection::ToDownstream,
+                MESSAGE_TYPE_SUBMIT_SHARES_SUCCESS,
+                AnyMessage::Mining(Mining::SubmitSharesError(SubmitSharesError {
+                    channel_id: 0,
+                    sequence_number: 0,
+                    error_code: "invalid-nonce".to_string().into_bytes().try_into().unwrap(),
+                })),
+            )
+            .into(),
+        ),
+    )
+    .await;
+    let (_pool_2, pool_addr_2) = start_pool(Some(tp_addr)).await;
+    // Sniffer between JDC and second pool
+    let (sniffer_2, sniffer_addr_2) =
+        start_sniffer("1".to_string(), pool_addr_2, false, None).await;
+    let (_jds_1, jds_addr_1) = start_jds(tp.rpc_info()).await;
+    // Sniffer between JDC and first JDS
+    let (sniffer_3, sniffer_addr_3) = start_sniffer("2".to_string(), jds_addr_1, false, None).await;
+    let (_jds_2, jds_addr_2) = start_jds(tp.rpc_info()).await;
+    // Sniffer between JDC and second JDS
+    let (sniffer_4, sniffer_addr_4) = start_sniffer("3".to_string(), jds_addr_2, false, None).await;
+    let (_jdc, jdc_addr) = start_jdc(
+        &[
+            (sniffer_addr_1, sniffer_addr_3),
+            (sniffer_addr_2, sniffer_addr_4),
+        ],
+        tp_addr,
+    )
+    .await;
+    // Assert that JDC has connected to the first (Pool,JDS) pair
+    sniffer_1
+        .wait_for_message_type(MessageDirection::ToUpstream, MESSAGE_TYPE_SETUP_CONNECTION)
+        .await;
+    sniffer_3
+        .wait_for_message_type(MessageDirection::ToUpstream, MESSAGE_TYPE_SETUP_CONNECTION)
+        .await;
+    let (_translator, sv2_translator_addr) = start_sv2_translator(jdc_addr).await;
+    let _ = start_mining_device_sv1(sv2_translator_addr, true, None).await;
+    // Assert that JDC switched to the second (Pool,JDS) pair
+    sniffer_2
+        .wait_for_message_type(MessageDirection::ToUpstream, MESSAGE_TYPE_SETUP_CONNECTION)
+        .await;
+    sniffer_4
+        .wait_for_message_type(MessageDirection::ToUpstream, MESSAGE_TYPE_SETUP_CONNECTION)
+        .await;
+}

scripts/message-generator-tests.sh

@@ -3,13 +3,8 @@
 search_dir="test/message-generator/test/"
 
 for entry in `ls $search_dir`; do
-    if [ "$entry" = "interop-jdc-change-upstream" ]; then
-        echo "Skipping $entry"
-        continue
-    fi
-
     echo $entry
-    $search_dir$entry/$entry.sh 
+    $search_dir$entry/$entry.sh
 done
 
 cd roles