File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
packages/subgraph-service/test/subgraphService Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ import { IHorizonStakingTypes } from "@graphprotocol/horizon/contracts/interface
1616import { Allocation } from "../../contracts/libraries/Allocation.sol " ;
1717import { AllocationManager } from "../../contracts/utilities/AllocationManager.sol " ;
1818import { ISubgraphService } from "../../contracts/interfaces/ISubgraphService.sol " ;
19+ import { LegacyAllocation } from "../../contracts/libraries/LegacyAllocation.sol " ;
1920import { SubgraphServiceSharedTest } from "../shared/SubgraphServiceShared.t.sol " ;
2021
2122contract SubgraphServiceTest is SubgraphServiceSharedTest {
@@ -386,9 +387,9 @@ contract SubgraphServiceTest is SubgraphServiceSharedTest {
386387
387388 subgraphService.migrateLegacyAllocation (_indexer, _allocationId, _subgraphDeploymentID);
388389
389- ( address afterIndexer , bytes32 afterSubgraphDeploymentId ) = subgraphService.legacyAllocations (_allocationId);
390- assertEq (afterIndexer , _indexer);
391- assertEq (afterSubgraphDeploymentId , _subgraphDeploymentID);
390+ LegacyAllocation.State memory afterLegacyAllocation = subgraphService.getLegacyAllocation (_allocationId);
391+ assertEq (afterLegacyAllocation.indexer , _indexer);
392+ assertEq (afterLegacyAllocation.subgraphDeploymentId , _subgraphDeploymentID);
392393 }
393394
394395 /*
You can’t perform that action at this time.
0 commit comments