feat(contracts): tests for BoltEigenLayerMiddlewareV2

chainbound · Oct 23, 2024 · d70980c · d70980c
1 parent 333d8dd
commit d70980c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bolt-contracts/src/contracts/BoltEigenLayerMiddlewareV2.sol b/bolt-contracts/src/contracts/BoltEigenLayerMiddlewareV2.sol
@@ -31,7 +31,7 @@ import {StrategyManagerStorage} from "@eigenlayer/src/contracts/core/StrategyMan
 /// See https://docs.openzeppelin.com/contracts/4.x/upgradeable#storage_gaps
 /// To validate the storage layout, use the Openzeppelin Foundry Upgrades toolkit.
 /// You can also validate manually with forge: forge inspect <contract> storage-layout --pretty
-contract BoltEigenLayerMiddlewareV1 is IBoltMiddlewareV1, IServiceManager, OwnableUpgradeable, UUPSUpgradeable {
+contract BoltEigenLayerMiddlewareV2 is IBoltMiddlewareV1, IServiceManager, OwnableUpgradeable, UUPSUpgradeable {
     using EnumerableSet for EnumerableSet.AddressSet;
     using EnumerableMap for EnumerableMap.AddressToUintMap;
     using MapWithTimeData for EnumerableMap.AddressToUintMap;

diff --git a/bolt-contracts/test/BoltManager.EigenLayer.t.sol b/bolt-contracts/test/BoltManager.EigenLayer.t.sol
@@ -6,7 +6,7 @@ import {Test, console} from "forge-std/Test.sol";
 import {BoltValidatorsV1} from "../src/contracts/BoltValidatorsV1.sol";
 import {BoltManagerV1} from "../src/contracts/BoltManagerV1.sol";
 import {BoltParametersV1} from "../src/contracts/BoltParametersV1.sol";
-import {BoltEigenLayerMiddlewareV1} from "../src/contracts/BoltEigenLayerMiddlewareV1.sol";
+import {BoltEigenLayerMiddlewareV2} from "../src/contracts/BoltEigenLayerMiddlewareV2.sol";
 import {BoltConfig} from "../src/lib/Config.sol";
 import {IBoltValidatorsV1} from "../src/interfaces/IBoltValidatorsV1.sol";
 import {IBoltManagerV1} from "../src/interfaces/IBoltManagerV1.sol";
@@ -30,7 +30,7 @@ contract BoltManagerEigenLayerTest is Test {
 
     BoltValidatorsV1 public validators;
     BoltManagerV1 public manager;
-    BoltEigenLayerMiddlewareV1 public middleware;
+    BoltEigenLayerMiddlewareV2 public middleware;
     EigenLayerDeployer public eigenLayerDeployer;
 
     uint128 public constant PRECONF_MAX_GAS_LIMIT = 5_000_000;
@@ -74,7 +74,7 @@ contract BoltManagerEigenLayerTest is Test {
         validators.initialize(admin, address(parameters));
         manager = new BoltManagerV1();
         manager.initialize(admin, address(parameters), address(validators));
-        middleware = new BoltEigenLayerMiddlewareV1();
+        middleware = new BoltEigenLayerMiddlewareV2();
 
         middleware.initialize(
             address(admin),