mongo/jstests/readonly/distinct.js

import {cycleN, runReadOnlyTest, zip2} from "jstests/readonly/lib/read_only_test.js";

runReadOnlyTest(
    (function () {
        return {
            name: "find",

            colors: ["blue", "green", "orange", "white"],
            nums: [1, 2, 3, 4, 5, 6],

            load: function (writableCollection) {
                let N = 1000;

                this.colors.sort();
                this.nums.sort();

                let bulk = writableCollection.initializeUnorderedBulkOp();

                for (let [color, num] of zip2(cycleN(this.colors, N), cycleN(this.nums, N))) {
                    bulk.insert({color, num});
                }
                assert.commandWorked(bulk.execute());
            },
            exec: function (readableCollection) {
                let distinctColors = readableCollection.distinct("color");
                let distinctNums = readableCollection.distinct("num");

                distinctColors.sort();
                distinctNums.sort();

                assert.eq(distinctColors.length, this.colors.length);
                for (var i = 0; i < distinctColors; ++i) {
                    assert.eq(distinctColors[i], this.colors[i]);
                }

                assert.eq(distinctNums.length, this.nums.length);
                for (var i = 0; i < distinctNums; ++i) {
                    assert.eq(distinctNums[i], this.nums[i]);
                }
            },
        };
    })(),
);