Skip to content

Commit 4223cbc

Browse files
committed
adjust lest tests
1 parent 021331b commit 4223cbc

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

test/RescueStrategyTest.sol

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ import {IAllowanceTransfer} from "../src/interfaces/IAllowanceTransfer.sol";
1010
import {EnumerableSet} from "openzeppelin-contracts/utils/structs/EnumerableSet.sol";
1111
import {IEVC} from "ethereum-vault-connector/interfaces/IEthereumVaultConnector.sol";
1212
import {RescueStrategy} from "../src/RescueStrategy.sol";
13-
import {EulerEarnVaultLens as EarnIndexerLens} from "../lib/euler-data-lenses/src/EulerEarnLens.sol";
13+
import {EulerEarnVaultLens as EarnIndexerLens, EulerEarnVaultInfoFull} from "../lib/euler-data-lenses/src/EulerEarnLens.sol";
1414
import "forge-std/Test.sol";
1515

1616
contract RescuePOC is Test {
@@ -27,7 +27,7 @@ contract RescuePOC is Test {
2727
IEulerEarn vault;
2828
IEulerEarn otherVault;
2929

30-
address indexerLens;
30+
EarnIndexerLens indexerLens;
3131

3232
string FORK_RPC_URL = vm.envOr("FORK_RPC_URL_MAINNET", string(""));
3333

@@ -55,7 +55,7 @@ contract RescuePOC is Test {
5555
vault.asset(), address(vault), type(uint160).max, type(uint48).max
5656
);
5757

58-
indexerLens = address(new EarnIndexerLens());
58+
indexerLens = new EarnIndexerLens();
5959
}
6060

6161
function testRescue_assertRescueMode() public {
@@ -304,12 +304,15 @@ contract RescuePOC is Test {
304304

305305
function testRescue_callLenses() external {
306306
_installRescueStrategy();
307+
// lens calls don't revert
307308

309+
// onchain lens
308310
(bool success, bytes memory data) = EARN_LENS.call(abi.encodeWithSignature("getVaultInfoFull(address)", address(vault)));
309311
assertTrue(success && data.length > 0);
310312

311-
(success, data) = indexerLens.call(abi.encodeWithSignature("getVaultInfoFull(address)", address(vault)));
312-
assertTrue(success && data.length > 0);
313+
// lens used in the indexer by setting the `code` in eth_call
314+
EulerEarnVaultInfoFull memory lensData = indexerLens.getVaultInfoFull(address(vault));
315+
assertEq(lensData.vault, address(vault));
313316
}
314317

315318
function testRescue_maxWithdrawView() external {

0 commit comments

Comments
 (0)