diff --git a/test/large/ReservoirPriceOracleLarge.t.sol b/test/large/ReservoirPriceOracleLarge.t.sol index 6a80359..1a7cf37 100644 --- a/test/large/ReservoirPriceOracleLarge.t.sol +++ b/test/large/ReservoirPriceOracleLarge.t.sol @@ -34,6 +34,14 @@ contract ReservoirPriceOracleLargeTest is ReservoirPriceOracleTest { aTokenAAddress.code.length == 0 && aTokenBAddress.code.length == 0 && aTokenCAddress.code.length == 0 && aTokenCAddress.code.length == 0 ); + assumeNotPrecompile(aTokenAAddress); + assumeNotPrecompile(aTokenBAddress); + assumeNotPrecompile(aTokenCAddress); + assumeNotPrecompile(aTokenDAddress); + assumeNotZeroAddress(aTokenAAddress); + assumeNotZeroAddress(aTokenBAddress); + assumeNotZeroAddress(aTokenCAddress); + assumeNotZeroAddress(aTokenDAddress); uint256 lPrice1 = bound(aPrice1, 1e12, 1e24); uint256 lPrice2 = bound(aPrice2, 1e12, 1e24); uint256 lPrice3 = bound(aPrice3, 1e12, 1e24); diff --git a/test/unit/ReservoirPriceOracle.t.sol b/test/unit/ReservoirPriceOracle.t.sol index 274d732..eac916c 100644 --- a/test/unit/ReservoirPriceOracle.t.sol +++ b/test/unit/ReservoirPriceOracle.t.sol @@ -260,6 +260,10 @@ contract ReservoirPriceOracleTest is BaseTest { ) external { // assume vm.assume(aTokenAAddress.code.length == 0 && aTokenBAddress.code.length == 0); + assumeNotPrecompile(aTokenAAddress); + assumeNotPrecompile(aTokenBAddress); + assumeNotZeroAddress(aTokenAAddress); + assumeNotZeroAddress(aTokenBAddress); uint256 lPrice = bound(aPrice, 1, 1e36); uint256 lAmtIn = bound(aAmtIn, 0, 1_000_000_000); uint256 lTokenADecimal = bound(aTokenADecimal, 0, 18); @@ -306,6 +310,12 @@ contract ReservoirPriceOracleTest is BaseTest { ) external { // assume vm.assume(aTokenAAddress.code.length == 0 && aTokenBAddress.code.length == 0 && aTokenCAddress.code.length == 0); + assumeNotPrecompile(aTokenAAddress); + assumeNotPrecompile(aTokenBAddress); + assumeNotPrecompile(aTokenCAddress); + assumeNotZeroAddress(aTokenAAddress); + assumeNotZeroAddress(aTokenBAddress); + assumeNotZeroAddress(aTokenCAddress); uint256 lPrice1 = bound(aPrice1, 1e9, 1e25); // need to bound price within this range as a price below this will go to zero as during the mul and div of prices uint256 lPrice2 = bound(aPrice2, 1e9, 1e25); uint256 lAmtIn = bound(aAmtIn, 0, 1_000_000_000);