fix: 🐛 getKYCData test method is working again

Author: shuffledex

src/contract_wrappers/modules/transfer_manager/__tests__/general_transfer_manager_wrapper.test.ts

@@ -465,27 +465,38 @@ describe('GeneralTransferManagerWrapper', () => {
   describe('Get KYC', () => {
     test('should getKYCData', async () => {
       const mockedParams = {
-        investors: ['0x3333333333333333333333333333333333333333', '0x5555555555555555555555555555555555555555'],
+        investors: [
+          '0x3333333333333333333333333333333333333333',
+          '0x4444444444444444444444444444444444444444',
+          '0x5555555555555555555555555555555555555555',
+        ],
       };
       // Address expected
       const expectedResult = [
-        [new BigNumber(1), new BigNumber(2), new BigNumber(3)],
-        [new BigNumber(1), new BigNumber(2), new BigNumber(3)],
+        [new BigNumber(100), new BigNumber(101), new BigNumber(102)],
+        [new BigNumber(103), new BigNumber(104), new BigNumber(105)],
+        [new BigNumber(106), new BigNumber(107), new BigNumber(108)],
       ];
       // Mocked method
       const mockedMethod = mock(MockedCallMethod);
       // Stub the method
       when(mockedContract.getKYCData).thenReturn(instance(mockedMethod));
       // Stub the request
-      when(mockedMethod.callAsync(mockedParams)).thenResolve(expectedResult);
+      when(mockedMethod.callAsync(mockedParams.investors)).thenResolve(expectedResult);
 
       // Real call
       const result = await target.getKYCData(mockedParams);
+
       // Result expectation
-      expect(result).toBe(expectedResult);
+      for (let i = 0; i < 3; i += 1) {
+        expect(dateToBigNumber(result[i].canSendAfter)).toEqual(expectedResult[0][i]);
+        expect(dateToBigNumber(result[i].canReceiveAfter)).toEqual(expectedResult[1][i]);
+        expect(dateToBigNumber(result[i].expiryTime)).toEqual(expectedResult[2][i]);
+      }
+
       // Verifications
       verify(mockedContract.getKYCData).once();
-      verify(mockedMethod.callAsync()).once();
+      verify(mockedMethod.callAsync(mockedParams.investors)).once();
     });
   });
 
@@ -538,9 +549,9 @@ describe('GeneralTransferManagerWrapper', () => {
       // Real call
       await expect(target.subscribeAsync(mockedParams)).rejects.toEqual(
         new Error(
-          `Expected eventName to be one of: 'ChangeIssuanceAddress', 'AllowAllTransfers', 'AllowAllWhitelistTransfers', 'AllowAllWhitelistIssuances', 'AllowAllBurnTransfers', 'ChangeSigningAddress', 'ChangeDefaults', 'ModifyWhitelist', 'Pause', 'Unpause', encountered: Transfer`,
+          `Expected eventName to be one of: 'ChangeIssuanceAddress', 'ChangeDefaults', 'ModifyKYCData', 'ModifyInvestorFlag', 'ModifyTransferRequirements', 'Pause', 'Unpause', encountered: Transfer`,
         ),
       );
     });
   });
-});
+});