calling super.tokentansfer before hook logic 🐞

defientco · Aug 4, 2023 · 4f039c8 · 4f039c8
1 parent c3b7f7f
commit 4f039c8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/ERC721ACH.sol b/src/ERC721ACH.sol
@@ -55,6 +55,7 @@ contract ERC721ACH is ERC721AC, IERC721ACH {
         uint256 startTokenId,
         uint256 quantity
     ) internal virtual override {
+        super._beforeTokenTransfers(from, to, startTokenId, quantity);
         IBeforeTokenTransfersHook beforeTokenTransfersHook = IBeforeTokenTransfersHook(hooks[HookType.BeforeTokenTransfers]);
         if (
             address(beforeTokenTransfersHook) != address(0) &&
@@ -78,6 +79,8 @@ contract ERC721ACH is ERC721AC, IERC721ACH {
         uint256 startTokenId,
         uint256 quantity
     ) internal virtual override {
+
+        super._afterTokenTransfers(from, to, startTokenId, quantity);
         IAfterTokenTransfersHook afterTokenTransfersHook = IAfterTokenTransfersHook(hooks[HookType.AfterTokenTransfers]);
         if (
             address(afterTokenTransfersHook) != address(0) &&

diff --git a/test/utils/ERC721ACHMock.sol b/test/utils/ERC721ACHMock.sol
@@ -7,6 +7,8 @@ contract ERC721ACHMock is ERC721ACH {
     bool public hooksEnabled;
     address public owner;
 
+    error Access_OnlyOwner();
+
     constructor(address _owner) ERC721ACH("ERC-721ACH Mock", "MOCK") {
         owner = _owner;
     }