fix: main branch, offset type got updated (#2171)

Signed-off-by: Vigith Maurice <[email protected]>
numaproj · Oct 17, 2024 · 187398c · 187398c
1 parent dc137c2
commit 187398c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 16 deletions.
diff --git a/rust/numaflow-core/src/sink/blackhole.rs b/rust/numaflow-core/src/sink/blackhole.rs
@@ -22,6 +22,7 @@ mod tests {
     use chrono::Utc;
 
     use super::BlackholeSink;
+    use crate::message::IntOffset;
     use crate::{
         message::{Message, MessageID, Offset, ResponseFromSink, ResponseStatusFromSink},
         sink::Sink,
@@ -34,10 +35,7 @@ mod tests {
             Message {
                 keys: vec![],
                 value: b"Hello, World!".to_vec(),
-                offset: Offset {
-                    offset: "1".to_string(),
-                    partition_id: 0,
-                },
+                offset: Some(Offset::Int(IntOffset::new(1, 0))),
                 event_time: Utc::now(),
                 headers: Default::default(),
                 id: MessageID {
@@ -49,10 +47,7 @@ mod tests {
             Message {
                 keys: vec![],
                 value: b"Hello, World!".to_vec(),
-                offset: Offset {
-                    offset: "2".to_string(),
-                    partition_id: 0,
-                },
+                offset: Some(Offset::Int(IntOffset::new(1, 0))),
                 event_time: Utc::now(),
                 headers: Default::default(),
                 id: MessageID {

diff --git a/rust/numaflow-core/src/sink/log.rs b/rust/numaflow-core/src/sink/log.rs
@@ -38,6 +38,7 @@ mod tests {
     use chrono::Utc;
 
     use super::LogSink;
+    use crate::message::IntOffset;
     use crate::{
         message::{Message, MessageID, Offset, ResponseFromSink, ResponseStatusFromSink},
         sink::Sink,
@@ -50,10 +51,7 @@ mod tests {
             Message {
                 keys: vec![],
                 value: b"Hello, World!".to_vec(),
-                offset: Offset {
-                    offset: "1".to_string(),
-                    partition_id: 0,
-                },
+                offset: Some(Offset::Int(IntOffset::new(1, 0))),
                 event_time: Utc::now(),
                 headers: Default::default(),
                 id: MessageID {
@@ -65,10 +63,7 @@ mod tests {
             Message {
                 keys: vec![],
                 value: b"Hello, World!".to_vec(),
-                offset: Offset {
-                    offset: "2".to_string(),
-                    partition_id: 0,
-                },
+                offset: Some(Offset::Int(IntOffset::new(1, 0))),
                 event_time: Utc::now(),
                 headers: Default::default(),
                 id: MessageID {