test: 💍 add unit tests

PolymeshAssociation · Oct 24, 2024 · 706b27c · 706b27c
1 parent 9634017
commit 706b27c
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 7 deletions.
diff --git a/src/api/procedures/__tests__/addInstruction.ts b/src/api/procedures/__tests__/addInstruction.ts
@@ -985,7 +985,7 @@ describe('addInstruction procedure', () => {
   });
 
   it('should handle NFT legs', async () => {
-    dsMockUtils.configureMocks({ contextOptions: { did: fromDid } });
+    dsMockUtils.configureMocks({ contextOptions: { did: fromDid, specVersion: 7000003 } });
     entityMockUtils.configureMocks({
       venueOptions: {
         exists: true,
@@ -1015,7 +1015,7 @@ describe('addInstruction procedure', () => {
             null,
             null,
             [undefined],
-            rawPortfolioIds,
+            [rawFrom, rawTo],
             null,
             rawEmptyMediatorSet,
           ],

diff --git a/src/api/procedures/__tests__/modifyInstructionAffirmation.ts b/src/api/procedures/__tests__/modifyInstructionAffirmation.ts
@@ -682,6 +682,7 @@ describe('modifyInstructionAffirmation procedure', () => {
       .calledWith(rawAffirmationStatus)
       .mockReturnValue(AffirmationStatus.Affirmed);
 
+    dsMockUtils.configureMocks({ contextOptions: { specVersion: 7000003 } });
     const proc = procedureMockUtils.getInstance<
       ModifyInstructionAffirmationParams,
       Instruction,
@@ -706,7 +707,7 @@ describe('modifyInstructionAffirmation procedure', () => {
     expect(result).toEqual({
       transaction,
       feeMultiplier: new BigNumber(2),
-      args: [rawInstructionId, rawPortfolioIds, mockAffirmCount],
+      args: [rawInstructionId, [rawPortfolioId, rawPortfolioId], mockAffirmCount],
       resolver: expect.objectContaining({ id }),
     });
   });

diff --git a/src/base/__tests__/Context.ts b/src/base/__tests__/Context.ts
@@ -83,10 +83,9 @@ describe('Context class', () => {
   beforeEach(() => {
     polymeshApi = dsMockUtils.getApiInstance();
 
-    polymeshApi.query.asset = {
-      ...polymeshApi.query.asset,
-      tickerAssetId: expect.any(Function),
-    };
+    dsMockUtils.setRuntimeVersion({
+      specVersion: dsMockUtils.createMockU32(new BigNumber(7000004)),
+    });
 
     dsMockUtils.setConstMock('system', 'ss58Prefix', {
       returnValue: dsMockUtils.createMockU8(new BigNumber(42)),

diff --git a/src/testUtils/mocks/dataSources.ts b/src/testUtils/mocks/dataSources.ts
@@ -431,6 +431,7 @@ interface TxMockData {
 interface ContextOptions {
   did?: string;
   isV6?: boolean;
+  specVersion?: number;
   withSigningManager?: boolean;
   balance?: AccountBalance;
   subsidy?: SubsidyWithAllowance;
@@ -671,6 +672,7 @@ let queryMultiMock = jest.fn();
 const defaultContextOptions: ContextOptions = {
   did: 'someDid',
   isV6: false,
+  specVersion: 7000004,
   withSigningManager: true,
   balance: {
     free: new BigNumber(100),
@@ -815,6 +817,7 @@ let signingManagerOptions = defaultSigningManagerOptions;
 function configureContext(opts: ContextOptions): void {
   const getSigningIdentity = jest.fn();
   const isV6 = opts.isV6;
+  const specVersion = opts.specVersion;
   const identity = {
     did: opts.did,
     hasRoles: jest.fn().mockResolvedValue(opts.hasRoles),
@@ -885,6 +888,7 @@ function configureContext(opts: ContextOptions): void {
   const queryMock = mockInstanceContainer.apolloInstance.query;
   const contextInstance = {
     isV6,
+    specVersion,
     signingAddress,
     nonce,
     getSigningIdentity,