-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
(test): Add Functions Gas tests (#10657)
- Loading branch information
1 parent
30dfa21
commit a35a7e1
Showing
11 changed files
with
734 additions
and
281 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
199 changes: 199 additions & 0 deletions
199
contracts/src/v0.8/functions/tests/v1_0_0/FunctionsBilling.t.sol
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,199 @@ | ||
| // SPDX-License-Identifier: MIT | ||
| pragma solidity ^0.8.19; | ||
|
|
||
| import {FunctionsCoordinator} from "../../dev/v1_0_0/FunctionsCoordinator.sol"; | ||
| import {FunctionsBilling} from "../../dev/v1_0_0/FunctionsBilling.sol"; | ||
| import {FunctionsRequest} from "../../dev/v1_0_0/libraries/FunctionsRequest.sol"; | ||
|
|
||
| import {FunctionsRouterSetup, FunctionsSubscriptionSetup, FunctionsMultipleFulfillmentsSetup} from "./Setup.t.sol"; | ||
|
|
||
| /// @notice #constructor | ||
| contract FunctionsBilling_Constructor { | ||
|
|
||
| } | ||
|
|
||
| /// @notice #getConfig | ||
| contract FunctionsBilling_GetConfig is FunctionsRouterSetup { | ||
| function test_GetConfig_Success() public { | ||
| // Send as stranger | ||
| vm.stopPrank(); | ||
| vm.startPrank(STRANGER_ADDRESS); | ||
|
|
||
| FunctionsBilling.Config memory config = s_functionsCoordinator.getConfig(); | ||
| assertEq(config.feedStalenessSeconds, getCoordinatorConfig().feedStalenessSeconds); | ||
| assertEq(config.gasOverheadBeforeCallback, getCoordinatorConfig().gasOverheadBeforeCallback); | ||
| assertEq(config.gasOverheadAfterCallback, getCoordinatorConfig().gasOverheadAfterCallback); | ||
| assertEq(config.requestTimeoutSeconds, getCoordinatorConfig().requestTimeoutSeconds); | ||
| assertEq(config.donFee, getCoordinatorConfig().donFee); | ||
| assertEq(config.maxSupportedRequestDataVersion, getCoordinatorConfig().maxSupportedRequestDataVersion); | ||
| assertEq(config.fulfillmentGasPriceOverEstimationBP, getCoordinatorConfig().fulfillmentGasPriceOverEstimationBP); | ||
| assertEq(config.fallbackNativePerUnitLink, getCoordinatorConfig().fallbackNativePerUnitLink); | ||
| } | ||
| } | ||
|
|
||
| /// @notice #updateConfig | ||
| contract FunctionsBilling_UpdateConfig { | ||
|
|
||
| } | ||
|
|
||
| /// @notice #getDONFee | ||
| contract FunctionsBilling_GetDONFee { | ||
|
|
||
| } | ||
|
|
||
| /// @notice #getAdminFee | ||
| contract FunctionsBilling_GetAdminFee { | ||
|
|
||
| } | ||
|
|
||
| /// @notice #getWeiPerUnitLink | ||
| contract FunctionsBilling_GetWeiPerUnitLink { | ||
|
|
||
| } | ||
|
|
||
| /// @notice #_getJuelsPerGas | ||
| contract FunctionsBilling__GetJuelsPerGas { | ||
| // TODO: make contract internal function helper | ||
| } | ||
|
|
||
| /// @notice #estimateCost | ||
| contract FunctionsBilling_EstimateCost is FunctionsSubscriptionSetup { | ||
| function setUp() public virtual override { | ||
| FunctionsSubscriptionSetup.setUp(); | ||
|
|
||
| // Get cost estimate as stranger | ||
| vm.stopPrank(); | ||
| vm.startPrank(STRANGER_ADDRESS); | ||
| } | ||
|
|
||
| uint256 private constant REASONABLE_GAS_PRICE_CEILING = 1_000_000_000_000_000; // 1 million gwei | ||
|
|
||
| function test_EstimateCost_RevertsIfGasPriceAboveCeiling() public { | ||
| // Build minimal valid request data | ||
| string memory sourceCode = "return 'hello world';"; | ||
| FunctionsRequest.Request memory request; | ||
| FunctionsRequest.initializeRequest( | ||
| request, | ||
| FunctionsRequest.Location.Inline, | ||
| FunctionsRequest.CodeLanguage.JavaScript, | ||
| sourceCode | ||
| ); | ||
| bytes memory requestData = FunctionsRequest.encodeCBOR(request); | ||
|
|
||
| uint32 callbackGasLimit = 5_500; | ||
| uint256 gasPriceWei = REASONABLE_GAS_PRICE_CEILING + 1; | ||
|
|
||
| vm.expectRevert(FunctionsBilling.InvalidCalldata.selector); | ||
|
|
||
| s_functionsCoordinator.estimateCost(s_subscriptionId, requestData, callbackGasLimit, gasPriceWei); | ||
| } | ||
|
|
||
| function test_EstimateCost_Success() public { | ||
| // Build minimal valid request data | ||
| string memory sourceCode = "return 'hello world';"; | ||
| FunctionsRequest.Request memory request; | ||
| FunctionsRequest.initializeRequest( | ||
| request, | ||
| FunctionsRequest.Location.Inline, | ||
| FunctionsRequest.CodeLanguage.JavaScript, | ||
| sourceCode | ||
| ); | ||
| bytes memory requestData = FunctionsRequest.encodeCBOR(request); | ||
|
|
||
| uint32 callbackGasLimit = 5_500; | ||
| uint256 gasPriceWei = 1; | ||
|
|
||
| uint96 costEstimate = s_functionsCoordinator.estimateCost( | ||
| s_subscriptionId, | ||
| requestData, | ||
| callbackGasLimit, | ||
| gasPriceWei | ||
| ); | ||
| uint96 expectedCostEstimate = 10873200; | ||
| assertEq(costEstimate, expectedCostEstimate); | ||
| } | ||
| } | ||
|
|
||
| /// @notice #_calculateCostEstimate | ||
| contract FunctionsBilling__CalculateCostEstimate { | ||
| // TODO: make contract internal function helper | ||
| } | ||
|
|
||
| /// @notice #_startBilling | ||
| contract FunctionsBilling__StartBilling { | ||
| // TODO: make contract internal function helper | ||
| } | ||
|
|
||
| /// @notice #_computeRequestId | ||
| contract FunctionsBilling__ComputeRequestId { | ||
| // TODO: make contract internal function helper | ||
| } | ||
|
|
||
| /// @notice #_fulfillAndBill | ||
| contract FunctionsBilling__FulfillAndBill { | ||
| // TODO: make contract internal function helper | ||
| } | ||
|
|
||
| /// @notice #deleteCommitment | ||
| contract FunctionsBilling_DeleteCommitment { | ||
|
|
||
| } | ||
|
|
||
| /// @notice #oracleWithdraw | ||
| contract FunctionsBilling_OracleWithdraw { | ||
|
|
||
| } | ||
|
|
||
| /// @notice #oracleWithdrawAll | ||
| contract FunctionsBilling_OracleWithdrawAll is FunctionsMultipleFulfillmentsSetup { | ||
| function setUp() public virtual override { | ||
| // Use no DON fee so that a transmitter has a balance of 0 | ||
| s_donFee = 0; | ||
|
|
||
| FunctionsMultipleFulfillmentsSetup.setUp(); | ||
| } | ||
|
|
||
| function test_OracleWithdrawAll_RevertIfNotOwner() public { | ||
| // Send as stranger | ||
| vm.stopPrank(); | ||
| vm.startPrank(STRANGER_ADDRESS); | ||
|
|
||
| vm.expectRevert("Only callable by owner"); | ||
| s_functionsCoordinator.oracleWithdrawAll(); | ||
| } | ||
|
|
||
| function test_OracleWithdrawAll_SuccessPaysTransmittersWithBalance() public { | ||
| uint256 transmitter1BalanceBefore = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_1); | ||
| assertEq(transmitter1BalanceBefore, 0); | ||
| uint256 transmitter2BalanceBefore = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_2); | ||
| assertEq(transmitter2BalanceBefore, 0); | ||
| uint256 transmitter3BalanceBefore = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_3); | ||
| assertEq(transmitter3BalanceBefore, 0); | ||
| uint256 transmitter4BalanceBefore = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_4); | ||
| assertEq(transmitter4BalanceBefore, 0); | ||
|
|
||
| s_functionsCoordinator.oracleWithdrawAll(); | ||
|
|
||
| uint96 expectedTransmitterBalance = s_fulfillmentCoordinatorBalance / 3; | ||
|
|
||
| uint256 transmitter1BalanceAfter = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_1); | ||
| assertEq(transmitter1BalanceAfter, expectedTransmitterBalance); | ||
| uint256 transmitter2BalanceAfter = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_2); | ||
| assertEq(transmitter2BalanceAfter, expectedTransmitterBalance); | ||
| uint256 transmitter3BalanceAfter = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_3); | ||
| assertEq(transmitter3BalanceAfter, expectedTransmitterBalance); | ||
| // Transmitter 4 has no balance | ||
| uint256 transmitter4BalanceAfter = s_linkToken.balanceOf(NOP_TRANSMITTER_ADDRESS_4); | ||
| assertEq(transmitter4BalanceAfter, 0); | ||
| } | ||
| } | ||
|
|
||
| /// @notice #_getTransmitters | ||
| contract FunctionsBilling__GetTransmitters { | ||
| // TODO: make contract internal function helper | ||
| } | ||
|
|
||
| /// @notice #_disperseFeePool | ||
| contract FunctionsBilling__DisperseFeePool { | ||
| // TODO: make contract internal function helper | ||
| } |
36 changes: 34 additions & 2 deletions
36
contracts/src/v0.8/functions/tests/v1_0_0/FunctionsClient.t.sol
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.