|
| 1 | +// SPDX-License-Identifier: UNLICENSED |
| 2 | +pragma solidity ^0.8.23; |
| 3 | + |
| 4 | +import {ERC20} from "@openzeppelin/contracts/token/ERC20/ERC20.sol"; |
| 5 | + |
| 6 | +import {ICreditFacadeV3, MultiCall} from "@gearbox-protocol/core-v3/contracts/interfaces/ICreditFacadeV3.sol"; |
| 7 | +import {ICreditManagerV3} from "@gearbox-protocol/core-v3/contracts/interfaces/ICreditManagerV3.sol"; |
| 8 | + |
| 9 | +import {ConstantPriceFeed} from "@gearbox-protocol/oracles-v3/contracts/oracles/ConstantPriceFeed.sol"; |
| 10 | + |
| 11 | +import {AttachTestBase} from "@gearbox-protocol/permissionless/contracts/test/suite/AttachTestBase.sol"; |
| 12 | + |
| 13 | +abstract contract IntegrationsAttachTestBase is AttachTestBase { |
| 14 | + address zeroPriceFeed; |
| 15 | + address onePriceFeed; |
| 16 | + |
| 17 | + address underlying; |
| 18 | + address pool; |
| 19 | + address creditManager; |
| 20 | + address creditFacade; |
| 21 | + |
| 22 | + address user; |
| 23 | + address creditAccount; |
| 24 | + |
| 25 | + function _setUp() internal { |
| 26 | + // NOTE: even though we compile our contracts under Shanghai EVM version, |
| 27 | + // more recent one might be needed to interact with third-party contracts |
| 28 | + vm.setEvmVersion("osaka"); |
| 29 | + |
| 30 | + _attachCore(); |
| 31 | + |
| 32 | + zeroPriceFeed = priceFeedStore.zeroPriceFeed(); |
| 33 | + if (bytecodeRepository.getAllowedBytecodeHash("PRICE_FEED::CONSTANT", 3_10) == 0) { |
| 34 | + // NOTE: this would fail on deployment if we bump `ConstantPriceFeed` to v3.1.1 in the oracles repo, |
| 35 | + // but we assume that in this case v3.1.0 would already be uploaded to the bytecode repository |
| 36 | + _uploadContract("PRICE_FEED::CONSTANT", 3_10, type(ConstantPriceFeed).creationCode); |
| 37 | + } |
| 38 | + onePriceFeed = _deploy("PRICE_FEED::CONSTANT", 3_10, abi.encode(1e8, "$1 price feed")); |
| 39 | + _addPriceFeed(onePriceFeed, 0, "$1 price feed"); |
| 40 | + |
| 41 | + _attachMarketConfigurator(); |
| 42 | + |
| 43 | + underlying = address(new ERC20("Mock Token", "MOCK")); |
| 44 | + _allowPriceFeed(underlying, onePriceFeed); |
| 45 | + pool = _createMockMarket(underlying, onePriceFeed); |
| 46 | + creditManager = _createMockCreditSuite({pool: pool, minDebt: 1e18, maxDebt: 1e18, debtLimit: 0}); |
| 47 | + creditFacade = ICreditManagerV3(creditManager).creditFacade(); |
| 48 | + |
| 49 | + user = makeAddr("user"); |
| 50 | + creditAccount = ICreditFacadeV3(creditFacade).openCreditAccount(user, new MultiCall[](0), 0); |
| 51 | + } |
| 52 | + |
| 53 | + // ----------------------- // |
| 54 | + // CONFIGURATION FUNCTIONS // |
| 55 | + // ----------------------- // |
| 56 | + |
| 57 | + function _addToken(address token) internal { |
| 58 | + _allowPriceFeed(token, zeroPriceFeed); |
| 59 | + _addToken( |
| 60 | + pool, |
| 61 | + TokenConfig({ |
| 62 | + token: token, priceFeed: zeroPriceFeed, reservePriceFeed: address(0), quotaLimit: 0, quotaRate: 0 |
| 63 | + }) |
| 64 | + ); |
| 65 | + _addCollateralToken(creditManager, token, 0); |
| 66 | + } |
| 67 | + |
| 68 | + function _getAdapterFor(address targetContract) internal view returns (address) { |
| 69 | + return ICreditManagerV3(creditManager).contractToAdapter(targetContract); |
| 70 | + } |
| 71 | + |
| 72 | + // -------------- // |
| 73 | + // USER FUNCTIONS // |
| 74 | + // -------------- // |
| 75 | + |
| 76 | + function _multicall(MultiCall[] memory calls) internal { |
| 77 | + vm.prank(user); |
| 78 | + ICreditFacadeV3(creditFacade).multicall(creditAccount, calls); |
| 79 | + } |
| 80 | + |
| 81 | + function _multicall(MultiCall memory call0) internal { |
| 82 | + MultiCall[] memory calls = new MultiCall[](1); |
| 83 | + calls[0] = call0; |
| 84 | + _multicall(calls); |
| 85 | + } |
| 86 | + |
| 87 | + function _multicall(MultiCall memory call0, MultiCall memory call1) internal { |
| 88 | + MultiCall[] memory calls = new MultiCall[](2); |
| 89 | + calls[0] = call0; |
| 90 | + calls[1] = call1; |
| 91 | + _multicall(calls); |
| 92 | + } |
| 93 | + |
| 94 | + function _multicall(MultiCall memory call0, MultiCall memory call1, MultiCall memory call2) internal { |
| 95 | + MultiCall[] memory calls = new MultiCall[](3); |
| 96 | + calls[0] = call0; |
| 97 | + calls[1] = call1; |
| 98 | + calls[2] = call2; |
| 99 | + _multicall(calls); |
| 100 | + } |
| 101 | + |
| 102 | + function _multicall(MultiCall memory call0, MultiCall memory call1, MultiCall memory call2, MultiCall memory call3) |
| 103 | + internal |
| 104 | + { |
| 105 | + MultiCall[] memory calls = new MultiCall[](4); |
| 106 | + calls[0] = call0; |
| 107 | + calls[1] = call1; |
| 108 | + calls[2] = call2; |
| 109 | + calls[3] = call3; |
| 110 | + _multicall(calls); |
| 111 | + } |
| 112 | + |
| 113 | + function _multicall( |
| 114 | + MultiCall memory call0, |
| 115 | + MultiCall memory call1, |
| 116 | + MultiCall memory call2, |
| 117 | + MultiCall memory call3, |
| 118 | + MultiCall memory call4 |
| 119 | + ) internal { |
| 120 | + MultiCall[] memory calls = new MultiCall[](5); |
| 121 | + calls[0] = call0; |
| 122 | + calls[1] = call1; |
| 123 | + calls[2] = call2; |
| 124 | + calls[3] = call3; |
| 125 | + calls[4] = call4; |
| 126 | + _multicall(calls); |
| 127 | + } |
| 128 | +} |
0 commit comments