core-js/tests/unit-global/es.iterator.take.js

import { STRICT } from '../helpers/constants.js';
import { createIterator } from '../helpers/helpers.js';

QUnit.test('Iterator#take', assert => {
  const { take } = Iterator.prototype;

  assert.isFunction(take);
  assert.arity(take, 1);
  assert.name(take, 'take');
  assert.looksNative(take);
  assert.nonEnumerable(Iterator.prototype, 'take');

  assert.arrayEqual(take.call(createIterator([1, 2, 3]), 2).toArray(), [1, 2], 'basic functionality');
  assert.arrayEqual(take.call(createIterator([1, 2, 3]), 1.5).toArray(), [1], 'float');
  assert.arrayEqual(take.call(createIterator([1, 2, 3]), 4).toArray(), [1, 2, 3], 'big');
  assert.arrayEqual(take.call(createIterator([1, 2, 3]), 0).toArray(), [], 'zero');
  assert.arrayEqual(take.call(createIterator([1, 2, 3]), Infinity).toArray(), [1, 2, 3], 'Infinity');

  if (STRICT) {
    assert.throws(() => take.call(undefined, 1), TypeError);
    assert.throws(() => take.call(null, 1), TypeError);
  }

  assert.throws(() => take.call({}, 1).next(), TypeError);
  assert.throws(() => take.call([], 1).next(), TypeError);
  assert.throws(() => take.call(createIterator([1, 2, 3]), -1), RangeError, 'negative');
  const it = createIterator([1], { return() { this.closed = true; } });
  assert.throws(() => take.call(it, NaN), RangeError, 'NaN');
  assert.throws(() => take.call(it, -Infinity), RangeError, '-Infinity');
  assert.throws(() => take.call(it, 0x20000000000000), RangeError, 'unsafe integer');
  assert.true(it.closed, 'take closes iterator on validation error');
  // https://issues.chromium.org/issues/336839115
  assert.throws(() => take.call({ next: null }, 1).next(), TypeError);
});