@@ -28,6 +28,7 @@ contract KlerosCoreTest is Test {
2828 BlockHashRNG rng;
2929 PNK pinakion;
3030 TestERC20 feeToken;
31+ TestERC20 wNative;
3132 ArbitrableExample arbitrable;
3233 DisputeTemplateRegistry registry;
3334 address governor;
@@ -65,6 +66,7 @@ contract KlerosCoreTest is Test {
6566 rng = new BlockHashRNG ();
6667 pinakion = new PNK ();
6768 feeToken = new TestERC20 ("Test " , "TST " );
69+ wNative = new TestERC20 ("wrapped ETH " , "wETH " );
6870
6971 governor = msg .sender ;
7072 guardian = vm.addr (1 );
@@ -96,7 +98,12 @@ contract KlerosCoreTest is Test {
9698
9799 UUPSProxy proxyCore = new UUPSProxy (address (coreLogic), "" );
98100
99- bytes memory initDataDk = abi.encodeWithSignature ("initialize(address,address) " , governor, address (proxyCore));
101+ bytes memory initDataDk = abi.encodeWithSignature (
102+ "initialize(address,address,address) " ,
103+ governor,
104+ address (proxyCore),
105+ address (wNative)
106+ );
100107
101108 UUPSProxy proxyDk = new UUPSProxy (address (dkLogic), initDataDk);
102109 disputeKit = DisputeKitClassic (address (proxyDk));
@@ -125,7 +132,8 @@ contract KlerosCoreTest is Test {
125132 [minStake, alpha, feeForJuror, jurorsForCourtJump],
126133 timesPerPeriod,
127134 sortitionExtraData,
128- sortitionModule
135+ sortitionModule,
136+ address (wNative)
129137 );
130138 vm.prank (staker1);
131139 pinakion.approve (address (core), 1 ether);
@@ -277,7 +285,12 @@ contract KlerosCoreTest is Test {
277285
278286 UUPSProxy proxyCore = new UUPSProxy (address (coreLogic), "" );
279287
280- bytes memory initDataDk = abi.encodeWithSignature ("initialize(address,address) " , governor, address (proxyCore));
288+ bytes memory initDataDk = abi.encodeWithSignature (
289+ "initialize(address,address,address) " ,
290+ governor,
291+ address (proxyCore),
292+ address (wNative)
293+ );
281294
282295 UUPSProxy proxyDk = new UUPSProxy (address (dkLogic), initDataDk);
283296 disputeKit = DisputeKitClassic (address (proxyDk));
@@ -325,7 +338,8 @@ contract KlerosCoreTest is Test {
325338 [minStake, alpha, feeForJuror, jurorsForCourtJump],
326339 timesPerPeriod,
327340 sortitionExtraData,
328- sortitionModule
341+ sortitionModule,
342+ address (wNative)
329343 );
330344 }
331345
@@ -2075,7 +2089,12 @@ contract KlerosCoreTest is Test {
20752089 uint256 disputeID = 0 ;
20762090 DisputeKitClassic dkLogic = new DisputeKitClassic ();
20772091 // Create a new DK and court to check the switch
2078- bytes memory initDataDk = abi.encodeWithSignature ("initialize(address,address) " , governor, address (core));
2092+ bytes memory initDataDk = abi.encodeWithSignature (
2093+ "initialize(address,address,address) " ,
2094+ governor,
2095+ address (core),
2096+ address (wNative)
2097+ );
20792098
20802099 UUPSProxy proxyDk = new UUPSProxy (address (dkLogic), initDataDk);
20812100 DisputeKitClassic newDisputeKit = DisputeKitClassic (address (proxyDk));
@@ -2856,7 +2875,12 @@ contract KlerosCoreTest is Test {
28562875 function test_castVote_differentDK () public {
28572876 DisputeKitClassic dkLogic = new DisputeKitClassic ();
28582877 // Create a new DK to check castVote.
2859- bytes memory initDataDk = abi.encodeWithSignature ("initialize(address,address) " , governor, address (core));
2878+ bytes memory initDataDk = abi.encodeWithSignature (
2879+ "initialize(address,address,address) " ,
2880+ governor,
2881+ address (core),
2882+ address (wNative)
2883+ );
28602884
28612885 UUPSProxy proxyDk = new UUPSProxy (address (dkLogic), initDataDk);
28622886 DisputeKitClassic newDisputeKit = DisputeKitClassic (address (proxyDk));
0 commit comments