lodash/test/nthArg.spec.js

import lodashStable from 'lodash';
import { args, falsey, stubA, stubB, noop } from './utils';
import nthArg from '../src/nthArg';

describe('nthArg', () => {
    const args = ['a', 'b', 'c', 'd'];

    it('should create a function that returns its nth argument', () => {
        const actual = lodashStable.map(args, (value, index) => {
            const func = nthArg(index);
            return func.apply(undefined, args);
        });

        expect(actual).toEqual(args);
    });

    it('should work with a negative `n`', () => {
        const actual = lodashStable.map(lodashStable.range(1, args.length + 1), (n) => {
            const func = nthArg(-n);
            return func.apply(undefined, args);
        });

        expect(actual).toEqual(['d', 'c', 'b', 'a']);
    });

    it('should coerce `n` to an integer', () => {
        let values = falsey;
        let expected = lodashStable.map(values, stubA);

        let actual = lodashStable.map(values, (n) => {
            const func = n ? nthArg(n) : nthArg();
            return func.apply(undefined, args);
        });

        expect(actual).toEqual(expected);

        values = ['1', 1.6];
        expected = lodashStable.map(values, stubB);

        actual = lodashStable.map(values, (n) => {
            const func = nthArg(n);
            return func.apply(undefined, args);
        });

        expect(actual).toEqual(expected);
    });

    it('should return `undefined` for empty arrays', () => {
        const func = nthArg(1);
        expect(func()).toBe(undefined);
    });

    it('should return `undefined` for non-indexes', () => {
        const values = [Infinity, args.length];
        const expected = lodashStable.map(values, noop);

        const actual = lodashStable.map(values, (n) => {
            const func = nthArg(n);
            return func.apply(undefined, args);
        });

        expect(actual).toEqual(expected);
    });
});