Update StateFuzzingTest

Author: kaihiroi

devkit/MCTest.sol

@@ -1,8 +1,9 @@
 // SPDX-License-Identifier: MIT
 pragma solidity ^0.8.24;
 
+import {Proxy as OZProxy} from "@oz.ucs/proxy/Proxy.sol";
+
 import {System} from "devkit/system/System.sol";
-import {DecodeErrorString} from "devkit/system/message/DecodeErrorString.sol";
 import {Receive} from "mc-std/functions/Receive.sol";
 import {Formatter} from "devkit/types/Formatter.sol";
 import {ProxyUtils} from "@ucs.mc/proxy/ProxyUtils.sol";
@@ -23,7 +24,7 @@ abstract contract MCTest is MCTestBase {
 }
 
 // 🌟 MC State Fuzzing Test
-abstract contract MCStateFuzzingTest is MCTestBase { // solhint-disable-line payable-fallback
+abstract contract MCStateFuzzingTest is MCTestBase, OZProxy { // solhint-disable-line payable-fallback
     struct Function {
         bytes4 selector;
         address implementation;
@@ -49,17 +50,10 @@ abstract contract MCStateFuzzingTest is MCTestBase { // solhint-disable-line pay
         vm.store(address(this), ProxyUtils.DICTIONARY_SLOT, Formatter.toBytes32(dictionary_));
     }
 
-    fallback(bytes calldata) external payable returns(bytes memory) {
-        address funcImpl = implementations[msg.sig];
-        require(funcImpl != address(0), "Called implementation is not registered.");
-        (bool success, bytes memory data) = funcImpl.delegatecall(msg.data);
-        if (success) {
-            return data;
-        } else {
-            // vm.expectRevert needs this.
-            revert(DecodeErrorString.decodeRevertReasonAndPanicCode(data));
-        }
+    function _implementation() internal view override returns (address) {
+        return implementations[msg.sig];
     }
+
 }
 
 // 🌟 MC TEST for DevKit

test/std/functions/Clone.t.sol

@@ -9,7 +9,6 @@ import {Clone} from "mc-std/functions/Clone.sol";
 import {ProxyCreator} from "mc-std/functions/internal/ProxyCreator.sol";
 import {ForgeHelper} from "devkit/utils/ForgeHelper.sol";
 import {Dummy} from "test/utils/Dummy.sol";
-import {StateFuzzing} from "devkit/utils/test/StateFuzzing.sol";
 
 contract CloneTest is MCStateFuzzingTest {
     function setUp() public {

test/std/functions/Create.t.sol

@@ -9,7 +9,6 @@ import {Create} from "mc-std/functions/Create.sol";
 import {ProxyCreator} from "mc-std/functions/internal/ProxyCreator.sol";
 import {ForgeHelper} from "devkit/utils/ForgeHelper.sol";
 import {Dummy} from "test/utils/Dummy.sol";
-import {StateFuzzing} from "devkit/utils/test/StateFuzzing.sol";
 
 contract CreateTest is MCStateFuzzingTest {
     function setUp() public {

test/std/functions/GetFunctions.t.sol

@@ -12,7 +12,6 @@ import {ForgeHelper} from "devkit/utils/ForgeHelper.sol";
 import {Dummy} from "test/utils/Dummy.sol";
 import {DummyFunction} from "test/utils/DummyFunction.sol";
 import {DummyFacade} from "test/utils/DummyFacade.sol";
-import {StateFuzzing} from "devkit/utils/test/StateFuzzing.sol";
 
 contract GetFunctionsTest is MCStateFuzzingTest {
     function setUp() public {