staking: take into account locked tokens for the delegated capacity

Author: abarmat

contracts/staking/Staking.sol

@@ -600,10 +600,10 @@ contract Staking is StakingV1Storage, GraphUpgradeable, IStaking {
      * @return Amount of tokens staked by the indexer
      */
     function getIndexerCapacity(address _indexer) public override view returns (uint256) {
-        Stakes.Indexer storage indexerStake = stakes[_indexer];
+        Stakes.Indexer memory indexerStake = stakes[_indexer];
         uint256 tokensDelegated = delegationPools[_indexer].tokens;
 
-        uint256 tokensDelegatedCap = indexerStake.tokensStaked.mul(uint256(delegationRatio));
+        uint256 tokensDelegatedCap = indexerStake.tokensSecureStake().mul(uint256(delegationRatio));
         uint256 tokensDelegatedCapacity = (tokensDelegated < tokensDelegatedCap)
             ? tokensDelegated
             : tokensDelegatedCap;

test/staking/delegation.test.ts

@@ -551,6 +551,19 @@ describe('Staking::Delegation', () => {
       expect(alloc.tokens).eq(tokensToAllocate)
     })
 
+    it('should account delegation for indexer capacity properly', async function () {
+      // 1:10 delegation capacity
+      await staking.connect(governor.signer).setDelegationRatio(10)
+
+      // Delegate
+      await staking.connect(delegator.signer).delegate(indexer.address, tokensToDelegate)
+
+      // If we unstake all, the indexer capacity should go to zero
+      // Should not be able to use delegated tokens
+      await staking.connect(indexer.signer).unstake(tokensToStake)
+      expect(await staking.getIndexerCapacity(indexer.address)).eq(0)
+    })
+
     it('should send delegation cut of query fees to delegation pool', async function () {
       // 1:10 delegation capacity
       await staking.connect(governor.signer).setDelegationRatio(10)