add test for user's staked balance

makerdao · Jul 8, 2021 · b8c55a3 · b8c55a3
1 parent 46979fd
commit b8c55a3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/packages/dai-plugin-governance/src/VoteDelegateService.js b/packages/dai-plugin-governance/src/VoteDelegateService.js
@@ -1,5 +1,6 @@
 import { LocalService } from '@makerdao/services-core';
 import VoteDelegate from './VoteDelegate';
+import BigNumber from 'bignumber.js';
 import { MKR, VOTE_DELEGATE_FACTORY, ZERO_ADDRESS } from './utils/constants';
 import { fromBuffer, getCurrency } from './utils/helpers';
 import voteDelegateAbi from '../contracts/abis/VoteDelegate.json';
@@ -68,7 +69,11 @@ export default class VoteDelegateService extends LocalService {
   }
 
   async getStakedBalanceForAddress(delegateAddress, address) {
-    return await this._getStakedBalanceForAddress(delegateAddress, address);
+    const bal = await this._getStakedBalanceForAddress(
+      delegateAddress,
+      address
+    );
+    return new BigNumber(bal).shiftedBy(-18);
   }
 
   async getAllDelegates() {

diff --git a/packages/dai-plugin-governance/test/VoteDelegateService.test.js b/packages/dai-plugin-governance/test/VoteDelegateService.test.js
@@ -78,6 +78,16 @@ test('user can lock MKR with a delegate', async () => {
   expect(postLockDeposits.toNumber()).toBe(amountToLock);
 });
 
+test("can check a user's delegated stake", async () => {
+  const stakedAmt = 3;
+  const deposits = await vds.getStakedBalanceForAddress(
+    delegateContractAddress,
+    maker.currentAccount().address
+  );
+
+  expect(deposits.toNumber()).toBe(stakedAmt);
+});
+
 test('delegate can cast an executive vote and retrieve voted on addresses from slate', async () => {
   maker.useAccountWithAddress(delegateAddress);