feat(gtest): add self-sent delayed message test (#4240)

examples/delayed-sender/src/wasm.rs

@@ -22,15 +22,34 @@ use gstd::{exec, msg, MessageId};
 static mut MID: Option<MessageId> = None;
 static mut DONE: bool = false;
 
+fn send_delayed_to_self() -> bool {
+    let to_self = msg::load_bytes().unwrap().as_slice() == b"self";
+    if to_self {
+        gstd::debug!("sending delayed message to self");
+        msg::send_bytes_delayed(exec::program_id(), b"self", 0, DELAY).unwrap();
+    }
+
+    to_self
+}
+
 #[no_mangle]
 extern "C" fn init() {
+    // Send message to self
+    if send_delayed_to_self() {
+        return;
+    }
+
     let delay: u32 = msg::load().unwrap();
 
     msg::send_bytes_delayed(msg::source(), "Delayed hello!", 0, delay).unwrap();
 }
 
 #[no_mangle]
 extern "C" fn handle() {
+    if send_delayed_to_self() {
+        return;
+    }
+
     let size = msg::size();
 
     if size == 0 {

gtest/Cargo.toml

@@ -40,3 +40,4 @@ demo-piggy-bank.workspace = true
 demo-ping.workspace = true
 demo-futures-unordered.workspace = true
 demo-constructor = { workspace = true, features = ["std"] }
+demo-delayed-sender.workspace = true

gtest/src/program.rs

@@ -1484,7 +1484,7 @@ mod tests {
     #[test]
     fn tests_unused_gas_value_not_transferred() {
         let sys = System::new();
-        sys.init_verbose_logger();
+        sys.init_logger();
 
         let user = 42;
         sys.mint_to(user, 2 * EXISTENTIAL_DEPOSIT);
@@ -1497,4 +1497,30 @@ mod tests {
         // lower than ED
         assert_eq!(sys.balance_of(user), 0)
     }
+
+    #[test]
+    fn tests_self_sent_delayed_message() {
+        use demo_delayed_sender::DELAY;
+
+        let sys = System::new();
+        sys.init_logger();
+
+        let user = DEFAULT_USER_ALICE;
+        let program_id = 69;
+
+        let prog = Program::from_binary_with_id(&sys, program_id, demo_delayed_sender::WASM_BINARY);
+
+        // Init message starts sequence of self-sent messages
+        prog.send_bytes(user, "self".as_bytes());
+        let res = sys.run_next_block();
+        assert_eq!(res.succeed.len(), 1);
+
+        let mut target_block_nb = sys.block_height() + DELAY;
+        let res = sys.run_to_block(target_block_nb);
+        assert_eq!(res.iter().last().unwrap().succeed.len(), 1);
+
+        target_block_nb += DELAY;
+        let res = sys.run_to_block(target_block_nb);
+        assert_eq!(res.iter().last().unwrap().succeed.len(), 1);
+    }
 }