Files
mongo/jstests/aggregation/expressions/objectToArray.js
Nick Zolnierz 3e0ff89ff1 SERVER-94971 Add query ownership for files under jstests/libs (#27375)
GitOrigin-RevId: 36330925597c2fca287c2d73bcb6743026f316b4
2024-10-02 11:50:41 +00:00

101 lines
3.7 KiB
JavaScript

// Tests for the $objectToArray aggregation expression.
import "jstests/libs/query/sbe_assert_error_override.js";
import {assertErrorCode} from "jstests/aggregation/extras/utils.js";
let coll = db.object_to_array_expr;
coll.drop();
let object_to_array_expr = {$project: {expanded: {$objectToArray: "$subDoc"}}};
// $objectToArray correctly converts a document to an array of key-value pairs.
assert.commandWorked(coll.insert({_id: 0, subDoc: {"a": 1, "b": 2, "c": "foo"}}));
let result = coll.aggregate([object_to_array_expr]).toArray();
assert.eq(result,
[{_id: 0, expanded: [{"k": "a", "v": 1}, {"k": "b", "v": 2}, {"k": "c", "v": "foo"}]}]);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 1, subDoc: {"y": []}}));
result = coll.aggregate([object_to_array_expr]).toArray();
assert.eq(result, [{_id: 1, expanded: [{"k": "y", "v": []}]}]);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 2, subDoc: {"a": 1, "b": {"d": "string"}, "c": [1, 2]}}));
result = coll.aggregate([object_to_array_expr]).toArray();
assert.eq(
result, [{
_id: 2,
expanded: [{"k": "a", "v": 1}, {"k": "b", "v": {"d": "string"}}, {"k": "c", "v": [1, 2]}]
}]);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 3, subDoc: {}}));
result = coll.aggregate([object_to_array_expr]).toArray();
assert.eq(result, [{_id: 3, expanded: []}]);
// Turns to array from the root of the document.
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 4, "a": 1, "b": 2, "c": 3}));
result = coll.aggregate([{$project: {document: {$objectToArray: "$$ROOT"}}}]).toArray();
assert.eq(
result, [{
_id: 4,
document: [{"k": "_id", "v": 4}, {"k": "a", "v": 1}, {"k": "b", "v": 2}, {"k": "c", "v": 3}]
}]);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 5, "date": ISODate("2017-01-24T00:00:00")}));
result =
coll.aggregate([{$project: {document: {$objectToArray: {dayOfWeek: {$dayOfWeek: "$date"}}}}}])
.toArray();
assert.eq(result, [{_id: 5, document: [{"k": "dayOfWeek", "v": 3}]}]);
// $objectToArray errors on non-document types.
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 6, subDoc: "string"}));
assertErrorCode(coll, [object_to_array_expr], 40390);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 7, subDoc: ObjectId()}));
assertErrorCode(coll, [object_to_array_expr], 40390);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 8, subDoc: NumberLong(0)}));
assertErrorCode(coll, [object_to_array_expr], 40390);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 9, subDoc: []}));
assertErrorCode(coll, [object_to_array_expr], 40390);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 10, subDoc: [0]}));
assertErrorCode(coll, [object_to_array_expr], 40390);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 11, subDoc: ["string"]}));
assertErrorCode(coll, [object_to_array_expr], 40390);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 12, subDoc: [{"a": "b"}]}));
assertErrorCode(coll, [object_to_array_expr], 40390);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 13, subDoc: NaN}));
assertErrorCode(coll, [object_to_array_expr], 40390);
// $objectToArray outputs null on null-ish types.
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 14, subDoc: null}));
result = coll.aggregate([object_to_array_expr]).toArray();
assert.eq(result, [{_id: 14, expanded: null}]);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 15, subDoc: undefined}));
result = coll.aggregate([object_to_array_expr]).toArray();
assert.eq(result, [{_id: 15, expanded: null}]);
assert(coll.drop());
assert.commandWorked(coll.insert({_id: 16}));
result = coll.aggregate([object_to_array_expr]).toArray();
assert.eq(result, [{_id: 16, expanded: null}]);