test.ts

import multiColumnSort from './index'
import unsortedRaw from './data/unsorted.json'
import firstNameASCSortedRaw from './data/firstNameASCSorted.json'
import firstNameDESCSortedRaw from './data/firstNameDESCSorted.json'
import firstNameASCBalanceDESCSortedRaw from './data/firstNameASCBalanceDESCSorted.json'
import firstNameDESCBalanceASCSortedRaw from './data/firstNameDESCBalanceASCSorted.json'

interface Data {
  firstName: string
  lastName: string
  balance: string
}

const unsorted: Data[] = unsortedRaw
const firstNameASCSorted: Data[] = firstNameASCSortedRaw
const firstNameDESCSorted: Data[] = firstNameDESCSortedRaw
const firstNameASCBalanceDESCSorted: Data[] = firstNameASCBalanceDESCSortedRaw
const firstNameDESCBalanceASCSorted: Data[] = firstNameDESCBalanceASCSortedRaw

describe('multiColumnSort', () => {
  const mapValues = (arr: Data[]) => arr.map(({ firstName }) => ({ firstName }))

  it('sorts by one column ASC', () => {
    expect(
      mapValues(multiColumnSort(unsorted, [['firstName', 'ASC']]))
    ).toEqual(mapValues(firstNameASCSorted))
  })

  it('sorts by one column DESC', () => {
    expect(
      mapValues(multiColumnSort(unsorted, [['firstName', 'DESC']]))
    ).toEqual(mapValues(firstNameDESCSorted))
  })

  test('object signature', () => {
    expect(mapValues(multiColumnSort(unsorted, { firstName: 'ASC' }))).toEqual(
      mapValues(firstNameASCSorted)
    )
  })

  describe('with getColumnValue', () => {
    const mapValues = (arr: Data[]) =>
      arr.map(({ firstName, balance }) => ({ firstName, balance }))

    const getColumnValue = (column: keyof Data, value: Data[keyof Data]) => {
      switch (column) {
        case 'balance':
          return parseFloat(value.replace(/\,|\$/g, ''))
        default:
          return value
      }
    }

    it('sorts by multiple columns', () => {
      expect(
        mapValues(
          multiColumnSort(
            unsorted,
            [
              ['firstName', 'DESC'],
              ['balance', 'ASC']
            ],
            getColumnValue
          )
        )
      ).toEqual(mapValues(firstNameDESCBalanceASCSorted))

      expect(
        mapValues(
          multiColumnSort(
            unsorted,
            [
              ['firstName', 'ASC'],
              ['balance', 'DESC']
            ],
            getColumnValue
          )
        )
      ).toEqual(mapValues(firstNameASCBalanceDESCSorted))
    })

    test('object signature', () => {
      expect(
        mapValues(
          multiColumnSort(
            unsorted,
            { firstName: 'DESC', balance: 'ASC' },
            getColumnValue
          )
        )
      ).toEqual(mapValues(firstNameDESCBalanceASCSorted))
    })
  })
})