test: sort transactions by first_seen and last_seen

Author: uvuvuwu

crates/chain/src/chain_data.rs

@@ -165,6 +165,7 @@ mod test {
     use crate::BlockId;
 
     use super::*;
+    use std::cmp::Ordering;
 
     #[test]
     fn chain_position_ord() {
@@ -250,4 +251,88 @@ mod test {
             ]
         );
     }
+
+    #[test]
+    fn test_sort_unconfirmed_chain_position() {
+        let mut v = vec![
+            ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                first_seen: Some(5),
+                last_seen: Some(20),
+            },
+            ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                first_seen: None,
+                last_seen: Some(30),
+            },
+            ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                first_seen: Some(1),
+                last_seen: Some(10),
+            },
+            ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                first_seen: Some(3),
+                last_seen: None,
+            },
+        ];
+
+        // Sort by first_seen: lowest first_seen first, None at the bottom
+        v.sort_by(|a, b| match (a, b) {
+            (
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed { first_seen: a, .. },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed { first_seen: b, .. },
+            ) => a.unwrap_or(u64::MAX).cmp(&b.unwrap_or(u64::MAX)),
+            _ => Ordering::Equal,
+        });
+
+        assert_eq!(
+            v,
+            vec![
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: Some(1),
+                    last_seen: Some(10)
+                },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: Some(3),
+                    last_seen: None
+                },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: Some(5),
+                    last_seen: Some(20)
+                },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: None,
+                    last_seen: Some(30)
+                },
+            ]
+        );
+
+        // Sort by last_seen: highest last_seen first, None at the bottom
+        v.sort_by(|a, b| match (a, b) {
+            (
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed { last_seen: a, .. },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed { last_seen: b, .. },
+            ) => b.unwrap_or(0).cmp(&a.unwrap_or(0)),
+            _ => Ordering::Equal,
+        });
+
+        assert_eq!(
+            v,
+            vec![
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: None,
+                    last_seen: Some(30)
+                },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: Some(5),
+                    last_seen: Some(20)
+                },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: Some(1),
+                    last_seen: Some(10)
+                },
+                ChainPosition::<ConfirmationBlockTime>::Unconfirmed {
+                    first_seen: Some(3),
+                    last_seen: None
+                },
+            ]
+        );
+    }
 }