import { customTokens } from 'reducers/customTokens';
import * as customTokensActions from 'actions/customTokens';
import { Token } from 'types/network';

describe('customTokens reducer', () => {
  const token1: Token = {
    address: 'address',
    symbol: 'OMG',
    decimal: 16
  };
  const token2: Token = {
    address: 'address',
    symbol: 'ANT',
    decimal: 16
  };

  it('should handle CUSTOM_TOKEN_ADD', () => {
    expect(customTokens(undefined, customTokensActions.addCustomToken(token1))).toEqual([token1]);
  });

  it('should handle CUSTOM_TOKEN_REMOVE', () => {
    const state1 = customTokens(undefined, customTokensActions.addCustomToken(token1));
    const state2 = customTokens(state1, customTokensActions.addCustomToken(token2));

    expect(customTokens(state2, customTokensActions.removeCustomToken(token2.symbol))).toEqual([
      token1
    ]);
  });
});