Improve test coverage

packages/cw-hooks/src/lib.rs

@@ -107,7 +107,7 @@ impl<'a> Hooks<'a> {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use cosmwasm_std::{coins, testing::mock_dependencies, BankMsg};
+    use cosmwasm_std::{coins, testing::mock_dependencies, BankMsg, Empty};
 
     // Shorthand for an unchecked address.
     macro_rules! addr {
@@ -151,6 +151,30 @@ mod tests {
             )]
         );
 
+        // Test prepare hooks with custom messages.
+        // In a real world scenario, you would be using something like
+        // TokenFactoryMsg.
+        let msgs = hooks
+            .prepare_hooks_custom_msg(storage, |a| {
+                Ok(SubMsg::<Empty>::reply_always(
+                    BankMsg::Burn {
+                        amount: coins(a.as_str().len() as u128, "uekez"),
+                    },
+                    2,
+                ))
+            })
+            .unwrap();
+
+        assert_eq!(
+            msgs,
+            vec![SubMsg::<Empty>::reply_always(
+                BankMsg::Burn {
+                    amount: coins(4, "uekez"),
+                },
+                2,
+            )]
+        );
+
         let HooksResponse { hooks: the_hooks } = hooks.query_hooks(deps.as_ref()).unwrap();
 
         assert_eq!(the_hooks, vec![addr!("meow")]);