Merge pull request #49 from Gearbox-protocol/legacy-test-fix

fix: add exception for forbidden tokens in Convex eq test
Gearbox-protocol · Aug 29, 2023 · 5324a48 · 5324a48
2 parents cde10cd + 8dff06b
commit 5324a48
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/contracts/test/adapters/mainnet/Live_ConvexEquivalenceTest.t.sol b/contracts/test/adapters/mainnet/Live_ConvexEquivalenceTest.t.sol
@@ -81,6 +81,14 @@ contract Live_ConvexEquivalenceTest is DSTest, LiveEnvHelper {
 
     /// HELPER
 
+    function _isSupportedPool(ICreditFacade creditFacade, address basePoolAdapter) internal view returns (bool) {
+        address spt = IConvexV1BaseRewardPoolAdapter(basePoolAdapter).stakedPhantomToken();
+
+        uint256 tokenMask = creditFacade.creditManager().tokenMasksMap(spt);
+
+        return (tokenMask != 0) && (creditFacade.creditManager().forbiddenTokenMask() & tokenMask == 0);
+    }
+
     function _getTokensToTrack(address basePoolAdapter) internal view returns (Tokens[] memory tokensToTrack) {
         address targetContract = IConvexV1BaseRewardPoolAdapter(basePoolAdapter).targetContract();
 
@@ -304,6 +312,8 @@ contract Live_ConvexEquivalenceTest is DSTest, LiveEnvHelper {
 
             address basePoolAdapter = lts.getAdapter(address(creditFacade.creditManager()), convexPools[i]);
 
+            if (!_isSupportedPool(creditFacade, basePoolAdapter)) continue;
+
             address creditAccount = openCreditAccountWithUnderlying(
                 creditFacade,
                 IConvexV1BaseRewardPoolAdapter(basePoolAdapter).curveLPtoken(),