ref(test): use function_selector in receiver mock

OpenZeppelin · Dec 10, 2024 · 22f6316 · 22f6316
1 parent bf2216d
commit 22f6316
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/examples/erc1155-supply/tests/mock/receiver.rs b/examples/erc1155-supply/tests/mock/receiver.rs
@@ -1,13 +1,31 @@
 #![allow(dead_code)]
 #![cfg(feature = "e2e")]
+use alloc::vec::Vec;
+
 use alloy::{
-    primitives::{fixed_bytes, Address, FixedBytes},
+    primitives::{fixed_bytes, Address, FixedBytes, U256},
     sol,
 };
 use e2e::Wallet;
+use stylus_sdk::{abi::Bytes, function_selector};
+
+const REC_RETVAL: FixedBytes<4> = FixedBytes(function_selector!(
+    "onERC1155Received",
+    Address,
+    Address,
+    U256,
+    U256,
+    Bytes
+));
 
-const REC_RETVAL: FixedBytes<4> = fixed_bytes!("f23a6e61");
-const BAT_RETVAL: FixedBytes<4> = fixed_bytes!("bc197c81");
+const BAT_RETVAL: FixedBytes<4> = FixedBytes(function_selector!(
+    "onERC1155BatchReceived",
+    Address,
+    Address,
+    Vec<U256>,
+    Vec<U256>,
+    Bytes
+));
 
 sol! {
     #[allow(missing_docs)]