metamask-extension/ui/app/pages/send/send-content/send-gas-row/tests/send-gas-row-component.test.js

import assert from 'assert';
import React from 'react';
import { shallow } from 'enzyme';
import sinon from 'sinon';
import SendGasRow from '../send-gas-row.component';

import SendRowWrapper from '../../send-row-wrapper/send-row-wrapper.component';
import GasFeeDisplay from '../gas-fee-display/gas-fee-display.component';
import GasPriceButtonGroup from '../../../../../components/app/gas-customization/gas-price-button-group';

const propsMethodSpies = {
  showCustomizeGasModal: sinon.spy(),
  resetGasButtons: sinon.spy(),
};

describe('SendGasRow Component', function () {
  let wrapper;

  describe('render', function () {
    beforeEach(function () {
      wrapper = shallow(
        <SendGasRow
          conversionRate={20}
          convertedCurrency="mockConvertedCurrency"
          gasFeeError
          gasLoadingError={false}
          gasTotal="mockGasTotal"
          gasButtonGroupShown={false}
          showCustomizeGasModal={propsMethodSpies.showCustomizeGasModal}
          resetGasButtons={propsMethodSpies.resetGasButtons}
          gasPriceButtonGroupProps={{
            someGasPriceButtonGroupProp: 'foo',
            anotherGasPriceButtonGroupProp: 'bar',
          }}
        />,
        { context: { t: (str) => `${str}_t`, metricsEvent: () => ({}) } },
      );
      wrapper.setProps({ isMainnet: true });
    });

    afterEach(function () {
      propsMethodSpies.resetGasButtons.resetHistory();
    });

    it('should render a SendRowWrapper component', function () {
      assert.strictEqual(wrapper.name(), 'Fragment');
      assert.strictEqual(wrapper.at(0).find(SendRowWrapper).length, 1);
    });

    it('should pass the correct props to SendRowWrapper', function () {
      const { label, showError, errorType } = wrapper
        .find(SendRowWrapper)
        .first()
        .props();

      assert.strictEqual(label, 'transactionFee_t:');
      assert.strictEqual(showError, true);
      assert.strictEqual(errorType, 'gasFee');
    });

    it('should render a GasFeeDisplay as a child of the SendRowWrapper', function () {
      assert(wrapper.find(SendRowWrapper).first().childAt(0).is(GasFeeDisplay));
    });

    it('should render the GasFeeDisplay', function () {
      const { gasLoadingError, gasTotal, onReset } = wrapper
        .find(SendRowWrapper)
        .first()
        .childAt(0)
        .props();
      assert.strictEqual(gasLoadingError, false);
      assert.strictEqual(gasTotal, 'mockGasTotal');
      assert.strictEqual(propsMethodSpies.resetGasButtons.callCount, 0);
      onReset();
      assert.strictEqual(propsMethodSpies.resetGasButtons.callCount, 1);
    });

    it('should render the GasPriceButtonGroup if gasButtonGroupShown is true', function () {
      wrapper.setProps({ gasButtonGroupShown: true });
      const rendered = wrapper.find(SendRowWrapper).first().childAt(0);
      assert.strictEqual(wrapper.children().length, 2);

      const gasPriceButtonGroup = rendered.childAt(0);
      assert(gasPriceButtonGroup.is(GasPriceButtonGroup));
      assert(gasPriceButtonGroup.hasClass('gas-price-button-group--small'));
      assert.strictEqual(gasPriceButtonGroup.props().showCheck, false);
      assert.strictEqual(
        gasPriceButtonGroup.props().someGasPriceButtonGroupProp,
        'foo',
      );
      assert.strictEqual(
        gasPriceButtonGroup.props().anotherGasPriceButtonGroupProp,
        'bar',
      );
    });

    it('should render an advanced options button if gasButtonGroupShown is true', function () {
      wrapper.setProps({ gasButtonGroupShown: true });
      const rendered = wrapper.find(SendRowWrapper).last();
      assert.strictEqual(wrapper.children().length, 2);

      const advancedOptionsButton = rendered.childAt(0);
      assert.strictEqual(advancedOptionsButton.text(), 'advancedOptions_t');

      assert.strictEqual(propsMethodSpies.showCustomizeGasModal.callCount, 0);
      advancedOptionsButton.props().onClick();
      assert.strictEqual(propsMethodSpies.showCustomizeGasModal.callCount, 1);
    });
  });
});