PHP WebShell

Текущая директория: /opt/BitGoJS/modules/bitgo/test/v2/unit/staking

Просмотр файла: goStakingWalletCommon.ts

import * as nock from 'nock';
import * as should from 'should';
import fixtures from '../../fixtures/staking/goStakingWallet';

import { Enterprise, Environments, GoStakingWallet, Wallet } from '@bitgo/sdk-core';
import { TestBitGo } from '@bitgo/sdk-test';
import { BitGo } from '../../../../src';
import * as sinon from 'sinon';
import { OfcToken } from '../../../../src/v2/coins';
import { tokens } from '@bitgo/statics';

describe('Go Staking Wallet Common', function () {
  const microservicesUri = Environments['mock'].uri;
  let bitgo;
  let baseCoin;
  let enterprise;
  let stakingWallet: GoStakingWallet;
  const coin = 'tsol';
  const ofcCoin = `ofc${coin}`;

  before(async function () {
    bitgo = TestBitGo.decorate(BitGo, { env: 'mock', microservicesUri } as any);
    bitgo.initializeTestVars();
    baseCoin = bitgo.coin(ofcCoin);
    baseCoin.keychains();
    const ofcToken = tokens.testnet.ofc.tokens.filter((token) => token.type === `ofc${coin}`)[0];
    const tokenConstructor = OfcToken.createTokenConstructor(ofcToken);
    bitgo.register(ofcToken.type, tokenConstructor);

    enterprise = new Enterprise(bitgo, baseCoin, { id: '5cf940949449412d00f53b3d92dbcaa3', name: 'Test Enterprise' });
    const walletData = {
      id: 'walletId',
      coin: ofcCoin,
      enterprise: enterprise.id,
      keys: ['5b3424f91bf349930e340175', '5b3424f91bf349930e340174', '5b3424f91bf349930e340173'],
    };
    const wallet = new Wallet(bitgo, baseCoin, walletData);
    stakingWallet = wallet.toGoStakingWallet();
    nock(microservicesUri)
      .get(`/api/v2/${ofcCoin}/key/${stakingWallet.wallet.keyIds()[0]}`)
      .reply(200, {
        id: stakingWallet.wallet.keyIds()[0],
        pub: 'xpub661MyMwAqRbcFq65dvGMeEVb81KKDRRkWkawSVesWcyevGc5gr8V27LjNfkktaMuKtM362jhgKy2eu35RdArcmmEAoULzAvgKkJpWQPvLXM',
        source: 'user',
        encryptedPrv: bitgo.encrypt({
          input:
            'xprvA41z7zogVVwxVSgdKUHDy1SKmdb533PjDz7J6N6mV6uS3ze1ai8FHa8kmHScGpWmj4WggLyQjgPie1rFSruoUihUZREPSL39UNdE3BBDu76',
          password: 'passphrase',
        }),
        coinSpecific: {},
      });

    nock(microservicesUri)
      .get(`/api/v2/${ofcCoin}/key/${stakingWallet.wallet.keyIds()[1]}`)
      .reply(200, {
        id: stakingWallet.wallet.keyIds()[1],
        pub: 'xpub661MyMwAqRbcFq65dvGMeEVb81KKDRRkWkawSVesWcyevGc5gr8V27LjNfkktaMuKtM362jhgKy2eu35RdArcmmEAoULzAvgKkJpWQPvLXM',
        source: 'user',
        encryptedPrv: bitgo.encrypt({
          input:
            'xprvA41z7zogVVwxVSgdKUHDy1SKmdb533PjDz7J6N6mV6uS3ze1ai8FHa8kmHScGpWmj4WggLyQjgPie1rFSruoUihUZREPSL39UNdE3BBDu76',
          password: 'passphrase',
        }),
        coinSpecific: {},
      });

    nock(microservicesUri)
      .get(`/api/v2/${ofcCoin}/key/${stakingWallet.wallet.keyIds()[2]}`)
      .reply(200, {
        id: stakingWallet.wallet.keyIds()[2],
        pub: 'xpub661MyMwAqRbcFq65dvGMeEVb81KKDRRkWkawSVesWcyevGc5gr8V27LjNfkktaMuKtM362jhgKy2eu35RdArcmmEAoULzAvgKkJpWQPvLXM',
        source: 'user',
        encryptedPrv: bitgo.encrypt({
          input:
            'xprvA41z7zogVVwxVSgdKUHDy1SKmdb533PjDz7J6N6mV6uS3ze1ai8FHa8kmHScGpWmj4WggLyQjgPie1rFSruoUihUZREPSL39UNdE3BBDu76',
          password: 'passphrase',
        }),
        coinSpecific: {},
      });
  });

  const sandbox = sinon.createSandbox();

  afterEach(function () {
    sandbox.verifyAndRestore();
  });

  describe('stake', function () {
    it('should call go-staking-service to stake', async function () {
      const preview = fixtures.previewGoStakingRequest(coin);
      const msScope1 = nock(microservicesUri)
        .post(`/api/go-staking/v1/${ofcCoin}/accounts/${stakingWallet.accountId}/requests/preview`, {
          amount: '1',
          clientId: 'clientId',
          type: 'STAKE',
        })
        .reply(201, preview);

      const expected = fixtures.finalizeGoStakingRequest(coin, 'STAKE');
      const msScope2 = nock(microservicesUri)
        .post(`/api/go-staking/v1/${ofcCoin}/accounts/${stakingWallet.accountId}/requests/finalize`, {
          amount: '1',
          clientId: 'clientId',
          frontTransferSendRequest: {
            halfSigned: {
              payload: preview.payload,
            },
          },
          type: 'STAKE',
        })
        .reply(201, expected);

      const stakingRequest = await stakingWallet.stake({
        amount: '1',
        clientId: 'clientId',
        walletPassphrase: 'passphrase',
      });

      should.exist(stakingRequest);

      stakingRequest.should.deepEqual(expected);
      msScope1.isDone().should.be.True();
      msScope2.isDone().should.be.True();
    });
  });

  describe('unstake', function () {
    it('should call go-staking-service to unstake', async function () {
      const expected = fixtures.finalizeGoStakingRequest(coin, 'UNSTAKE');
      const msScope = nock(microservicesUri)
        .post(`/api/go-staking/v1/${ofcCoin}/accounts/${stakingWallet.accountId}/requests/finalize`, {
          amount: '1',
          clientId: 'clientId',
          type: 'UNSTAKE',
        })
        .reply(201, expected);

      const stakingRequest = await stakingWallet.unstake({
        amount: '1',
        clientId: 'clientId',
      });

      should.exist(stakingRequest);

      stakingRequest.should.deepEqual(expected);
      msScope.isDone().should.be.True();
    });
  });

  describe('getGoStakingRequest', function () {
    it('should call gostaking-service to get go staking request', async function () {
      const stakingRequestId = '8638284a-dab2-46b9-b07f-21109a6e7220';
      const expected = fixtures.finalizeGoStakingRequest(coin, 'STAKE');
      const msScope = nock(microservicesUri)
        .get(`/api/go-staking/v1/${ofcCoin}/accounts/${stakingWallet.accountId}/requests/${stakingRequestId}`)
        .reply(200, expected);

      const stakingRequest = await stakingWallet.getGoStakingRequest(stakingRequestId);

      should.exist(stakingRequest);

      stakingRequest.should.deepEqual(expected);
      msScope.isDone().should.be.True();
    });
  });
});

Выполнить команду

Для локальной разработки. Не используйте в интернете!