Provide correct message type in Sniffer

Previously we pushed the correct message but wrong message type after an intercept. This is now fixed with using the message type from the intercepet message directly.
stratum-mining · Jan 27, 2025 · f024e1d · f024e1d
1 parent 4c7992d
commit f024e1d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/roles/tests-integration/lib/sniffer.rs b/roles/tests-integration/lib/sniffer.rs
@@ -234,8 +234,10 @@ impl Sniffer {
                         )
                         .expect("Failed to create the frame"),
                     );
-                    downstream_messages
-                        .add_message(msg_type, intercept_message.replacement_message.clone());
+                    downstream_messages.add_message(
+                        intercept_message.replacement_message.message_type(),
+                        intercept_message.replacement_message.clone(),
+                    );
                     let _ = send.send(frame).await;
                 }
             }
@@ -271,8 +273,10 @@ impl Sniffer {
                         )
                         .expect("Failed to create the frame"),
                     );
-                    upstream_messages
-                        .add_message(msg_type, intercept_message.replacement_message.clone());
+                    upstream_messages.add_message(
+                        intercept_message.replacement_message.message_type(),
+                        intercept_message.replacement_message.clone(),
+                    );
                     let _ = send.send(frame).await;
                 }
             }