@bananapus/core-v6 0.0.36 → 0.0.38

package/test/units/static/JBTerminalStore/TestCurrentSurplusOf.sol

@@ -1,463 +0,0 @@
-// SPDX-License-Identifier: MIT
-pragma solidity 0.8.28;
-
-import {IJBController} from "../../../../src/interfaces/IJBController.sol";
-import {IJBDirectory} from "../../../../src/interfaces/IJBDirectory.sol";
-import {IJBFundAccessLimits} from "../../../../src/interfaces/IJBFundAccessLimits.sol";
-import {IJBPrices} from "../../../../src/interfaces/IJBPrices.sol";
-import {IJBRulesetApprovalHook} from "../../../../src/interfaces/IJBRulesetApprovalHook.sol";
-import {IJBRulesets} from "../../../../src/interfaces/IJBRulesets.sol";
-import {IJBTerminal} from "../../../../src/interfaces/IJBTerminal.sol";
-import {IJBToken} from "../../../../src/interfaces/IJBToken.sol";
-import {JBConstants} from "../../../../src/libraries/JBConstants.sol";
-import {JBRulesetMetadataResolver} from "../../../../src/libraries/JBRulesetMetadataResolver.sol";
-import {JBAccountingContext} from "../../../../src/structs/JBAccountingContext.sol";
-import {JBCurrencyAmount} from "../../../../src/structs/JBCurrencyAmount.sol";
-import {JBRuleset} from "../../../../src/structs/JBRuleset.sol";
-import {JBRulesetMetadata} from "../../../../src/structs/JBRulesetMetadata.sol";
-import {JBTerminalStoreSetup} from "./JBTerminalStoreSetup.sol";
-
-contract TestCurrentSurplusOf_Local is JBTerminalStoreSetup {
-    using JBRulesetMetadataResolver for JBRulesetMetadata;
-
-    uint256 _projectId = 1;
-    uint256 _balance = 1e18;
-
-    // Mocks
-    IJBTerminal _terminal = IJBTerminal(makeAddr("terminal"));
-    IJBToken _token = IJBToken(makeAddr("token"));
-    IJBController _controller = IJBController(makeAddr("controller"));
-    IJBFundAccessLimits _accessLimits = IJBFundAccessLimits(makeAddr("funds"));
-
-    uint32 _currency = uint32(uint160(address(_token)));
-    address _nativeAddress = JBConstants.NATIVE_TOKEN;
-    uint32 _nativeCurrency = uint32(uint160(JBConstants.NATIVE_TOKEN));
-
-    function setUp() public {
-        super.terminalStoreSetup();
-    }
-
-    /// @notice Helper to register an accounting context with the store (pranks as the terminal).
-    function _registerContext(JBAccountingContext memory ctx) internal {
-        JBAccountingContext[] memory ctxs = new JBAccountingContext[](1);
-        ctxs[0] = ctx;
-        vm.prank(address(_terminal));
-        _store.recordAccountingContextOf(_projectId, ctxs);
-    }
-
-    /// @notice Helper to call currentSurplusOf with the new multi-terminal signature.
-    function _currentSurplusOf(uint256 decimals, uint256 currency) internal view returns (uint256) {
-        IJBTerminal[] memory terminals = new IJBTerminal[](1);
-        terminals[0] = _terminal;
-        return _store.currentSurplusOf(_projectId, terminals, new address[](0), decimals, currency);
-    }
-
-    modifier whenProjectHasBalance() {
-        // Find the storage slot
-        bytes32 balanceOfSlot = keccak256(abi.encode(address(_terminal), uint256(0)));
-        bytes32 projectSlot = keccak256(abi.encode(_projectId, uint256(balanceOfSlot)));
-        bytes32 slot = keccak256(abi.encode(address(_token), uint256(projectSlot)));
-
-        bytes32 balanceBytes = bytes32(_balance);
-
-        // Set balance
-        vm.store(address(_store), slot, balanceBytes);
-
-        // Ensure balance is set correctly
-        uint256 _balanceCallReturn = _store.balanceOf(address(_terminal), _projectId, address(_token));
-        assertEq(_balanceCallReturn, _balance);
-        _;
-    }
-
-    function test_GivenSurplusRequiresDecimalAdjustmentAndAccountingCurrencyMatchesTargetCurrency()
-        external
-        whenProjectHasBalance
-    {
-        // it will adjust surplus to target decimals without conversion
-        JBAccountingContext[] memory _contexts = new JBAccountingContext[](1);
-
-        _contexts[0] = JBAccountingContext({token: address(_token), decimals: 18, currency: _currency});
-        _registerContext(_contexts[0]);
-
-        // JBRulesets calldata
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-
-        // mock call to JBRulesets currentOf
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // mock call to JBDirectory controllerOf
-        mockExpect(
-            address(directory),
-            abi.encodeCall(IJBDirectory.controllerOf, (_projectId)),
-            abi.encode(address(_controller))
-        );
-
-        // mock call to controller FUND_ACCESS_LIMITS
-        mockExpect(
-            address(_controller), abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ()), abi.encode(_accessLimits)
-        );
-
-        JBCurrencyAmount[] memory _payoutLimits = new JBCurrencyAmount[](1);
-        _payoutLimits[0] = JBCurrencyAmount({amount: 1e17, currency: _currency});
-
-        // mock call to fundAccessLimits payoutLimitsOf
-        bytes memory _payoutLimitsCall = abi.encodeCall(
-            IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal), address(_token))
-        );
-        bytes memory _payoutLimitsReturn = abi.encode(_payoutLimits);
-        mockExpect(address(_accessLimits), _payoutLimitsCall, _payoutLimitsReturn);
-
-        uint256 currentSurplus = _currentSurplusOf(6, _currency);
-
-        // assert correct calcs
-        uint256 expectedSurplus = (1e18 - 1e17) / 10 ** (18 - 6);
-        assertEq(expectedSurplus, currentSurplus);
-    }
-
-    function test_GivenSurplusRequiresDecimalAdjustmentAndAccountingCurrencyDoesNotMatchTargetCurrency()
-        external
-        whenProjectHasBalance
-    {
-        // it will convert surplus to target currency with decimal adjustment
-        JBAccountingContext[] memory _contexts = new JBAccountingContext[](1);
-
-        _contexts[0] = JBAccountingContext({token: address(_token), decimals: 18, currency: _currency});
-        _registerContext(_contexts[0]);
-
-        // JBRulesets calldata
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-
-        // mock call to JBRulesets currentOf
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // mock call to JBDirectory controllerOf
-        mockExpect(
-            address(directory),
-            abi.encodeCall(IJBDirectory.controllerOf, (_projectId)),
-            abi.encode(address(_controller))
-        );
-
-        // mock call to controller FUND_ACCESS_LIMITS
-        bytes memory _accessLimitsCall = abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ());
-        bytes memory _accessLimitsReturn = abi.encode(_accessLimits);
-        mockExpect(address(_controller), _accessLimitsCall, _accessLimitsReturn);
-
-        JBCurrencyAmount[] memory _payoutLimits = new JBCurrencyAmount[](1);
-        _payoutLimits[0] = JBCurrencyAmount({amount: 1e17, currency: _currency});
-
-        // mock call to fundAccessLimits payoutLimitsOf
-        bytes memory _payoutLimitsCall = abi.encodeCall(
-            IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal), address(_token))
-        );
-        bytes memory _payoutLimitsReturn = abi.encode(_payoutLimits);
-        mockExpect(address(_accessLimits), _payoutLimitsCall, _payoutLimitsReturn);
-
-        // mock call to JBPrices pricePerUnitOf
-        bytes memory _pricePerCall =
-            abi.encodeCall(IJBPrices.pricePerUnitOf, (_projectId, _currency, _nativeCurrency, 18));
-        bytes memory _pricePerReturn = abi.encode(uint256(1));
-        mockExpect(address(prices), _pricePerCall, _pricePerReturn);
-
-        uint256 currentSurplus = _currentSurplusOf(6, _nativeCurrency);
-
-        // assert correct calcs
-        uint256 expectedSurplus = ((1e18 - 1e17) * 1e18) / 10 ** (18 - 6);
-        assertEq(expectedSurplus, currentSurplus);
-    }
-
-    function test_GivenAccountingCurrencyMatchesTargetCurrency() external whenProjectHasBalance {
-        // it will return standard surplus
-
-        JBAccountingContext[] memory _contexts = new JBAccountingContext[](1);
-
-        _contexts[0] = JBAccountingContext({token: address(_token), decimals: 18, currency: _currency});
-        _registerContext(_contexts[0]);
-
-        // JBRulesets calldata
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-
-        // mock call to JBRulesets currentOf
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // mock call to JBDirectory controllerOf
-        mockExpect(
-            address(directory),
-            abi.encodeCall(IJBDirectory.controllerOf, (_projectId)),
-            abi.encode(address(_controller))
-        );
-
-        // mock call to controller FUND_ACCESS_LIMITS
-        mockExpect(
-            address(_controller), abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ()), abi.encode(_accessLimits)
-        );
-
-        JBCurrencyAmount[] memory _payoutLimits = new JBCurrencyAmount[](1);
-        _payoutLimits[0] = JBCurrencyAmount({amount: 1e17, currency: _currency});
-
-        // mock call to fundAccessLimits payoutLimitsOf
-        bytes memory _payoutLimitsCall = abi.encodeCall(
-            IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal), address(_token))
-        );
-        bytes memory _payoutLimitsReturn = abi.encode(_payoutLimits);
-        mockExpect(address(_accessLimits), _payoutLimitsCall, _payoutLimitsReturn);
-
-        uint256 currentSurplus = _currentSurplusOf(18, _currency);
-
-        // assert correct calcs
-        uint256 expectedSurplus = 1e18 - 1e17;
-        assertEq(expectedSurplus, currentSurplus);
-    }
-
-    function test_GivenAccountingCurrencyDoesNotMatchTargetCurrency() external whenProjectHasBalance {
-        // it will convert surplus to target currency without decimal adjustment
-        JBAccountingContext[] memory _contexts = new JBAccountingContext[](1);
-
-        _contexts[0] = JBAccountingContext({token: address(_token), decimals: 18, currency: _currency});
-        _registerContext(_contexts[0]);
-
-        // JBRulesets calldata
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-
-        // mock call to JBRulesets currentOf
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // mock call to JBDirectory controllerOf
-        mockExpect(
-            address(directory),
-            abi.encodeCall(IJBDirectory.controllerOf, (_projectId)),
-            abi.encode(address(_controller))
-        );
-
-        // mock call to controller FUND_ACCESS_LIMITS
-        bytes memory _accessLimitsCall = abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ());
-        bytes memory _accessLimitsReturn = abi.encode(_accessLimits);
-        mockExpect(address(_controller), _accessLimitsCall, _accessLimitsReturn);
-
-        JBCurrencyAmount[] memory _payoutLimits = new JBCurrencyAmount[](1);
-        _payoutLimits[0] = JBCurrencyAmount({amount: 1e17, currency: _currency});
-
-        // mock call to fundAccessLimits payoutLimitsOf
-        bytes memory _payoutLimitsCall = abi.encodeCall(
-            IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal), address(_token))
-        );
-        bytes memory _payoutLimitsReturn = abi.encode(_payoutLimits);
-        mockExpect(address(_accessLimits), _payoutLimitsCall, _payoutLimitsReturn);
-
-        // mock call to JBPrices pricePerUnitOf
-        bytes memory _pricePerCall =
-            abi.encodeCall(IJBPrices.pricePerUnitOf, (_projectId, _currency, _nativeCurrency, 18));
-        bytes memory _pricePerReturn = abi.encode(uint256(1));
-        mockExpect(address(prices), _pricePerCall, _pricePerReturn);
-
-        uint256 currentSurplus = _currentSurplusOf(18, _nativeCurrency);
-
-        // assert correct calcs
-        uint256 expectedSurplus = (1e18 - 1e17) * 1e18;
-        assertEq(expectedSurplus, currentSurplus);
-    }
-
-    function test_GivenPayoutLimitCurrencyMatchesTargetCurrency() external whenProjectHasBalance {
-        // it will return standard surplus
-
-        JBAccountingContext[] memory _contexts = new JBAccountingContext[](1);
-
-        _contexts[0] = JBAccountingContext({token: address(_token), decimals: 18, currency: _currency});
-        _registerContext(_contexts[0]);
-
-        // JBRulesets calldata
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-
-        // mock call to JBRulesets currentOf
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // mock call to JBDirectory controllerOf
-        mockExpect(
-            address(directory),
-            abi.encodeCall(IJBDirectory.controllerOf, (_projectId)),
-            abi.encode(address(_controller))
-        );
-
-        // mock call to controller FUND_ACCESS_LIMITS
-        bytes memory _accessLimitsCall = abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ());
-        bytes memory _accessLimitsReturn = abi.encode(_accessLimits);
-        mockExpect(address(_controller), _accessLimitsCall, _accessLimitsReturn);
-
-        JBCurrencyAmount[] memory _payoutLimits = new JBCurrencyAmount[](1);
-        _payoutLimits[0] = JBCurrencyAmount({amount: 1e17, currency: _currency});
-
-        // mock call to fundAccessLimits payoutLimitsOf
-        bytes memory _payoutLimitsCall = abi.encodeCall(
-            IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal), address(_token))
-        );
-        bytes memory _payoutLimitsReturn = abi.encode(_payoutLimits);
-        mockExpect(address(_accessLimits), _payoutLimitsCall, _payoutLimitsReturn);
-
-        uint256 currentSurplus = _currentSurplusOf(18, _currency);
-
-        // assert correct calcs
-        uint256 expectedSurplus = (1e18 - 1e17);
-        assertEq(expectedSurplus, currentSurplus);
-    }
-
-    function test_GivenPayoutLimitCurrencyDoesNotMatchTargetCurrency() external whenProjectHasBalance {
-        // it will convert payout limit target currency without decimal adjustment
-
-        JBAccountingContext[] memory _contexts = new JBAccountingContext[](1);
-
-        _contexts[0] = JBAccountingContext({token: address(_token), decimals: 18, currency: _nativeCurrency});
-        _registerContext(_contexts[0]);
-
-        // JBRulesets calldata
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-
-        // mock call to JBRulesets currentOf
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // mock call to JBDirectory controllerOf
-        mockExpect(
-            address(directory),
-            abi.encodeCall(IJBDirectory.controllerOf, (_projectId)),
-            abi.encode(address(_controller))
-        );
-
-        // mock call to controller FUND_ACCESS_LIMITS
-        bytes memory _accessLimitsCall = abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ());
-        bytes memory _accessLimitsReturn = abi.encode(_accessLimits);
-        mockExpect(address(_controller), _accessLimitsCall, _accessLimitsReturn);
-
-        JBCurrencyAmount[] memory _payoutLimits = new JBCurrencyAmount[](1);
-        _payoutLimits[0] = JBCurrencyAmount({amount: 1e17, currency: _currency});
-
-        // mock call to fundAccessLimits payoutLimitsOf
-        bytes memory _payoutLimitsCall = abi.encodeCall(
-            IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal), address(_token))
-        );
-        bytes memory _payoutLimitsReturn = abi.encode(_payoutLimits);
-        mockExpect(address(_accessLimits), _payoutLimitsCall, _payoutLimitsReturn);
-
-        // mock call to JBPrices pricePerUnitOf
-        bytes memory _pricePerCall =
-            abi.encodeCall(IJBPrices.pricePerUnitOf, (_projectId, _currency, _nativeCurrency, 18));
-        bytes memory _pricePerReturn = abi.encode(uint256(1e18));
-        mockExpect(address(prices), _pricePerCall, _pricePerReturn);
-
-        uint256 currentSurplus = _currentSurplusOf(18, _nativeCurrency);
-
-        // assert correct calcs
-        uint256 expectedSurplus = 1e18 - 1e17;
-        assertEq(expectedSurplus, currentSurplus);
-    }
-
-    function test_GivenCumulativeBalanceNotGreaterThanCumulativePayoutLimit() external whenProjectHasBalance {
-        // it will return zero
-
-        JBAccountingContext[] memory _contexts = new JBAccountingContext[](1);
-
-        _contexts[0] = JBAccountingContext({token: address(_token), decimals: 18, currency: _currency});
-        _registerContext(_contexts[0]);
-
-        // JBRulesets calldata
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-
-        // mock call to JBRulesets currentOf
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // mock call to JBDirectory controllerOf
-        mockExpect(
-            address(directory),
-            abi.encodeCall(IJBDirectory.controllerOf, (_projectId)),
-            abi.encode(address(_controller))
-        );
-
-        // mock call to controller FUND_ACCESS_LIMITS
-        bytes memory _accessLimitsCall = abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ());
-        bytes memory _accessLimitsReturn = abi.encode(_accessLimits);
-        mockExpect(address(_controller), _accessLimitsCall, _accessLimitsReturn);
-
-        JBCurrencyAmount[] memory _payoutLimits = new JBCurrencyAmount[](1);
-        _payoutLimits[0] = JBCurrencyAmount({amount: 1e19, currency: _currency});
-
-        // mock call to fundAccessLimits payoutLimitsOf
-        bytes memory _payoutLimitsCall = abi.encodeCall(
-            IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal), address(_token))
-        );
-        bytes memory _payoutLimitsReturn = abi.encode(_payoutLimits);
-        mockExpect(address(_accessLimits), _payoutLimitsCall, _payoutLimitsReturn);
-
-        uint256 currentSurplus = _currentSurplusOf(18, _currency);
-
-        // assert correct calcs
-        uint256 expectedSurplus = 0;
-        assertEq(expectedSurplus, currentSurplus);
-    }
-}

package/test/units/static/JBTerminalStore/TestCurrentTotalSurplusOf.sol

@@ -1,135 +0,0 @@
-// SPDX-License-Identifier: MIT
-pragma solidity 0.8.28;
-
-import {IJBController} from "../../../../src/interfaces/IJBController.sol";
-import {IJBDirectory} from "../../../../src/interfaces/IJBDirectory.sol";
-import {IJBFundAccessLimits} from "../../../../src/interfaces/IJBFundAccessLimits.sol";
-import {IJBRulesetApprovalHook} from "../../../../src/interfaces/IJBRulesetApprovalHook.sol";
-import {IJBRulesets} from "../../../../src/interfaces/IJBRulesets.sol";
-import {IJBTerminal} from "../../../../src/interfaces/IJBTerminal.sol";
-import {JBAccountingContext} from "../../../../src/structs/JBAccountingContext.sol";
-import {JBCurrencyAmount} from "../../../../src/structs/JBCurrencyAmount.sol";
-import {JBRuleset} from "../../../../src/structs/JBRuleset.sol";
-import {JBTerminalStoreSetup} from "./JBTerminalStoreSetup.sol";
-
-contract TestCurrentTotalSurplusOf_Local is JBTerminalStoreSetup {
-    uint256 _projectId = 1;
-    uint256 _decimals = 18;
-
-    // Mocks
-    IJBTerminal _terminal1 = IJBTerminal(makeAddr("terminal1"));
-    IJBTerminal _terminal2 = IJBTerminal(makeAddr("terminal2"));
-    address _token = makeAddr("token");
-    IJBController _controller = IJBController(makeAddr("controller"));
-    IJBFundAccessLimits _accessLimits = IJBFundAccessLimits(makeAddr("funds"));
-
-    // forge-lint: disable-next-line(unsafe-typecast)
-    uint32 _currency = uint32(uint160(_token));
-
-    function setUp() public {
-        super.terminalStoreSetup();
-    }
-
-    /// @notice Helper to set balance for a terminal/project/token via vm.store.
-    function _setBalance(address terminal, uint256 projectId, address token, uint256 balance) internal {
-        bytes32 balanceOfSlot = keccak256(abi.encode(terminal, uint256(0)));
-        bytes32 projectSlot = keccak256(abi.encode(projectId, uint256(balanceOfSlot)));
-        bytes32 slot = keccak256(abi.encode(token, uint256(projectSlot)));
-        vm.store(address(_store), slot, bytes32(balance));
-    }
-
-    /// @notice Helper to register an accounting context with the store (pranks as the terminal).
-    function _registerContext(address terminal, JBAccountingContext memory ctx) internal {
-        JBAccountingContext[] memory ctxs = new JBAccountingContext[](1);
-        ctxs[0] = ctx;
-        vm.prank(terminal);
-        _store.recordAccountingContextOf(_projectId, ctxs);
-    }
-
-    function test_WhenTerminalsAreConfiguredInJBDirectory() external {
-        // it will return the cumulative surplus
-
-        // Set up terminals in directory
-        IJBTerminal[] memory _terminals = new IJBTerminal[](2);
-        _terminals[0] = _terminal1;
-        _terminals[1] = _terminal2;
-
-        mockExpect(address(directory), abi.encodeCall(IJBDirectory.terminalsOf, (_projectId)), abi.encode(_terminals));
-
-        // Set up accounting contexts for each terminal
-        JBAccountingContext memory ctx = JBAccountingContext({token: _token, decimals: 18, currency: _currency});
-        _registerContext(address(_terminal1), ctx);
-        _registerContext(address(_terminal2), ctx);
-
-        // Set balances: terminal1 = 1e18, terminal2 = 2e18
-        _setBalance(address(_terminal1), _projectId, _token, 1e18);
-        _setBalance(address(_terminal2), _projectId, _token, 2e18);
-
-        // Mock ruleset
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        // Mock controller and fund access limits (zero payout limits = all balance is surplus)
-        mockExpect(address(directory), abi.encodeCall(IJBDirectory.controllerOf, (_projectId)), abi.encode(_controller));
-        mockExpect(
-            address(_controller), abi.encodeCall(IJBController.FUND_ACCESS_LIMITS, ()), abi.encode(_accessLimits)
-        );
-
-        JBCurrencyAmount[] memory _emptyLimits = new JBCurrencyAmount[](0);
-        // Mock payoutLimitsOf for terminal1
-        mockExpect(
-            address(_accessLimits),
-            abi.encodeCall(
-                IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal1), _token)
-            ),
-            abi.encode(_emptyLimits)
-        );
-        // Mock payoutLimitsOf for terminal2
-        mockExpect(
-            address(_accessLimits),
-            abi.encodeCall(
-                IJBFundAccessLimits.payoutLimitsOf, (_projectId, block.timestamp, address(_terminal2), _token)
-            ),
-            abi.encode(_emptyLimits)
-        );
-
-        uint256 sum = _store.currentTotalSurplusOf(_projectId, _decimals, _currency);
-        assertEq(3e18, sum);
-    }
-
-    function test_WhenTerminalsAreNotConfiguredInJBDirectory() external {
-        // it will return zero
-
-        IJBTerminal[] memory _terminals = new IJBTerminal[](0);
-
-        // mock call to JBDirectory terminalsOf
-        mockExpect(address(directory), abi.encodeCall(IJBDirectory.terminalsOf, (_projectId)), abi.encode(_terminals));
-
-        // Mock ruleset (needed by currentSurplusOf)
-        JBRuleset memory _returnedRuleset = JBRuleset({
-            cycleNumber: uint48(block.timestamp),
-            id: uint48(block.timestamp),
-            basedOnId: 0,
-            start: uint48(block.timestamp),
-            duration: uint32(block.timestamp + 1000),
-            weight: 1e18,
-            weightCutPercent: 0,
-            approvalHook: IJBRulesetApprovalHook(address(0)),
-            metadata: 0
-        });
-        mockExpect(address(rulesets), abi.encodeCall(IJBRulesets.currentOf, (_projectId)), abi.encode(_returnedRuleset));
-
-        uint256 sum = _store.currentTotalSurplusOf(_projectId, _decimals, _currency);
-        assertEq(0, sum);
-    }
-}