ADD: Add status test data

databento · Jul 3, 2024 · fa7afa2 · fa7afa2
1 parent e2375e2
commit fa7afa2
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/rust/dbn/src/decode/dbn/async.rs b/rust/dbn/src/decode/dbn/async.rs
@@ -665,7 +665,7 @@ mod tests {
             DbnEncodable,
         },
         rtype, CbboMsg, Error, ErrorMsg, ImbalanceMsg, InstrumentDefMsg, MboMsg, Mbp10Msg, Mbp1Msg,
-        OhlcvMsg, RecordHeader, Result, Schema, StatMsg, TbboMsg, TradeMsg, WithTsOut,
+        OhlcvMsg, RecordHeader, Result, Schema, StatMsg, StatusMsg, TbboMsg, TradeMsg, WithTsOut,
     };
 
     #[rstest]
@@ -682,6 +682,7 @@ mod tests {
     #[case::definitions(Schema::Definition, InstrumentDefMsg::default())]
     #[case::imbalance(Schema::Imbalance, ImbalanceMsg::default())]
     #[case::statistics(Schema::Statistics, StatMsg::default())]
+    #[case::status(Schema::Status, StatusMsg::default())]
     #[tokio::test]
     async fn test_dbn_identity<R: DbnEncodable + HasRType + PartialEq + Clone>(
         #[case] schema: Schema,
@@ -726,6 +727,7 @@ mod tests {
     #[case::definitions(Schema::Definition, InstrumentDefMsg::default())]
     #[case::imbalance(Schema::Imbalance, ImbalanceMsg::default())]
     #[case::statistics(Schema::Statistics, StatMsg::default())]
+    #[case::status(Schema::Status, StatusMsg::default())]
     #[tokio::test]
     async fn test_dbn_zstd_identity<R: DbnEncodable + HasRType + PartialEq + Clone>(
         #[case] schema: Schema,

diff --git a/rust/dbn/src/decode/dbn/sync.rs b/rust/dbn/src/decode/dbn/sync.rs
@@ -685,8 +685,8 @@ mod tests {
             dbn::Encoder, DbnEncodable, DbnRecordEncoder, DynWriter, EncodeDbn, EncodeRecord,
         },
         rtype, CbboMsg, Compression, Error, ErrorMsg, ImbalanceMsg, InstrumentDefMsg, MboMsg,
-        Mbp10Msg, Mbp1Msg, MetadataBuilder, OhlcvMsg, RecordHeader, Result, StatMsg, TbboMsg,
-        TradeMsg, WithTsOut, SYMBOL_CSTR_LEN,
+        Mbp10Msg, Mbp1Msg, MetadataBuilder, OhlcvMsg, RecordHeader, Result, StatMsg, StatusMsg,
+        TbboMsg, TradeMsg, WithTsOut, SYMBOL_CSTR_LEN,
     };
 
     #[test]
@@ -865,6 +865,7 @@ mod tests {
         Compression::None,
         StatMsg::default()
     )]
+    #[case::uncompressed_status_v2(2, Schema::Status, Compression::None, StatusMsg::default())]
     #[case::zstd_mbo_v2(2, Schema::Mbo, Compression::ZStd, MboMsg::default())]
     #[case::zstd_trades_v2(2, Schema::Trades, Compression::ZStd, TradeMsg::default())]
     #[case::zstd_tbbo_v2(2, Schema::Tbbo, Compression::ZStd, TbboMsg::default())]
@@ -908,6 +909,7 @@ mod tests {
     )]
     #[case::zstd_imbalance_v2(2, Schema::Imbalance, Compression::ZStd, ImbalanceMsg::default())]
     #[case::zstd_statistics_v2(2, Schema::Statistics, Compression::ZStd, StatMsg::default())]
+    #[case::zstd_status_v2(2, Schema::Status, Compression::ZStd, StatusMsg::default())]
     fn test_dbn_identity<R: DbnEncodable + HasRType + PartialEq + Clone>(
         #[case] version: u8,
         #[case] schema: Schema,

diff --git a/tests/data/test_data.status.dbn b/tests/data/test_data.status.dbn
diff --git a/tests/data/test_data.status.dbn.zst b/tests/data/test_data.status.dbn.zst