chore: use StreamIDType in tests

ceramicnetwork · Aug 23, 2024 · 4c61204 · 4c61204
1 parent 5a1a0f2
commit 4c61204
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/flight/Cargo.toml b/flight/Cargo.toml
@@ -13,6 +13,7 @@ cid.workspace = true
 serde.workspace = true
 chrono = { version = "0.4", features = ["serde"] }
 arrow.workspace = true
+ceramic-core = { path = "../core" }
 
 [dev-dependencies]
 tokio = { workspace = true, features = ["macros", "rt"] }

diff --git a/flight/src/tests.rs b/flight/src/tests.rs
@@ -1,6 +1,7 @@
 use super::*;
 use crate::types::{ConclusionData, ConclusionEvent, ConclusionInit};
 use arrow::array::{Array, BinaryArray, ListArray, StringArray, UInt8Array};
+use ceramic_core::StreamIdType;
 use cid::Cid;
 
 /// Tests the conversion of ConclusionEvents to Arrow RecordBatch.
@@ -20,7 +21,7 @@ fn test_conclusion_events_to_record_batch() {
         ConclusionEvent::Data(ConclusionData {
             id: Cid::default(),
             init: ConclusionInit {
-                stream_type: 0,
+                stream_type: StreamIdType::Model as u8,
                 controller: "did:key:test1".to_string(),
                 dimensions: vec![],
             },
@@ -30,7 +31,7 @@ fn test_conclusion_events_to_record_batch() {
         ConclusionEvent::Data(ConclusionData {
             id: Cid::default(),
             init: ConclusionInit {
-                stream_type: 2,
+                stream_type: StreamIdType::Model as u8,
                 controller: "did:key:test2".to_string(),
                 dimensions: vec![],
             },
@@ -70,7 +71,7 @@ fn test_conclusion_events_to_record_batch() {
         .as_any()
         .downcast_ref::<UInt8Array>()
         .unwrap();
-    assert_eq!(stream_types.value(0), 0);
+    assert_eq!(stream_types.value(0), 2);
     assert_eq!(stream_types.value(1), 2);
 
     let controllers = record_batch