From 568b11d71e69855ebd379230abed51464d207af5 Mon Sep 17 00:00:00 2001
From: Alex Beregszaszi <alex@rtfs.hu>
Date: Thu, 4 Jul 2019 16:54:10 +0100
Subject: [PATCH] rust: add two test cases to cover ExecutionMessage

---
 bindings/rust/evmc-vm/src/lib.rs | 68 ++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)

diff --git a/bindings/rust/evmc-vm/src/lib.rs b/bindings/rust/evmc-vm/src/lib.rs
index ea8a329..1a79841 100644
--- a/bindings/rust/evmc-vm/src/lib.rs
+++ b/bindings/rust/evmc-vm/src/lib.rs
@@ -567,6 +567,74 @@ mod tests {
         }
     }
 
+    #[test]
+    fn message_from_ffi() {
+        let destination = ffi::evmc_address { bytes: [32u8; 20] };
+        let sender = ffi::evmc_address { bytes: [128u8; 20] };
+        let value = ffi::evmc_uint256be { bytes: [0u8; 32] };
+        let create2_salt = ffi::evmc_bytes32 { bytes: [255u8; 32] };
+
+        let msg = ffi::evmc_message {
+            kind: ffi::evmc_call_kind::EVMC_CALL,
+            flags: 44,
+            depth: 66,
+            gas: 4466,
+            destination: destination,
+            sender: sender,
+            input_data: std::ptr::null(),
+            input_size: 0,
+            value: value,
+            create2_salt: create2_salt,
+        };
+
+        let ret: ExecutionMessage = (&msg).into();
+
+        assert_eq!(ret.kind(), msg.kind);
+        assert_eq!(ret.flags(), msg.flags);
+        assert_eq!(ret.depth(), msg.depth);
+        assert_eq!(ret.gas(), msg.gas);
+        assert_eq!(*ret.destination(), msg.destination);
+        assert_eq!(*ret.sender(), msg.sender);
+        assert!(ret.input().is_none());
+        assert_eq!(*ret.value(), msg.value);
+        assert_eq!(*ret.create2_salt(), msg.create2_salt);
+    }
+
+    #[test]
+    fn message_from_ffi_with_input() {
+        let input = vec![0xc0, 0xff, 0xee];
+        let destination = ffi::evmc_address { bytes: [32u8; 20] };
+        let sender = ffi::evmc_address { bytes: [128u8; 20] };
+        let value = ffi::evmc_uint256be { bytes: [0u8; 32] };
+        let create2_salt = ffi::evmc_bytes32 { bytes: [255u8; 32] };
+
+        let msg = ffi::evmc_message {
+            kind: ffi::evmc_call_kind::EVMC_CALL,
+            flags: 44,
+            depth: 66,
+            gas: 4466,
+            destination: destination,
+            sender: sender,
+            input_data: input.as_ptr(),
+            input_size: input.len(),
+            value: value,
+            create2_salt: create2_salt,
+        };
+
+        let ret: ExecutionMessage = (&msg).into();
+
+        assert_eq!(ret.kind(), msg.kind);
+        assert_eq!(ret.flags(), msg.flags);
+        assert_eq!(ret.depth(), msg.depth);
+        assert_eq!(ret.gas(), msg.gas);
+        assert_eq!(*ret.destination(), msg.destination);
+        assert_eq!(*ret.sender(), msg.sender);
+        assert!(ret.input().is_some());
+        assert_eq!(*ret.input().unwrap(), input);
+        assert_eq!(*ret.value(), msg.value);
+        assert_eq!(*ret.create2_salt(), msg.create2_salt);
+    }
+
     unsafe extern "C" fn get_dummy_tx_context(
         _context: *mut ffi::evmc_context,
     ) -> ffi::evmc_tx_context {