mirror of https://github.com/mongodb/mongo
89 lines
3.0 KiB
JavaScript
89 lines
3.0 KiB
JavaScript
// SERVER-10689 introduced the $switch expression. In this file, we test the error cases of the
|
|
// expression.
|
|
import "jstests/libs/query/sbe_assert_error_override.js";
|
|
|
|
import {assertErrorCode} from "jstests/aggregation/extras/utils.js";
|
|
|
|
let coll = db.switch;
|
|
coll.drop();
|
|
|
|
let pipeline = {"$project": {"output": {"$switch": "not an object"}}};
|
|
assertErrorCode(coll, pipeline, 40060, "$switch requires an object as an argument.");
|
|
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {"branches": "not an array"}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40061, "$switch requires 'branches' to be an array.");
|
|
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {"branches": ["not an object"]}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40062, "$switch requires each branch to be an object.");
|
|
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {"branches": [{}]}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40064, "$switch requires each branch have a 'case'.");
|
|
|
|
pipeline = {
|
|
"$project": {
|
|
"output": {
|
|
"$switch": {
|
|
"branches": [
|
|
{
|
|
"case": 1,
|
|
},
|
|
],
|
|
},
|
|
},
|
|
},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40065, "$switch requires each branch have a 'then'.");
|
|
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {"branches": [{"case": true, "then": false, "badKey": 1}]}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40063, "$switch found a branch with an unknown argument");
|
|
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {"notAnArgument": 1}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40067, "$switch found an unknown argument");
|
|
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {"branches": []}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40068, "$switch requires at least one branch");
|
|
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40068, "$switch requires at least one branch");
|
|
|
|
assert.commandWorked(coll.insert({x: 1}));
|
|
pipeline = {
|
|
"$project": {"output": {"$switch": {"branches": [{"case": {"$eq": ["$x", 0]}, "then": 1}]}}},
|
|
};
|
|
assertErrorCode(coll, pipeline, 40066, "$switch has no default and an input matched no case");
|
|
|
|
// This query was designed to reproduce SERVER-70190. The first branch of the $switch can be
|
|
// optimized away and the $ifNull can be optimized to 2. If the field "x" exists in the input
|
|
// document and is truthy, then the expression should return 2. Otherwise it should throw because no
|
|
// case statement matched and there is no "default" expression.
|
|
pipeline = [
|
|
{
|
|
$sortByCount: {
|
|
$switch: {
|
|
branches: [
|
|
{case: {$literal: false}, then: 1},
|
|
{case: "$x", then: {$ifNull: [2, "$y"]}},
|
|
],
|
|
},
|
|
},
|
|
},
|
|
];
|
|
assert.eq([{"_id": 2, "count": 1}], coll.aggregate(pipeline).toArray());
|
|
assert.commandWorked(coll.remove({x: 1}));
|
|
assert.commandWorked(coll.insert({z: 1}));
|
|
assertErrorCode(coll, pipeline, 40066, "$switch has no default and an input matched no case");
|