// Copyright 2022 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Flags: --allow-natives-syntax

function verifyGroupByResult(result) {
  assertInstanceof(result, Map);
  %HeapObjectVerify(result);
}

var array = [-0, 1, 0, 2];
var groupToMap = () => {
  let result = Map.groupBy(array, v => v > 0);
  verifyGroupByResult(result);
  result = Array.from(result.entries());
  return result;
}

// entry order matters
assertEquals(groupToMap(), [
  [false, [-0, 0]],
  [true, [1, 2]],
]);

Object.defineProperty(array, 4, {
  enumerable: true,
  configurable: true,
  writable: true,
  value: 3,
});
assertEquals(groupToMap(), [
  [false, [-0, 0]],
  [true, [1, 2, 3]],
]);

Object.defineProperty(array, 5, {
  enumerable: true,
  configurable: true,
  get: () => 4,
});
var result = groupToMap();
assertEquals(result, [
  [false, [-0, 0]],
  [true, [1, 2, 3, 4]],
]);
assertSame(result[0][1][0], -0);


// fairly large result array
var length = 20000;
var array = new Array(length);
for (var idx = 0; idx < length; idx++) {
  array[idx] = idx;
}
var groupToMap = () => {
  let result = Map.groupBy(array, v => v % 2);
  verifyGroupByResult(result);
  result = Array.from(result.entries());
  return result;
}
var result = groupToMap();
assertEquals(result, [
  [0, array.filter(v => v % 2 === 0)],
  [1, array.filter(v => v % 2 === 1)],
]);


// check #sec-array.prototype.groupbymap 6.d
var array = [-0, 0];
var result = Map.groupBy(array, v => v);
verifyGroupByResult(result);
assertEquals(result.get(0), [-0, 0]);


// check array changed by callbackfn
var array = [-0, 0, 1, 2];
var groupToMap = () => {
  let result = Map.groupBy(array, (v, idx) => {
    if (idx === 1) {
      array[2] = {a: 'b'};
    }
    return v > 0;
  });
  verifyGroupByResult(result);
  result = Array.from(result.entries());
  return result;
}

assertEquals(groupToMap(), [
  [false, [-0, 0, {a: 'b'}]],
  [true, [2]],
]);

// check array with holes
var array = [1, , 2, , 3, , 4];
var groupToMap = () => {
  let result = Map.groupBy(array, v => v % 2 === 0 ? 'even' : 'not_even');
  verifyGroupByResult(result);
  result = Array.from(result.entries());
  return result;
};
function checkNoHoles(arr) {
  for (let idx = 0; idx < arr.length; idx++) {
    assertTrue(Object.getOwnPropertyDescriptor(arr, idx) !== undefined);
  }
}
var result = groupToMap();
assertEquals(result, [
  ['not_even', [1, undefined, undefined, 3, undefined]],
  ['even', [2, 4]],
]);
checkNoHoles(result[0][1]);
checkNoHoles(result[1][1]);

var array = [1, undefined, 2, undefined, 3, undefined, 4];
result = groupToMap();
assertEquals(result, [
  ['not_even', [1, undefined, undefined, 3, undefined]],
  ['even', [2, 4]],
]);
checkNoHoles(result[0][1]);
checkNoHoles(result[1][1]);

// array like objects
var arrayLikeObjects = [
  (function* f(){
    yield -1;
    yield 1;
    yield 2;
  })(),
  (function () { return arguments })(-1, 1, 2),
  Int8Array.from([-1, 1, 2]),
  Float32Array.from([-1, 1, 2]),
];
var groupToMap = () => {
  let result = Map.groupBy(array, v => v > 0);
  verifyGroupByResult(result);
  result = Array.from(result.entries());
  return result;
};
for (var array of arrayLikeObjects) {
  assertEquals(groupToMap(), [
    [false, [-1]],
    [true, [1, 2]],
  ]);
}

// check proto elements
var array = [,];
var groupToMap = () => {
  let result = Map.groupBy(array, v => v);
  verifyGroupByResult(result);
  result = Array.from(result.entries());
  return result;
}

assertEquals(groupToMap(), [
  [undefined, [undefined]],
]);
array.__proto__.push(6);
assertEquals(groupToMap(), [
  [6, [6]],
]);


// callbackfn throws
var array = [-0, 1, 0, 2];
assertThrows(
  () => Map.groupBy(array, () => { throw new Error('foobar'); }),
  Error,
  'foobar'
);


// callbackfn is not callable
var array = [-0, 1, 0, 2];
assertThrows(
  () => Map.groupBy(array, 'foobar'),
  TypeError,
);

// Lots of groups to hit grow path in the intermediate OrderedHashMap
Map.groupBy("Strings are iterable, actually,", (x) => x);

// Large group.
Map.groupBy(new Int8Array(65536), function() {});