fix: imrpove code and fix tedt

PolymeshAssociation · Jun 26, 2020 · b75d02a · b75d02a
1 parent 9ac5289
commit b75d02a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 24 deletions.
diff --git a/src/api/entities/SecurityToken/TokenHolders.ts b/src/api/entities/SecurityToken/TokenHolders.ts
@@ -49,7 +49,7 @@ export class TokenHolders extends Namespace<SecurityToken> {
 
     const securityToken = new SecurityToken({ ticker }, context);
 
-    const areFrozen = opts?.mintStatus ? await securityToken.transfers.areFrozen() : true;
+    const areFrozen = opts?.mintStatus ? await securityToken.transfers.areFrozen() : undefined;
 
     if (opts?.mintStatus && !areFrozen) {
       const entriesChunks = chunk(entries, 10);
@@ -86,7 +86,7 @@ export class TokenHolders extends Namespace<SecurityToken> {
           ),
           balance: balanceToBigNumber(balance),
         };
-        if (opts && opts.mintStatus && areFrozen) {
+        if (opts?.mintStatus) {
           return { ...entrie, canMint: false };
         }
         return entrie;

diff --git a/src/api/entities/SecurityToken/__tests__/TokenHolders.ts b/src/api/entities/SecurityToken/__tests__/TokenHolders.ts
@@ -112,17 +112,6 @@ describe('TokenHolders class', () => {
     test('should retrieve all the token holders with balance and mint status in false', async () => {
       dsMockUtils.createQueryStub('asset', 'balanceOf');
 
-      const fakeData = [
-        {
-          identity: 'someIdentity',
-          value: 1000,
-        },
-        {
-          identity: 'otherIdentity',
-          value: 2000,
-        },
-      ];
-
       const expectedHolders: IdentityBalance[] = [];
 
       entityMockUtils.configureMocks({
@@ -167,17 +156,6 @@ describe('TokenHolders class', () => {
     test('should retrieve all the token holders with balance and canMint attribute setting in true', async () => {
       dsMockUtils.createQueryStub('asset', 'balanceOf');
 
-      const fakeData = [
-        {
-          identity: 'someIdentity',
-          value: 1000,
-        },
-        {
-          identity: 'otherIdentity',
-          value: 2000,
-        },
-      ];
-
       const expectedHolders: IdentityBalance[] = [];
 
       entityMockUtils.configureMocks({