David Walsh
67 lines
2.5 KiB
TypeScript
67 lines
2.5 KiB
TypeScript
import { BigNumber } from '@0x/utils';
|
|
import { instance, mock, when } from 'ts-mockito';
|
|
|
|
import { GWEI_DECIMALS } from '../../src/core/constants';
|
|
import { GasOracleType0 } from '../../src/utils/GasOracleType0';
|
|
import { GasStationAttendantPolygon } from '../../src/utils/GasStationAttendantPolygon';
|
|
|
|
let gasOracleMock: GasOracleType0;
|
|
|
|
describe('GasStationAttendantPolygon', () => {
|
|
beforeAll(() => {
|
|
gasOracleMock = mock(GasOracleType0);
|
|
});
|
|
|
|
describe('getWorkerBalanceForTradeAsync', () => {
|
|
it('gets the balance to trade', async () => {
|
|
when(gasOracleMock.getGasWeiAsync('fast')).thenResolve(new BigNumber(1000).shiftedBy(GWEI_DECIMALS));
|
|
|
|
const attendant = new GasStationAttendantPolygon(instance(gasOracleMock));
|
|
|
|
const workerGasToTrade = await attendant.getWorkerBalanceForTradeAsync();
|
|
|
|
// 1000 GWEI * (1.1^3) = 1331
|
|
// Gas estimate = 100,000 * 1.1 = 110,000
|
|
// Total = 1331 * 110,000 = 146,410,000 GWEI
|
|
|
|
expect(workerGasToTrade.toPrecision(2).toString()).toEqual(
|
|
new BigNumber(146_410_000).times(Math.pow(10, GWEI_DECIMALS)).toPrecision(2).toString(),
|
|
);
|
|
});
|
|
|
|
it('throws if gas oracle rejects', async () => {
|
|
when(gasOracleMock.getGasWeiAsync('fast')).thenReject();
|
|
|
|
const attendant = new GasStationAttendantPolygon(instance(gasOracleMock));
|
|
|
|
expect(attendant.getWorkerBalanceForTradeAsync()).rejects.toThrow();
|
|
});
|
|
});
|
|
|
|
describe('getExpectedTransactionGasRateAsync', () => {
|
|
it('estimates the transaction gas rate', async () => {
|
|
when(gasOracleMock.getGasWeiAsync('fast')).thenResolve(new BigNumber(666).shiftedBy(GWEI_DECIMALS));
|
|
|
|
const attendant = new GasStationAttendantPolygon(instance(gasOracleMock));
|
|
|
|
const gasRate = await attendant.getExpectedTransactionGasRateAsync();
|
|
|
|
// 666 * 1.1^1.5 = 768.35736217 GWEI
|
|
expect(gasRate.toString()).toEqual(
|
|
new BigNumber(768.35736217)
|
|
.times(Math.pow(10, GWEI_DECIMALS))
|
|
.integerValue(BigNumber.ROUND_CEIL)
|
|
.toString(),
|
|
);
|
|
});
|
|
|
|
it('throws if gas oracle rejects', async () => {
|
|
when(gasOracleMock.getGasWeiAsync('fast')).thenReject();
|
|
|
|
const attendant = new GasStationAttendantPolygon(instance(gasOracleMock));
|
|
|
|
expect(attendant.getExpectedTransactionGasRateAsync()).rejects.toThrow();
|
|
});
|
|
});
|
|
});
|