add logging tests from the suite (#93)

dagster-io · Jan 10, 2025 · b9675a6 · b9675a6
1 parent 02d2476
commit b9675a6
Show file tree

Hide file tree

Showing 3 changed files with 548 additions and 3 deletions.
diff --git a/libraries/pipes/implementations/rust/pipes.toml b/libraries/pipes/implementations/rust/pipes.toml
@@ -2,7 +2,7 @@
 error_reporting = false
 
 [messages]
-log = false
+log = true
 report_custom_message = false
 report_asset_materialization = false
 report_asset_check = false

diff --git a/libraries/pipes/implementations/rust/src/bin/pipes_tests.rs b/libraries/pipes/implementations/rust/src/bin/pipes_tests.rs
@@ -1,5 +1,5 @@
 use clap::Parser;
-use dagster_pipes_rust::{open_dagster_pipes, DagsterPipesError};
+use dagster_pipes_rust::{open_dagster_pipes, DagsterPipesError, MessageWriter, PipesContext};
 use dagster_pipes_rust::{DAGSTER_PIPES_CONTEXT_ENV_VAR, DAGSTER_PIPES_MESSAGES_ENV_VAR};
 use std::collections::HashMap;
 use std::fs::File;
@@ -49,5 +49,22 @@ pub fn main() -> Result<(), DagsterPipesError> {
             serde_json::from_reader(file).expect("extras could not be parsed");
         assert_eq!(context.data.extras, Some(json));
     }
+
+    match args.test_name.as_str() {
+        "test_message_log" => test_message_log(context),
+        _ => Ok(()),
+    }?;
+    Ok(())
+}
+
+fn test_message_log<W>(mut context: PipesContext<W>) -> Result<(), DagsterPipesError>
+where
+    W: MessageWriter,
+{
+    context.logger.debug("Debug message")?;
+    context.logger.info("Info message")?;
+    context.logger.warning("Warning message")?;
+    context.logger.error("Error message")?;
+    context.logger.critical("Critical message")?;
     Ok(())
 }