From 9875c591b6ef77217a597e75f0550104e2741ae6 Mon Sep 17 00:00:00 2001
From: refcell <abigger87@gmail.com>
Date: Tue, 8 Oct 2024 18:36:21 -0400
Subject: [PATCH 1/2] chore(derive): batch timestamp tests

---
 crates/derive/src/batch/mod.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/crates/derive/src/batch/mod.rs b/crates/derive/src/batch/mod.rs
index 8e617466b4..4494ba0da4 100644
--- a/crates/derive/src/batch/mod.rs
+++ b/crates/derive/src/batch/mod.rs
@@ -106,3 +106,26 @@ impl Batch {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_timestamp() {
+        let single_batch = SingleBatch {
+            timestamp: 123,
+            ..Default::default()
+        };
+        let span_batch = SpanBatch {
+            batches: vec![SpanBatchElement {
+                timestamp: 456,
+                ..Default::default()
+            }],
+            ..Default::default()
+        };
+
+        assert_eq!(Batch::Single(single_batch).timestamp(), 123);
+        assert_eq!(Batch::Span(span_batch).timestamp(), 456);
+    }
+}

From cdb425704e90dcf2e2b63c9b9dc4816c4799f4cd Mon Sep 17 00:00:00 2001
From: refcell <abigger87@gmail.com>
Date: Tue, 8 Oct 2024 18:40:39 -0400
Subject: [PATCH 2/2] fix: lints

---
 crates/derive/src/batch/mod.rs | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/crates/derive/src/batch/mod.rs b/crates/derive/src/batch/mod.rs
index 4494ba0da4..1921bd714b 100644
--- a/crates/derive/src/batch/mod.rs
+++ b/crates/derive/src/batch/mod.rs
@@ -113,15 +113,9 @@ mod tests {
 
     #[test]
     fn test_timestamp() {
-        let single_batch = SingleBatch {
-            timestamp: 123,
-            ..Default::default()
-        };
+        let single_batch = SingleBatch { timestamp: 123, ..Default::default() };
         let span_batch = SpanBatch {
-            batches: vec![SpanBatchElement {
-                timestamp: 456,
-                ..Default::default()
-            }],
+            batches: vec![SpanBatchElement { timestamp: 456, ..Default::default() }],
             ..Default::default()
         };