From d6a1e7498fa1aa5bde5859833c8c4f717ab7f36d Mon Sep 17 00:00:00 2001
From: apfitzge <apfitzge@users.noreply.github.com>
Date: Wed, 7 Sep 2022 12:52:18 -0500
Subject: [PATCH] Add tests for deserialize_and_collect_packets (#27623)

---
 core/src/packet_deserializer.rs | 52 +++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/core/src/packet_deserializer.rs b/core/src/packet_deserializer.rs
index 07fc9322f..3511d2540 100644
--- a/core/src/packet_deserializer.rs
+++ b/core/src/packet_deserializer.rs
@@ -136,3 +136,55 @@ impl PacketDeserializer {
         })
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use {
+        super::*,
+        solana_perf::packet::to_packet_batches,
+        solana_sdk::{
+            hash::Hash, pubkey::Pubkey, signature::Keypair, system_transaction,
+            transaction::Transaction,
+        },
+    };
+
+    fn random_transfer() -> Transaction {
+        system_transaction::transfer(&Keypair::new(), &Pubkey::new_unique(), 1, Hash::default())
+    }
+
+    #[test]
+    fn test_deserialize_and_collect_packets_empty() {
+        let results = PacketDeserializer::deserialize_and_collect_packets(&[], None);
+        assert_eq!(results.deserialized_packets.len(), 0);
+        assert!(results.new_tracer_stats_option.is_none());
+        assert_eq!(results.passed_sigverify_count, 0);
+        assert_eq!(results.failed_sigverify_count, 0);
+    }
+
+    #[test]
+    fn test_deserialize_and_collect_packets_simple_batches() {
+        let transactions = vec![random_transfer(), random_transfer()];
+        let packet_batches = to_packet_batches(&transactions, 1);
+        assert_eq!(packet_batches.len(), 2);
+
+        let results = PacketDeserializer::deserialize_and_collect_packets(&packet_batches, None);
+        assert_eq!(results.deserialized_packets.len(), 2);
+        assert!(results.new_tracer_stats_option.is_none());
+        assert_eq!(results.passed_sigverify_count, 2);
+        assert_eq!(results.failed_sigverify_count, 0);
+    }
+
+    #[test]
+    fn test_deserialize_and_collect_packets_simple_batches_with_failure() {
+        let transactions = vec![random_transfer(), random_transfer()];
+        let mut packet_batches = to_packet_batches(&transactions, 1);
+        assert_eq!(packet_batches.len(), 2);
+        packet_batches[0][0].meta.set_discard(true);
+
+        let results = PacketDeserializer::deserialize_and_collect_packets(&packet_batches, None);
+        assert_eq!(results.deserialized_packets.len(), 1);
+        assert!(results.new_tracer_stats_option.is_none());
+        assert_eq!(results.passed_sigverify_count, 1);
+        assert_eq!(results.failed_sigverify_count, 1);
+    }
+}