Add multiple sections test

Author: iddm

Cargo.toml

@@ -92,6 +92,10 @@ crate-type = ["cdylib"]
 name = "info_handler_struct"
 crate-type = ["cdylib"]
 
+[[example]]
+name = "info_handler_multiple_sections"
+crate-type = ["cdylib"]
+
 [[example]]
 name = "info"
 crate-type = ["cdylib"]

examples/info_handler_multiple_sections.rs

@@ -0,0 +1,39 @@
+use std::collections::HashMap;
+
+use redis_module::InfoContext;
+use redis_module::{redis_module, RedisResult};
+use redis_module_macros::{info_command_handler, InfoSection};
+
+#[derive(Debug, Clone, InfoSection)]
+struct InfoSection1 {
+    field_1: String,
+}
+
+#[derive(Debug, Clone, InfoSection)]
+struct InfoSection2 {
+    field_2: String,
+}
+
+#[info_command_handler]
+fn add_info(ctx: &InfoContext, _for_crash_report: bool) -> RedisResult<()> {
+    let data = InfoSection1 {
+        field_1: "value1".to_owned(),
+    };
+    let _ = ctx.build_one_section(data);
+
+    let data = InfoSection2 {
+        field_2: "value2".to_owned(),
+    };
+
+    ctx.build_one_section(data)
+}
+
+//////////////////////////////////////////////////////
+
+redis_module! {
+    name: "info_handler_multiple_sections",
+    version: 1,
+    allocator: (redis_module::alloc::RedisAlloc, redis_module::alloc::RedisAlloc),
+    data_types: [],
+    commands: [],
+}

tests/integration.rs

@@ -133,7 +133,7 @@ fn test_helper_info() -> Result<()> {
                 .arg(module)
                 .query(&mut con)
                 .with_context(|| format!("failed to run INFO {module}"))?;
-            println!("Now processing {module}. Result: {res}.");
+
             assert!(res.contains(&format!("{module}_field:value")));
             if has_dictionary {
                 assert!(res.contains("dictionary:key=value"));
@@ -143,6 +143,29 @@ fn test_helper_info() -> Result<()> {
         })
 }
 
+#[test]
+fn test_info_handler_multiple_sections() -> Result<()> {
+    const MODULES: [&str; 1] = ["info_handler_multiple_sections"];
+
+    MODULES.into_iter().try_for_each(|module| {
+        let port: u16 = 6500;
+        let _guards = vec![start_redis_server_with_module(module, port)
+            .with_context(|| "failed to start redis server")?];
+        let mut con =
+            get_redis_connection(port).with_context(|| "failed to connect to redis server")?;
+
+        let res: String = redis::cmd("INFO")
+            .arg(format!("{module}_InfoSection2"))
+            .query(&mut con)
+            .with_context(|| format!("failed to run INFO {module}"))?;
+
+        assert!(res.contains(&format!("{module}_field_2:value2")));
+        assert!(!res.contains(&format!("{module}_field_1:value1")));
+
+        Ok(())
+    })
+}
+
 #[allow(unused_must_use)]
 #[test]
 fn test_test_helper_err() -> Result<()> {