Merge pull request #5942 from BitGo/WIN-5187-wemix

feat: adding test cases for non bitgo recovery for Wemix

Author: prithvishet2503

modules/sdk-coin-wemix/package.json

@@ -43,7 +43,8 @@
     "@bitgo/abstract-eth": "^24.1.0",
     "@bitgo/sdk-core": "^32.1.0",
     "@bitgo/statics": "^51.7.0",
-    "@ethereumjs/common": "^2.6.5"
+    "@ethereumjs/common": "^2.6.5",
+    "@ethereumjs/tx": "^3.3.0"
   },
   "devDependencies": {
     "@bitgo/sdk-api": "^1.61.5",

modules/sdk-coin-wemix/test/resources.ts

@@ -54,3 +54,47 @@ export const mockDataUnsignedSweep = {
   getBalanceRequest: getBalanceRequestUnsignedSweep,
   getBalanceResponse: getBalanceResponseUnsignedSweep,
 };
+
+const getTxListRequestNonBitGoRecovery: Record<string, string> = {
+  module: 'account',
+  action: 'txlist',
+  address: '0xcf05f9c25579832d2237e52ffb4d16ca3153bb15',
+};
+
+const getTxListResponseNonBitGoRecovery: Record<string, unknown> = {
+  status: '1',
+  result: [
+    {
+      hash: '0xede855d43d70ea1bb75db63d4f75113dae0845f0d4bdb0b2d8bda55249c70812',
+      nonce: '23',
+      from: '0xcf05f9c25579832d2237e52ffb4d16ca3153bb15',
+    },
+  ],
+  message: 'OK',
+};
+
+const getBalanceRequestNonBitGoRecovery: Record<string, string> = {
+  module: 'account',
+  action: 'balance',
+  address: '0xcf05f9c25579832d2237e52ffb4d16ca3153bb15',
+};
+
+const getBalanceResponseNonBitGoRecovery: Record<string, unknown> = {
+  status: '1',
+  result: '100000000000000000',
+  message: 'OK',
+};
+
+export const mockDataNonBitGoRecovery = {
+  recoveryDestination: '0x26efa259beeb4373aff0f0e37167a7b6255fe34e',
+  userKeyData:
+    '{"iv":"LkXud+OJjiSgFSz2GUF90g==","v":1,"iter":10000,"ks":256,"ts":64,"mode":"ccm","adata":"","cipher":"aes","salt":"qpjtz9Fp4u8=","ct":"syqUtFTqWy/itfdnwckTSSk8a2tYVQIApUP811rfWBth0ODZzjLzWs+pM0IYi9IM9boQ5E94/Kutk1HVNWC7q1vG88B/ZI5wVc5Scr2wgOJ62TifS6VNgmaGfAlMfke0tL+2icn/NqiEAPXlan4VpiSIzGV50V+BPW6x/zWOqzA8OiZ7WApsqpIMiNHal+27g1F41ALGnpVhuDBr/M67FLvzdCz38dZoqcbV67CncqXzGFvDmw4Z2h86DP4jJFhGQUkgJqwKjFVOFREnFClkg1VgPryYpWvx5liHZv4URTSlqTUj7VS/FtYoJFobilSnHz0tCKM8q81sJ4hz9mzxsQRL6xuV4SoM44TGSPutQfCTNWyXqq9PNe4er3BpRS4FGl6qrBFkyAkx4zF8NJDI3+miFlG1YmLSq7aLRuU5RAnQTTY+XnvbE7xsF9zhAzTsJk9zjZUQVDZDAC+kOiaYtZBC8wsJ26i1zbo9g6V9O+9X5tVFauCt47W9kodi+PNservj8tBYvG6YjjJ1SK/l2SQAffa+PZX4+b51Qf8FBawdcRkCfasZPjUYjYZ0KiiRQmSg3OMz3AqEw8yelxJ/lkbBOccrNcbVpYZh+OTB2Kjvvhk5aAs4WiaO+7PM3YrhjGhjln1eAvh4ABzXlOHV31f41p5/MeOgWwf0SH2aZocM1erCpk0+dk6Mu4Z46D9NTYY0JoaljKsvxpWidtRHp93YkuNjVI1qIaMyJj3O8pe6782JfT+sRQfJJxTvLThzm0ljy9hQGHK98GRejge4T1p66Jkhu94kmKpJJHOt9iVuU5mbc3kA0pi148B5HTRzH6/I0v2m25tcspsagvaiQeaU4+RAsoTrWiFtHG3UByCMaS6jiIU0b5zKKPbWqoQV85ywyZOi4xtIQoqFUte7v1ziTzaywGb3tG2iAHYRrJgAPg2wNYDYVOOIJWdhtxw8G/uKMq9EZ8+l2FLdXpUHDc1UJzNqf6ijYbixMv0RHDMXtPC+870ciyVjGAAfPwWrnXafgJewRsYpeUxiJvf1A9DEZ0IyVKT4TjohEmuHlUxT5d0QJ1e1Pxa0qaeLrNoU"}',
+  backupKeyData:
+    '{"iv":"j01iJ6D6yrIAvQG39rUK7A==","v":1,"iter":10000,"ks":256,"ts":64,"mode":"ccm","adata":"","cipher":"aes","salt":"FZhlN/ZL7EQ=","ct":"l8rWds8F0MUrnyvIkfoU9Pcf5rR9eJwZkyjzzrrQaor4oIYcqNl3+OfuXYFssSMhor3AuylZ85hLaovqndvfYtPBC2AZ1sM1AAMmwmptGWeuRuzIjd9bVUzGJdAljDZ3/ikcmMbV/drT7HSu6l2Z2tsc5XJwJujO2YobCOlBBGWM8lIqCcCORa+oliLXW2YxsRzEEM5b2H/RqqQKpZeJvYK3l5EX8aAyTIMqrIv2ZS9MaWXcY95/IYB5xO/a1zZRfxDv5WBu4RsVtS5uIH99PPNJIcAXAAFN+n58EsfZbtJ6aDWCDngtEaghiwlSUhcdwU59P6eUmzEjEvY4RJwDMX0Mq9C1kfwRxNn9+BwdaO24HaH4uWsJlHNp++onlTEb9pDp0nmMgWLQjh1n0CrF5XThSt9i5s2jMFNEE7LruwRT+TKzpN48If2L8IvwFiFWLuUtQDJ0jdl/8pRB0C0yc443DIxROBluh8hwmYqglK5YtaBZitXl++jSaXdRlgJ4wBfTuDp4Lg62wWgprPhaRFWfMJpXedDfzZEDjc2UFZvKKe7V1+f7kSB9ol18aE4xpEN5vta6L81ZMgVWkbWu7Ascu4P4vOl1M8tPWkLmg6SpPbbxYdp5Q52DMkhLz8zuAaUD5tbt6tVKTCd0USEus2s+r4VJd6WImdvNUsg9anSpWQp48jWx0ZcLHnACOyT1bRC/VcYwEB6XE81uyhWdZYrhly/MKcJNO9Auij/RxNPpELYNL1Ru+Vh7ZNSE3EotzoRba0+QkHCOq3xGi2tSzZlmJe329LZEjpiQyGM7tLV8ZnSngXXUEXGQXVDaCeQQz82VYhZpKeC9xCTPLCUMETWfiqU4XhFfRzyLjIAZKlzq6yOMsLrXFpsA85XpJciyYhO9MHd2/8yEyZfernqN3qLY0dl4dVR53E4UwWSwLSbXFV6dD1ZheB4G8tSp2oq/IRHw7ZuQlRcW0Bn5kaDr3gs5+7enQ9SSBLkPrAYfeloAVyz/ugKJ3+ZtEBFkTbobORUHtfsrjYqxb7fn6/n5kvrm8fZzy3Oq+SG//B1guuyElM/cJ8J+yHQvNv0AKQCm"}',
+  walletPassphrase: 'prithvi-bitgo',
+  walletRootAddress: '0xcf05f9c25579832d2237e52ffb4d16ca3153bb15',
+  getTxListRequest: getTxListRequestNonBitGoRecovery,
+  getTxListResponse: getTxListResponseNonBitGoRecovery,
+  getBalanceRequest: getBalanceRequestNonBitGoRecovery,
+  getBalanceResponse: getBalanceResponseNonBitGoRecovery,
+};

modules/sdk-coin-wemix/test/unit/wemix.ts

@@ -5,9 +5,11 @@ import { BitGoAPI } from '@bitgo/sdk-api';
 
 import { Wemix, Twemix } from '../../src/index';
 import { UnsignedSweepTxMPCv2 } from '@bitgo/abstract-eth';
-import { mockDataUnsignedSweep } from '../resources';
+import { mockDataUnsignedSweep, mockDataNonBitGoRecovery } from '../resources';
 import nock from 'nock';
 import { common } from '@bitgo/sdk-core';
+import { FeeMarketEIP1559Transaction } from '@ethereumjs/tx';
+import { stripHexPrefix } from '@ethereumjs/util';
 
 const bitgo: TestBitGoAPI = TestBitGo.decorate(BitGoAPI, { env: 'test' });
 
@@ -96,3 +98,47 @@ describe('Build Unsigned Sweep for Self-Custody Cold Wallets (MPCv2)', function
     tx.unsignedTx.parsedTx?.should.have.property('outputs');
   });
 });
+
+describe('Non Bitgo Recovery for Hot Wallets', function () {
+  const bitgo = TestBitGo.decorate(BitGoAPI, { env: 'test' });
+  const explorerUrl = common.Environments[bitgo.getEnv()].wemixExplorerBaseUrl as string;
+  const maxFeePerGasvalue = 20000000000;
+  const maxPriorityFeePerGasValue = 10000000000;
+  const chain_id = 1112;
+  const gasLimitvalue = 500000;
+
+  it('should generate a signed non-bitgo recovery tx', async () => {
+    nock(explorerUrl)
+      .get('/api')
+      .twice()
+      .query(mockDataNonBitGoRecovery.getTxListRequest)
+      .reply(200, mockDataNonBitGoRecovery.getTxListResponse);
+    nock(explorerUrl)
+      .get('/api')
+      .query(mockDataNonBitGoRecovery.getBalanceRequest)
+      .reply(200, mockDataNonBitGoRecovery.getBalanceResponse);
+
+    const baseCoin: any = bitgo.coin('twemix');
+    const transaction = await baseCoin.recover({
+      userKey: mockDataNonBitGoRecovery.userKeyData,
+      backupKey: mockDataNonBitGoRecovery.backupKeyData,
+      walletContractAddress: mockDataNonBitGoRecovery.walletRootAddress,
+      walletPassphrase: mockDataNonBitGoRecovery.walletPassphrase,
+      recoveryDestination: mockDataNonBitGoRecovery.recoveryDestination,
+      isTss: true,
+      eip1559: { maxFeePerGas: maxFeePerGasvalue, maxPriorityFeePerGas: maxPriorityFeePerGasValue },
+      gasLimit: gasLimitvalue,
+      replayProtectionOptions: {
+        chain: chain_id,
+        hardfork: 'london',
+      },
+    });
+    should.exist(transaction);
+    transaction.should.have.property('id');
+    transaction.should.have.property('tx');
+    const tx = FeeMarketEIP1559Transaction.fromSerializedTx(Buffer.from(stripHexPrefix(transaction.tx), 'hex'));
+    tx.getSenderAddress().toString().should.equal(mockDataNonBitGoRecovery.walletRootAddress);
+    const jsonTx = tx.toJSON();
+    jsonTx.to?.should.equal(mockDataNonBitGoRecovery.recoveryDestination);
+  });
+});