// Make sure many locations in one doc works, in the form of an array
// @tags: [
//   does_not_support_stepdowns,
//   requires_replication,
//   resource_intensive,
//   # We are choosing not to test on MacOS since it's too noisy.
//   slow_on_macos,
// ]
import {ReplSetTest} from "jstests/libs/replsettest.js";

const numLocations = 300;
let locObj = [];
// Add locations everywhere
for (let i = 0; i < 10; i++) {
    for (let j = 0; j < 10; j++) {
        if (j % 2 == 0) locObj.push([i, j]);
        else locObj.push({x: i, y: j});
    }
}

// Add docs with all these locations
let docs = [];
for (let i = 0; i < numLocations; i++) {
    docs.push({_id: i, loc: locObj});
}

const collNamePrefix = "geo_array1_";
let collCount = 0;

function test(conn, index) {
    const testDB = conn.getDB("test");
    let t = testDB.getCollection(collNamePrefix + collCount++);
    t.drop();

    if (index) {
        assert.commandWorked(t.createIndex({loc: "2d"}));
    }

    assert.commandWorked(t.insert(docs));

    // Pull them back
    for (let i = 0; i < 10; i++) {
        for (let j = 0; j < 10; j++) {
            const locations = t
                .find({
                    loc: {
                        $within: {
                            $box: [
                                [i - 0.5, j - 0.5],
                                [i + 0.5, j + 0.5],
                            ],
                        },
                    },
                })
                .toArray();
            assert.eq(
                numLocations,
                locations.length,
                "index: " + index + "; i: " + i + "; j: " + j + "; locations: " + tojson(locations),
            );
        }
    }
}

const rst = new ReplSetTest({nodes: 1});
rst.startSet();
rst.initiate();

const primary = rst.getPrimary();

test(primary, /*index=*/ true);
test(primary, /*index=*/ false);

rst.stopSet();