add non_aggregated_translator_correctly_deals_with_group_channels Integration Test

Author: plebhash

integration-tests/tests/translator_integration.rs

@@ -309,4 +309,142 @@ async fn aggregated_translator_correctly_deals_with_group_channels() {
         share_channel_id, group_channel_id,
         "Share NOT submitted to the group channel ID"
     );
-}
+}
+
+// This test launches a tProxy in non-aggregated mode and leverages a MockUpstream to test the
+// correct functionalities of grouping standard channels.
+#[tokio::test]
+async fn non_aggregated_translator_correctly_deals_with_group_channels() {
+    start_tracing();
+    let (_tp, tp_addr) = start_template_provider(None, DifficultyLevel::Low);
+    let (_pool, pool_addr) = start_pool(sv2_tp_config(tp_addr), vec![], vec![]).await;
+    let (sniffer, sniffer_addr) = start_sniffer("0", pool_addr, false, vec![], None);
+
+    // non-aggregated tProxy
+    let (_, tproxy_addr) = start_sv2_translator(&[sniffer_addr], false, vec![], vec![]).await;
+
+    sniffer
+        .wait_for_message_type_and_clean_queue(
+            MessageDirection::ToDownstream,
+            MESSAGE_TYPE_SETUP_CONNECTION_SUCCESS,
+        )
+        .await;
+
+    // connect first minerd process
+    let (_minerd_process_1, _minerd_addr_1) = start_minerd(tproxy_addr, None, None, false).await;
+
+    sniffer
+        .wait_for_message_type_and_clean_queue(
+            MessageDirection::ToUpstream,
+            MESSAGE_TYPE_OPEN_EXTENDED_MINING_CHANNEL,
+        )
+        .await;
+
+    sniffer
+        .wait_for_message_type(
+            MessageDirection::ToDownstream,
+            MESSAGE_TYPE_OPEN_EXTENDED_MINING_CHANNEL_SUCCESS,
+        )
+        .await;
+
+    let (channel_id_a, group_channel_id) = match sniffer.next_message_from_upstream() {
+        Some((
+            _,
+            AnyMessage::Mining(parsers_sv2::Mining::OpenExtendedMiningChannelSuccess(msg)),
+        )) => (msg.channel_id, msg.group_channel_id),
+        msg => panic!(
+            "Expected OpenExtendedMiningChannelSuccess message, found: {:?}",
+            msg
+        ),
+    };
+
+    sniffer
+        .wait_for_message_type_and_clean_queue(
+            MessageDirection::ToDownstream,
+            MESSAGE_TYPE_NEW_EXTENDED_MINING_JOB,
+        )
+        .await;
+    sniffer
+        .wait_for_message_type_and_clean_queue(
+            MessageDirection::ToDownstream,
+            MESSAGE_TYPE_MINING_SET_NEW_PREV_HASH,
+        )
+        .await;
+
+    assert_ne!(
+        channel_id_a, group_channel_id,
+        "Channel ID must be different from the group channel ID"
+    );
+
+    // connect second minerd process
+    let (_minerd_process_2, _minerd_addr_2) = start_minerd(tproxy_addr, None, None, false).await;
+
+    sniffer
+        .wait_for_message_type_and_clean_queue(
+            MessageDirection::ToUpstream,
+            MESSAGE_TYPE_OPEN_EXTENDED_MINING_CHANNEL,
+        )
+        .await;
+
+    sniffer
+        .wait_for_message_type(
+            MessageDirection::ToDownstream,
+            MESSAGE_TYPE_OPEN_EXTENDED_MINING_CHANNEL_SUCCESS,
+        )
+        .await;
+
+    let (channel_id_b, group_channel_id) = match sniffer.next_message_from_upstream() {
+        Some((
+            _,
+            AnyMessage::Mining(parsers_sv2::Mining::OpenExtendedMiningChannelSuccess(msg)),
+        )) => (msg.channel_id, msg.group_channel_id),
+        msg => panic!(
+            "Expected OpenExtendedMiningChannelSuccess message, found: {:?}",
+            msg
+        ),
+    };
+
+    assert_ne!(
+        channel_id_b, group_channel_id,
+        "Channel ID must be different from the group channel ID"
+    );
+
+    sniffer
+        .wait_for_message_type_and_clean_queue(
+            MessageDirection::ToDownstream,
+            MESSAGE_TYPE_NEW_EXTENDED_MINING_JOB,
+        )
+        .await;
+    sniffer
+        .wait_for_message_type_and_clean_queue(
+            MessageDirection::ToDownstream,
+            MESSAGE_TYPE_MINING_SET_NEW_PREV_HASH,
+        )
+        .await;
+
+    // NewExtendedMiningJob and SetNewPrevHash messages arrived to tProxy
+    // eventually a share will be submitted
+
+    sniffer
+        .wait_for_message_type(
+            MessageDirection::ToUpstream,
+            MESSAGE_TYPE_SUBMIT_SHARES_EXTENDED,
+        )
+        .await;
+    let share_channel_id = match sniffer.next_message_from_downstream() {
+        Some((_, AnyMessage::Mining(parsers_sv2::Mining::SubmitSharesExtended(msg)))) => {
+            msg.channel_id
+        }
+        msg => panic!("Expected SubmitSharesExtended message, found: {:?}", msg),
+    };
+
+    assert!(
+        share_channel_id == channel_id_a || share_channel_id == channel_id_b,
+        "Share submitted to the correct channel ID"
+    );
+
+    assert_ne!(
+        share_channel_id, group_channel_id,
+        "Share NOT submitted to the group channel ID"
+    );
+}