@@ -6,6 +6,7 @@ import "forge-std/Test.sol";
6
6
import { ECDSA } from "@openzeppelin/contracts/utils/cryptography/ECDSA.sol " ;
7
7
import { TokenUtils } from "@graphprotocol/contracts/contracts/utils/TokenUtils.sol " ;
8
8
import { Controller } from "@graphprotocol/contracts/contracts/governance/Controller.sol " ;
9
+ import { UnsafeUpgrades } from "openzeppelin-foundry-upgrades/Upgrades.sol " ;
9
10
10
11
import { DisputeManager } from "../contracts/DisputeManager.sol " ;
11
12
import { IDisputeManager } from "../contracts/interfaces/IDisputeManager.sol " ;
@@ -72,16 +73,24 @@ contract DisputeManagerTest is Test {
72
73
controller.setContractProxy (keccak256 ("GraphToken " ), address (graphToken));
73
74
controller.setContractProxy (keccak256 ("Staking " ), address (staking));
74
75
controller.setContractProxy (keccak256 ("RewardsManager " ), address (rewardsManager));
76
+ controller.setContractProxy (keccak256 ("GraphPayments " ), address (0x100 ));
77
+ controller.setContractProxy (keccak256 ("PaymentsEscrow " ), address (0x101 ));
78
+ controller.setContractProxy (keccak256 ("EpochManager " ), address (0x102 ));
79
+ controller.setContractProxy (keccak256 ("GraphTokenGateway " ), address (0x103 ));
80
+ controller.setContractProxy (keccak256 ("GraphProxyAdmin " ), address (0x104 ));
81
+ controller.setContractProxy (keccak256 ("Curation " ), address (0x105 ));
75
82
vm.stopPrank ();
76
83
77
- disputeManager = new DisputeManager (address (controller));
78
- disputeManager.initialize (
79
- arbitrator,
80
- disputePeriod,
81
- minimumDeposit,
82
- fishermanRewardPercentage,
83
- maxSlashingPercentage
84
+ address disputeManagerImplementation = address (new DisputeManager (address (controller)));
85
+ address disputeManagerProxy = UnsafeUpgrades.deployTransparentProxy (
86
+ disputeManagerImplementation,
87
+ governor,
88
+ abi.encodeCall (
89
+ DisputeManager.initialize,
90
+ (arbitrator, disputePeriod, minimumDeposit, fishermanRewardPercentage, maxSlashingPercentage)
91
+ )
84
92
);
93
+ disputeManager = DisputeManager (disputeManagerProxy);
85
94
86
95
subgraphService = new SubgraphService (address (controller), address (disputeManager), tapVerifier, curation);
87
96
subgraphService.initialize (1000 ether, 16 );
@@ -110,7 +119,7 @@ contract DisputeManagerTest is Test {
110
119
vm.startPrank (fisherman);
111
120
graphToken.mint (fisherman, tokens);
112
121
graphToken.approve (address (disputeManager), tokens);
113
- bytes32 _disputeID = disputeManager.createIndexingDispute (_allocationID, tokens);
122
+ bytes32 _disputeID = disputeManager.createIndexingDispute (_allocationID, bytes32 ( " POI1234 " ), tokens);
114
123
vm.stopPrank ();
115
124
return _disputeID;
116
125
}
@@ -218,7 +227,7 @@ contract DisputeManagerTest is Test {
218
227
graphToken.approve (address (disputeManager), tokens);
219
228
bytes memory expectedError = abi.encodeWithSignature ("DisputeManagerDisputeAlreadyCreated(bytes32) " , disputeID);
220
229
vm.expectRevert (expectedError);
221
- disputeManager.createIndexingDispute (allocationID, tokens);
230
+ disputeManager.createIndexingDispute (allocationID, bytes32 ( " POI1234 " ), tokens);
222
231
vm.stopPrank ();
223
232
}
224
233
@@ -232,7 +241,7 @@ contract DisputeManagerTest is Test {
232
241
100 ether
233
242
);
234
243
vm.expectRevert (expectedError);
235
- disputeManager.createIndexingDispute (allocationID, 50 ether);
244
+ disputeManager.createIndexingDispute (allocationID, bytes32 ( " POI1234 " ), 50 ether);
236
245
vm.stopPrank ();
237
246
}
238
247
@@ -244,7 +253,7 @@ contract DisputeManagerTest is Test {
244
253
graphToken.approve (address (disputeManager), tokens);
245
254
bytes memory expectedError = abi.encodeWithSignature ("DisputeManagerIndexerNotFound(address) " , allocationID);
246
255
vm.expectRevert (expectedError);
247
- disputeManager.createIndexingDispute (allocationID, tokens);
256
+ disputeManager.createIndexingDispute (allocationID, bytes32 ( " POI1234 " ), tokens);
248
257
vm.stopPrank ();
249
258
}
250
259
0 commit comments