Add ERC1155

src/tests/mocks/erc1155_mocks.cairo

@@ -4,8 +4,8 @@ mod DualCaseERC1155Mock {
     use openzeppelin::token::erc1155::ERC1155Component;
     use starknet::ContractAddress;
 
-    component!(path: SRC5Component, storage: src5, event: SRC5Event);
     component!(path: ERC1155Component, storage: erc1155, event: ERC1155Event);
+    component!(path: SRC5Component, storage: src5, event: SRC5Event);
 
     // ERC1155
     #[abi(embed_v0)]
@@ -15,7 +15,6 @@ mod DualCaseERC1155Mock {
         ERC1155Component::ERC1155MetadataURIImpl<ContractState>;
     #[abi(embed_v0)]
     impl ERC721Camel = ERC1155Component::ERC1155CamelImpl<ContractState>;
-
     impl ERC1155InternalImpl = ERC1155Component::InternalImpl<ContractState>;
 
     // SRC5
@@ -48,11 +47,7 @@ mod DualCaseERC1155Mock {
         uri: ByteArray
     ) {
         self.erc1155.initializer(uri);
-
-        // mint
-        let token_ids = array![token_id].span();
-        let values = array![value].span();
-        self.erc1155.update(Zeroable::zero(), recipient, token_ids, values);
+        self.erc1155.mint_with_acceptance_check(recipient, token_id, value, array![].span());
     }
 }
 
@@ -62,16 +57,15 @@ mod SnakeERC1155Mock {
     use openzeppelin::token::erc1155::ERC1155Component;
     use starknet::ContractAddress;
 
-    component!(path: SRC5Component, storage: src5, event: SRC5Event);
     component!(path: ERC1155Component, storage: erc1155, event: ERC1155Event);
+    component!(path: SRC5Component, storage: src5, event: SRC5Event);
 
     // ERC1155
     #[abi(embed_v0)]
     impl ERC1155Impl = ERC1155Component::ERC1155Impl<ContractState>;
     #[abi(embed_v0)]
     impl ERC1155MetadataURIImpl =
         ERC1155Component::ERC1155MetadataURIImpl<ContractState>;
-
     impl ERC1155InternalImpl = ERC1155Component::InternalImpl<ContractState>;
 
     // SRC5
@@ -104,28 +98,25 @@ mod SnakeERC1155Mock {
         uri: ByteArray
     ) {
         self.erc1155.initializer(uri);
-
-        // mint
-        let token_ids = array![token_id].span();
-        let values = array![value].span();
-        self.erc1155.update(Zeroable::zero(), recipient, token_ids, values);
+        self.erc1155.mint_with_acceptance_check(recipient, token_id, value, array![].span());
     }
 }
 
 #[starknet::contract]
 mod CamelERC1155Mock {
     use openzeppelin::introspection::src5::SRC5Component;
-    use openzeppelin::token::erc1155::ERC1155Component::{ERC1155Impl, ERC1155MetadataURIImpl};
     use openzeppelin::token::erc1155::ERC1155Component;
     use starknet::ContractAddress;
 
-    component!(path: SRC5Component, storage: src5, event: SRC5Event);
     component!(path: ERC1155Component, storage: erc1155, event: ERC1155Event);
+    component!(path: SRC5Component, storage: src5, event: SRC5Event);
 
     // ERC1155
     #[abi(embed_v0)]
     impl ERC1155Camel = ERC1155Component::ERC1155CamelImpl<ContractState>;
-
+    #[abi(embed_v0)]
+    impl ERC1155MetadataURIImpl =
+        ERC1155Component::ERC1155MetadataURIImpl<ContractState>;
     impl ERC1155InternalImpl = ERC1155Component::InternalImpl<ContractState>;
 
     // SRC5
@@ -158,11 +149,7 @@ mod CamelERC1155Mock {
         uri: ByteArray
     ) {
         self.erc1155.initializer(uri);
-
-        // mint
-        let token_ids = array![token_id].span();
-        let values = array![value].span();
-        self.erc1155.update(Zeroable::zero(), recipient, token_ids, values);
+        self.erc1155.mint_with_acceptance_check(recipient, token_id, value, array![].span());
     }
 }
 
@@ -177,44 +164,24 @@ mod SnakeERC1155PanicMock {
     #[abi(per_item)]
     #[generate_trait]
     impl ExternalImpl of ExternalTrait {
-        #[external(v0)]
-        fn supports_interface(self: @ContractState, interface_id: felt252) -> bool {
-            panic!("Some error");
-            false
-        }
-
         #[external(v0)]
         fn uri(self: @ContractState, token_id: u256) -> ByteArray {
             panic!("Some error");
             "3"
         }
 
         #[external(v0)]
-        fn balance_of(self: @ContractState, account: ContractAddress) -> u256 {
+        fn balance_of(self: @ContractState, account: ContractAddress, token_id: u256) -> u256 {
             panic!("Some error");
             u256 { low: 3, high: 3 }
         }
 
         #[external(v0)]
-        fn is_approved_for_all(
-            self: @ContractState, owner: ContractAddress, operator: ContractAddress
-        ) -> bool {
-            panic!("Some error");
-            false
-        }
-
-        #[external(v0)]
-        fn set_approval_for_all(
-            ref self: ContractState, operator: ContractAddress, approved: bool
-        ) {
-            panic!("Some error");
-        }
-
-        #[external(v0)]
-        fn transfer_from(
-            ref self: ContractState, from: ContractAddress, to: ContractAddress, token_id: u256
-        ) {
+        fn balance_of_batch(
+            self: @ContractState, accounts: Span<ContractAddress>, token_ids: Span<u256>
+        ) -> Span<u256> {
             panic!("Some error");
+            array![u256 { low: 3, high: 3 }].span()
         }
 
         #[external(v0)]
@@ -223,19 +190,12 @@ mod SnakeERC1155PanicMock {
             from: ContractAddress,
             to: ContractAddress,
             token_id: u256,
+            value: u256,
             data: Span<felt252>
         ) {
             panic!("Some error");
         }
 
-        #[external(v0)]
-        fn balance_of_batch(
-            self: @ContractState, accounts: Span<ContractAddress>, token_ids: Span<u256>
-        ) -> Span<u256> {
-            panic!("Some error");
-            array![u256 { low: 3, high: 3 }].span()
-        }
-
         #[external(v0)]
         fn safe_batch_transfer_from(
             ref self: ContractState,
@@ -249,15 +209,25 @@ mod SnakeERC1155PanicMock {
         }
 
         #[external(v0)]
-        fn batch_transfer_from(
-            ref self: ContractState,
-            from: starknet::ContractAddress,
-            to: starknet::ContractAddress,
-            token_ids: Span<u256>,
-            values: Span<u256>
+        fn is_approved_for_all(
+            self: @ContractState, owner: ContractAddress, operator: ContractAddress
+        ) -> bool {
+            panic!("Some error");
+            false
+        }
+
+        #[external(v0)]
+        fn set_approval_for_all(
+            ref self: ContractState, operator: ContractAddress, approved: bool
         ) {
             panic!("Some error");
         }
+
+        #[external(v0)]
+        fn supports_interface(self: @ContractState, interface_id: felt252) -> bool {
+            panic!("Some error");
+            false
+        }
     }
 }
 
@@ -272,54 +242,24 @@ mod CamelERC1155PanicMock {
     #[abi(per_item)]
     #[generate_trait]
     impl ExternalImpl of ExternalTrait {
-        #[external(v0)]
-        fn supportsInterface(self: @ContractState, interfaceId: felt252) -> bool {
-            panic!("Some error");
-            false
-        }
-
-        #[external(v0)]
-        fn name(self: @ContractState) -> ByteArray {
-            panic!("Some error");
-            "3"
-        }
-
-        #[external(v0)]
-        fn symbol(self: @ContractState) -> ByteArray {
-            panic!("Some error");
-            "3"
-        }
-
         #[external(v0)]
         fn uri(self: @ContractState, tokenId: u256) -> ByteArray {
             panic!("Some error");
             "3"
         }
 
         #[external(v0)]
-        fn balanceOf(self: @ContractState, account: ContractAddress) -> u256 {
+        fn balanceOf(self: @ContractState, account: ContractAddress, tokenId: u256) -> u256 {
             panic!("Some error");
             u256 { low: 3, high: 3 }
         }
 
         #[external(v0)]
-        fn isApprovedForAll(
-            self: @ContractState, owner: ContractAddress, operator: ContractAddress
-        ) -> bool {
-            panic!("Some error");
-            false
-        }
-
-        #[external(v0)]
-        fn setApprovalForAll(ref self: ContractState, operator: ContractAddress, approved: bool) {
-            panic!("Some error");
-        }
-
-        #[external(v0)]
-        fn transferFrom(
-            ref self: ContractState, from: ContractAddress, to: ContractAddress, tokenId: u256
-        ) {
+        fn balanceOfBatch(
+            self: @ContractState, accounts: Span<ContractAddress>, token_ids: Span<u256>
+        ) -> Span<u256> {
             panic!("Some error");
+            array![u256 { low: 3, high: 3 }].span()
         }
 
         #[external(v0)]
@@ -328,19 +268,12 @@ mod CamelERC1155PanicMock {
             from: ContractAddress,
             to: ContractAddress,
             tokenId: u256,
+            value: u256,
             data: Span<felt252>
         ) {
             panic!("Some error");
         }
 
-        #[external(v0)]
-        fn balanceOfBatch(
-            self: @ContractState, accounts: Span<ContractAddress>, token_ids: Span<u256>
-        ) -> Span<u256> {
-            panic!("Some error");
-            array![u256 { low: 3, high: 3 }].span()
-        }
-
         #[external(v0)]
         fn safeBatchTransferFrom(
             ref self: ContractState,
@@ -354,14 +287,22 @@ mod CamelERC1155PanicMock {
         }
 
         #[external(v0)]
-        fn batchTransferFrom(
-            ref self: ContractState,
-            from: starknet::ContractAddress,
-            to: starknet::ContractAddress,
-            token_ids: Span<u256>,
-            values: Span<u256>
-        ) {
+        fn isApprovedForAll(
+            self: @ContractState, owner: ContractAddress, operator: ContractAddress
+        ) -> bool {
+            panic!("Some error");
+            false
+        }
+
+        #[external(v0)]
+        fn setApprovalForAll(ref self: ContractState, operator: ContractAddress, approved: bool) {
+            panic!("Some error");
+        }
+
+        #[external(v0)]
+        fn supportsInterface(self: @ContractState, interfaceId: felt252) -> bool {
             panic!("Some error");
+            false
         }
     }
 }

src/tests/mocks/erc1155_receiver_mocks.cairo

@@ -211,7 +211,20 @@ mod SnakeERC1155ReceiverPanicMock {
         value: u256,
         data: Span<felt252>
     ) -> felt252 {
-        panic_with_felt252('Some error');
+        panic!("Some error");
+        3
+    }
+
+    #[external(v0)]
+    fn on_erc1155_batch_received(
+        self: @ContractState,
+        operator: ContractAddress,
+        from: ContractAddress,
+        tokenIds: Span<u256>,
+        values: Span<u256>,
+        data: Span<felt252>
+    ) -> felt252 {
+        panic!("Some error");
         3
     }
 }
@@ -232,7 +245,20 @@ mod CamelERC1155ReceiverPanicMock {
         value: u256,
         data: Span<felt252>
     ) -> felt252 {
-        panic_with_felt252('Some error');
+        panic!("Some error");
+        3
+    }
+
+    #[external(v0)]
+    fn onERC1155BatchReceived(
+        self: @ContractState,
+        operator: ContractAddress,
+        from: ContractAddress,
+        tokenIds: Span<u256>,
+        values: Span<u256>,
+        data: Span<felt252>
+    ) -> felt252 {
+        panic!("Some error");
         3
     }
 }

src/tests/presets.cairo

@@ -1,4 +1,5 @@
 mod test_account;
+mod test_erc1155;
 mod test_erc20;
 mod test_erc721;
 mod test_eth_account;