generated from PolymeshAssociation/typescript-boilerplate
-
Notifications
You must be signed in to change notification settings - Fork 11
/
Portfolios.ts
75 lines (64 loc) · 2.54 KB
/
Portfolios.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import { u64 } from '@polkadot/types';
import BigNumber from 'bignumber.js';
import { IdentityId } from 'polymesh-types/types';
import sinon from 'sinon';
import { Identity, Namespace, NumberedPortfolio, Portfolio } from '~/api/entities';
import { Context } from '~/base';
import { dsMockUtils, entityMockUtils } from '~/testUtils/mocks';
import { Mocked } from '~/testUtils/types';
import { tuple } from '~/types/utils';
import * as utilsModule from '~/utils';
import { Portfolios } from '../Portfolios';
describe('Portfolios class', () => {
const did = 'someDid';
const rawIdentityId = dsMockUtils.createMockIdentityId(did);
const numberedPortfolioId = new BigNumber(1);
const rawNumberedPortfolioId = dsMockUtils.createMockU64(numberedPortfolioId.toNumber());
let mockContext: Mocked<Context>;
let stringToIdentityIdStub: sinon.SinonStub<[string, Context], IdentityId>;
let u64ToBigNumberStub: sinon.SinonStub<[u64], BigNumber>;
let portfolios: Portfolios;
let identity: Identity;
beforeAll(() => {
dsMockUtils.initMocks();
entityMockUtils.initMocks();
stringToIdentityIdStub = sinon.stub(utilsModule, 'stringToIdentityId');
u64ToBigNumberStub = sinon.stub(utilsModule, 'u64ToBigNumber');
});
beforeEach(() => {
mockContext = dsMockUtils.getContextInstance();
identity = new Identity({ did }, mockContext);
portfolios = new Portfolios(identity, mockContext);
});
afterEach(() => {
entityMockUtils.reset();
dsMockUtils.reset();
});
afterAll(() => {
entityMockUtils.cleanup();
dsMockUtils.cleanup();
});
test('should extend namespace', () => {
expect(Portfolios.prototype instanceof Namespace).toBe(true);
});
describe('method: getPortfolios', () => {
test('should retrieve all the portfolios for the current identity', async () => {
dsMockUtils.createQueryStub('portfolio', 'portfolios', {
entries: [
tuple(
[dsMockUtils.createMockIdentityId(did), rawNumberedPortfolioId],
dsMockUtils.createMockBytes()
),
],
});
stringToIdentityIdStub.withArgs(did, mockContext).returns(rawIdentityId);
u64ToBigNumberStub.returns(numberedPortfolioId);
const result = await portfolios.getPortfolios();
expect(result).toHaveLength(2);
expect(result[0] instanceof Portfolio).toBe(true);
expect(result[1] instanceof NumberedPortfolio).toBe(true);
expect((result[0] as Portfolio).owner.did).toEqual(did);
expect((result[1] as NumberedPortfolio).id).toEqual(numberedPortfolioId);
});
});
});