import chai, { expect } from 'chai';
import sinonChai from 'sinon-chai';
import { exportPrivateKey } from '../../../src/rpc/export-private-key';
import { WalletMock } from '../wallet.mock.test';
import { testBip44Entropy, testPrivateKey } from './key-pair-test-constants';
import { SnapConfig } from '@solana-tools/solsnap-types';

chai.use(sinonChai);

describe('Test rpc handler function: exportSeed', function () {
  const walletStub = new WalletMock();

  afterEach(function () {
    walletStub.reset();
  });

  it('should return seed on positive prompt confirmation and keyring saved in state', async function () {
    walletStub.send.returns(true);
    walletStub.send.returns(testBip44Entropy);
    walletStub.getPluginState.returns({
      solana: {
        config: {
          network: 's',
          derivationPath: "m/44'/501'/0'/0/0",
        } as SnapConfig,
      },
    });

    const result = await exportPrivateKey(walletStub);

    expect(walletStub.getAppKey).to.have.not.been.called;
    expect(walletStub.send).to.have.been.calledTwice;
    expect(walletStub.getPluginState).to.have.been.calledOnce;
    expect(result).to.be.eq(testPrivateKey);
  });

  it('should not return seed on negative prompt confirmation', async function () {
    walletStub.send.returns(false);

    const result = await exportPrivateKey(walletStub);

    expect(walletStub.send).to.have.been.calledOnce;
    expect(result).to.be.eq(null);
  });
});