Skip to content

Commit a9b7e46

Browse files
committed
Fix BaseStrategyUpgradeable test initialization via proxy
1 parent b3ccf64 commit a9b7e46

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

pkg/contracts/test/BaseStrategyUpgradeable.t.sol

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import {BaseStrategyUpgradeable} from "../src/BaseStrategyUpgradeable.sol";
77
import {IAllo} from "allo-v2-contracts/core/interfaces/IAllo.sol";
88
import {IStrategy} from "allo-v2-contracts/core/interfaces/IStrategy.sol";
99
import {Errors} from "allo-v2-contracts/core/libraries/Errors.sol";
10+
import {ERC1967Proxy} from "@openzeppelin/contracts/proxy/ERC1967/ERC1967Proxy.sol";
1011

1112
contract MockAllo {
1213
mapping(uint256 => mapping(address => bool)) public managers;
@@ -88,8 +89,12 @@ contract BaseStrategyUpgradeableTest is Test {
8889

8990
function setUp() public {
9091
allo = new MockAllo();
91-
strategy = new BaseStrategyUpgradeableHarness();
92-
strategy.initializeHarness(address(allo), "TEST_STRATEGY", owner);
92+
BaseStrategyUpgradeableHarness impl = new BaseStrategyUpgradeableHarness();
93+
ERC1967Proxy proxy = new ERC1967Proxy(
94+
address(impl),
95+
abi.encodeWithSelector(BaseStrategyUpgradeableHarness.initializeHarness.selector, address(allo), "TEST_STRATEGY", owner)
96+
);
97+
strategy = BaseStrategyUpgradeableHarness(payable(address(proxy)));
9398
}
9499

95100
function test_initSetsOwnerAndAllo() public {

0 commit comments

Comments
 (0)