@@ -10,7 +10,7 @@ import {IAllowanceTransfer} from "../src/interfaces/IAllowanceTransfer.sol";
1010import {EnumerableSet} from "openzeppelin-contracts/utils/structs/EnumerableSet.sol " ;
1111import {IEVC} from "ethereum-vault-connector/interfaces/IEthereumVaultConnector.sol " ;
1212import {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 " ;
1414import "forge-std/Test.sol " ;
1515
1616contract 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