@@ -6,6 +6,7 @@ import "forge-std/Test.sol";
66import { ECDSA } from "@openzeppelin/contracts/utils/cryptography/ECDSA.sol " ;
77import { TokenUtils } from "@graphprotocol/contracts/contracts/utils/TokenUtils.sol " ;
88import { Controller } from "@graphprotocol/contracts/contracts/governance/Controller.sol " ;
9+ import { UnsafeUpgrades } from "openzeppelin-foundry-upgrades/Upgrades.sol " ;
910
1011import { DisputeManager } from "../contracts/DisputeManager.sol " ;
1112import { IDisputeManager } from "../contracts/interfaces/IDisputeManager.sol " ;
@@ -72,16 +73,24 @@ contract DisputeManagerTest is Test {
7273 controller.setContractProxy (keccak256 ("GraphToken " ), address (graphToken));
7374 controller.setContractProxy (keccak256 ("Staking " ), address (staking));
7475 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 ));
7582 vm.stopPrank ();
7683
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+ )
8492 );
93+ disputeManager = DisputeManager (disputeManagerProxy);
8594
8695 subgraphService = new SubgraphService (address (controller), address (disputeManager), tapVerifier, curation);
8796 subgraphService.initialize (1000 ether, 16 );
@@ -110,7 +119,7 @@ contract DisputeManagerTest is Test {
110119 vm.startPrank (fisherman);
111120 graphToken.mint (fisherman, tokens);
112121 graphToken.approve (address (disputeManager), tokens);
113- bytes32 _disputeID = disputeManager.createIndexingDispute (_allocationID, tokens);
122+ bytes32 _disputeID = disputeManager.createIndexingDispute (_allocationID, bytes32 ( " POI1234 " ), tokens);
114123 vm.stopPrank ();
115124 return _disputeID;
116125 }
@@ -218,7 +227,7 @@ contract DisputeManagerTest is Test {
218227 graphToken.approve (address (disputeManager), tokens);
219228 bytes memory expectedError = abi.encodeWithSignature ("DisputeManagerDisputeAlreadyCreated(bytes32) " , disputeID);
220229 vm.expectRevert (expectedError);
221- disputeManager.createIndexingDispute (allocationID, tokens);
230+ disputeManager.createIndexingDispute (allocationID, bytes32 ( " POI1234 " ), tokens);
222231 vm.stopPrank ();
223232 }
224233
@@ -232,7 +241,7 @@ contract DisputeManagerTest is Test {
232241 100 ether
233242 );
234243 vm.expectRevert (expectedError);
235- disputeManager.createIndexingDispute (allocationID, 50 ether);
244+ disputeManager.createIndexingDispute (allocationID, bytes32 ( " POI1234 " ), 50 ether);
236245 vm.stopPrank ();
237246 }
238247
@@ -244,7 +253,7 @@ contract DisputeManagerTest is Test {
244253 graphToken.approve (address (disputeManager), tokens);
245254 bytes memory expectedError = abi.encodeWithSignature ("DisputeManagerIndexerNotFound(address) " , allocationID);
246255 vm.expectRevert (expectedError);
247- disputeManager.createIndexingDispute (allocationID, tokens);
256+ disputeManager.createIndexingDispute (allocationID, bytes32 ( " POI1234 " ), tokens);
248257 vm.stopPrank ();
249258 }
250259
0 commit comments