update: refuse a positional path instead of destroying the array
`{$set: {"y.$[i].b": 2}}` did not fail to update the array. It replaced
`y: [{b: 3}, {b: 1}]` with `y: {"$[i]": {"b": 2}}` -- every element
discarded -- and answered ok: 1, modifiedCount: 1. Remotely reachable by any
client issuing an ordinary MongoDB update.
`arrayFilters` was not implicated: the string appears nowhere in src/, the
option is accepted off the wire and dropped. The destruction was in the
path, at `set_path`'s "treat as non-array: replace with a doc" branch, so it
fired for all three spellings of "descend into this array" -- `$`, `$[]` and
`$[<ident>]` -- under every operator. `$inc` through `$[i]` stored its
operand rather than incrementing.
Two refusals, because the branch held two different mistakes:
- a positional segment is refused up front, before anything is applied, so
an update naming a good path and a positional one lands neither. Its
code is BadValue (2), which is what mongod answers for every positional
path failure.
- a plain non-numeric segment under an array -- `y.nope.b`, `y.$x.b` -- is
PathNotViable (28), measured. It is never a field to create, which is
the opposite of what `set_path` does for a missing *document* field and
the reason this branch existed at all.
Numeric segments are untouched, including the null padding past the end,
which a test now pins.
Messages are this server's own words. mongod's PathNotViable text embeds a
shell-syntax rendering of the offending element (`Cannot create field 'nope'
in element {y: [ { b: 3 }, { b: 1 } ]}`) and no BSON formatter here produces
it. The code is what the corpus asserts and the code is exact; a half-copy
of the text would be worse than a clear sentence that does not pretend.
Re-running the 17-case probe against both servers: 16 of 17 diverged before,
0 are destructive now, and 5 agree with mongod's code exactly -- every case
where mongod also refuses. The rest refuse where mongod succeeds, which is
the honest not-implemented state and is what the design review chose.
Scorecard unchanged at 204 pass / 87 fail: the 14 arrayFilters cases still
fail, now reporting the refusal rather than a corrupted document. That was
the gate this review picked -- `docs/M3_ARRAYFILTERS_DESIGN_REVIEW.md` §5,
option D -- because the corpus has no `$[]` case and no bare `$` case at
all, so passing it would have certified two live ways to destroy an array.
206/206 unit (8 new) in ReleaseFast and ReleaseSafe, 83/83 fuzz, aggregation
corpus 70/0, full e2e matrix and crash-fuzz green. Both refusals are
mutation-checked: dropping the up-front scan reddens four tests on the
array's contents rather than on the error, and restoring the destructive
branch reddens the PathNotViable test.
This commit is contained in:
301
src/update.zig
301
src/update.zig
@@ -6,7 +6,42 @@ const std = @import("std");
|
||||
const bson = @import("bson.zig");
|
||||
const query = @import("query.zig");
|
||||
|
||||
pub const UpdateError = error{ ImmutableId, InvalidUpdate, OutOfMemory };
|
||||
pub const UpdateError = error{
|
||||
ImmutableId,
|
||||
InvalidUpdate,
|
||||
/// A `$`, `$[]` or `$[<identifier>]` segment: MongoDB's three ways of
|
||||
/// saying "descend into this array", none of them implemented here.
|
||||
PositionalUnsupported,
|
||||
/// A non-numeric segment applied to an array, which is never a field to
|
||||
/// create. mongod's `PathNotViable`.
|
||||
PathNotViable,
|
||||
OutOfMemory,
|
||||
};
|
||||
|
||||
/// Which path a refusal was about, so the reply can name it instead of saying
|
||||
/// "bad update". Borrowed from the update document, which outlives the call.
|
||||
pub const Diagnostic = struct {
|
||||
path: []const u8 = "",
|
||||
segment: []const u8 = "",
|
||||
};
|
||||
|
||||
fn note(diag: ?*Diagnostic, path: []const u8, segment: []const u8) void {
|
||||
if (diag) |d| d.* = .{ .path = path, .segment = segment };
|
||||
}
|
||||
|
||||
/// `$`, `$[]` and `$[<identifier>]` -- the three spellings of "descend into
|
||||
/// this array".
|
||||
///
|
||||
/// None is implemented, and until they are each has to be refused rather than
|
||||
/// walked. `set_path` used to reach the array, fail to read the segment as an
|
||||
/// index, and overwrite the array with a document keyed by the segment's
|
||||
/// literal text: `{$set: {"y.$[i].b": 2}}` turned `y: [{b: 3}, {b: 1}]` into
|
||||
/// `y: {"$[i]": {"b": 2}}` and answered ok: 1. Every element was discarded,
|
||||
/// under every operator -- `$inc` stored its operand rather than incrementing.
|
||||
fn is_positional(seg: []const u8) bool {
|
||||
if (std.mem.eql(u8, seg, "$")) return true;
|
||||
return seg.len >= 3 and std.mem.startsWith(u8, seg, "$[") and seg[seg.len - 1] == ']';
|
||||
}
|
||||
|
||||
const max_path_segments = 16;
|
||||
|
||||
@@ -28,8 +63,16 @@ fn is_operator_key(key: []const u8) bool {
|
||||
}
|
||||
|
||||
/// Apply an update document to `doc`: a replacement, or a set of operators.
|
||||
pub fn apply(doc: *bson.Document, update: *const bson.Document) UpdateError!void {
|
||||
pub fn apply(
|
||||
doc: *bson.Document,
|
||||
update: *const bson.Document,
|
||||
diag: ?*Diagnostic,
|
||||
) UpdateError!void {
|
||||
// A replacement carries data, not paths, so nothing below applies to it --
|
||||
// which is also why mongod ignores `arrayFilters` alongside one, the single
|
||||
// case out of seventeen where this server already agreed with it.
|
||||
if (is_replacement(update.pairs)) return apply_replacement(doc, update.pairs);
|
||||
try reject_positional(update.pairs, diag);
|
||||
const arena = doc.arena.allocator();
|
||||
var pairs = try copy_to_list(bson.Pair, arena, doc.pairs);
|
||||
for (update.pairs) |op| {
|
||||
@@ -37,11 +80,40 @@ pub fn apply(doc: *bson.Document, update: *const bson.Document) UpdateError!void
|
||||
// is not one is a mixed document -- which MongoDB rejects rather than
|
||||
// guessing at.
|
||||
if (!is_operator_key(op.key)) return error.InvalidUpdate;
|
||||
try apply_operator(arena, &pairs, op.key, op.value);
|
||||
try apply_operator(arena, &pairs, op.key, op.value, diag);
|
||||
}
|
||||
doc.pairs = try pairs.toOwnedSlice(arena);
|
||||
}
|
||||
|
||||
/// Refuse every positional path in the update before any of it is applied.
|
||||
///
|
||||
/// Up front rather than at the point of use, because one update names several
|
||||
/// paths: checking as we walk would refuse the third path having already
|
||||
/// rewritten what the first two named. The caller discards its copy on error
|
||||
/// either way, so this is not what makes the refusal safe -- it is what makes
|
||||
/// the refusal *about the update* rather than about however far a walk got.
|
||||
fn reject_positional(update: []const bson.Pair, diag: ?*Diagnostic) UpdateError!void {
|
||||
for (update) |op| {
|
||||
const ops = doc_pairs(op.value) orelse continue;
|
||||
for (ops) |p| {
|
||||
try reject_positional_path(p.key, diag);
|
||||
// `$rename`'s destination is a path too, and it is the *value*.
|
||||
if (std.mem.eql(u8, op.key, "$rename") and p.value == .string) {
|
||||
try reject_positional_path(p.value.string, diag);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn reject_positional_path(path: []const u8, diag: ?*Diagnostic) UpdateError!void {
|
||||
var it = std.mem.splitScalar(u8, path, '.');
|
||||
while (it.next()) |seg| {
|
||||
if (!is_positional(seg)) continue;
|
||||
note(diag, path, seg);
|
||||
return error.PositionalUnsupported;
|
||||
}
|
||||
}
|
||||
|
||||
/// Replace every field of `doc` with `replacement`'s, except `_id`.
|
||||
///
|
||||
/// `_id` is immutable, so it survives and keeps its position at the front (which
|
||||
@@ -97,6 +169,7 @@ fn apply_operator(
|
||||
pairs: *std.ArrayListUnmanaged(bson.Pair),
|
||||
op: []const u8,
|
||||
value: bson.Value,
|
||||
diag: ?*Diagnostic,
|
||||
) UpdateError!void {
|
||||
if (std.mem.eql(u8, op, "$set")) {
|
||||
const ops = doc_pairs(value) orelse return error.InvalidUpdate;
|
||||
@@ -104,7 +177,7 @@ fn apply_operator(
|
||||
if (std.mem.eql(u8, p.key, "_id")) return error.ImmutableId;
|
||||
var segs: [max_path_segments][]const u8 = undefined;
|
||||
const n = split_path(p.key, &segs) orelse return error.InvalidUpdate;
|
||||
try set_path(arena, pairs, segs[0..n], try bson.copy_value(arena, p.value));
|
||||
try set_path(arena, pairs, segs[0..n], try bson.copy_value(arena, p.value), p.key, diag);
|
||||
}
|
||||
return;
|
||||
}
|
||||
@@ -125,7 +198,7 @@ fn apply_operator(
|
||||
const current = get_value(pairs.items, segs[0..n]) orelse bson.Value{ .int32 = 0 };
|
||||
if (!current.is_number() or !p.value.is_number()) return error.InvalidUpdate;
|
||||
const sum = try numeric_add(current, p.value);
|
||||
try set_path(arena, pairs, segs[0..n], sum);
|
||||
try set_path(arena, pairs, segs[0..n], sum, p.key, diag);
|
||||
}
|
||||
return;
|
||||
}
|
||||
@@ -151,12 +224,12 @@ fn apply_operator(
|
||||
else => return error.InvalidUpdate,
|
||||
};
|
||||
for (arr) |item| try items.append(arena, try bson.copy_value(arena, item));
|
||||
try set_path(arena, pairs, segs[0..n], .{ .array = try items.toOwnedSlice(arena) });
|
||||
try set_path(arena, pairs, segs[0..n], .{ .array = try items.toOwnedSlice(arena) }, p.key, diag);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
try items.append(arena, try bson.copy_value(arena, p.value));
|
||||
try set_path(arena, pairs, segs[0..n], .{ .array = try items.toOwnedSlice(arena) });
|
||||
try set_path(arena, pairs, segs[0..n], .{ .array = try items.toOwnedSlice(arena) }, p.key, diag);
|
||||
}
|
||||
return;
|
||||
}
|
||||
@@ -177,7 +250,7 @@ fn apply_operator(
|
||||
try items.append(arena, elem);
|
||||
}
|
||||
}
|
||||
try set_path(arena, pairs, segs[0..n], .{ .array = try items.toOwnedSlice(arena) });
|
||||
try set_path(arena, pairs, segs[0..n], .{ .array = try items.toOwnedSlice(arena) }, p.key, diag);
|
||||
}
|
||||
return;
|
||||
}
|
||||
@@ -192,7 +265,7 @@ fn apply_operator(
|
||||
unset_path(arena, pairs, old_segs[0..old_n]);
|
||||
var new_segs: [max_path_segments][]const u8 = undefined;
|
||||
const new_n = split_path(p.value.string, &new_segs) orelse return error.InvalidUpdate;
|
||||
try set_path(arena, pairs, new_segs[0..new_n], v);
|
||||
try set_path(arena, pairs, new_segs[0..new_n], v, p.value.string, diag);
|
||||
}
|
||||
return;
|
||||
}
|
||||
@@ -258,6 +331,8 @@ fn set_path(
|
||||
pairs: *std.ArrayListUnmanaged(bson.Pair),
|
||||
segs: []const []const u8,
|
||||
value: bson.Value,
|
||||
path: []const u8,
|
||||
diag: ?*Diagnostic,
|
||||
) UpdateError!void {
|
||||
if (segs.len == 1) {
|
||||
if (find_pair(pairs.items, segs[0])) |idx| {
|
||||
@@ -270,23 +345,31 @@ fn set_path(
|
||||
const idx = find_pair(pairs.items, segs[0]) orelse {
|
||||
const is_array = parse_index(segs[1]) != null;
|
||||
try pairs.append(arena, .{ .key = try arena.dupe(u8, segs[0]), .value = if (is_array) .{ .array = &.{} } else .{ .doc = &.{} } });
|
||||
return set_path(arena, pairs, segs, value);
|
||||
return set_path(arena, pairs, segs, value, path, diag);
|
||||
};
|
||||
switch (pairs.items[idx].value) {
|
||||
.doc => |sub| {
|
||||
var sub_pairs = try copy_to_list(bson.Pair, arena, sub);
|
||||
defer sub_pairs.deinit(arena);
|
||||
try set_path(arena, &sub_pairs, segs[1..], value);
|
||||
try set_path(arena, &sub_pairs, segs[1..], value, path, diag);
|
||||
pairs.items[idx].value = .{ .doc = try sub_pairs.toOwnedSlice(arena) };
|
||||
},
|
||||
.array => |arr| {
|
||||
const index = parse_index(segs[1]) orelse {
|
||||
// treat as non-array: replace with a doc
|
||||
var sub_pairs: std.ArrayListUnmanaged(bson.Pair) = .empty;
|
||||
defer sub_pairs.deinit(arena);
|
||||
try set_path(arena, &sub_pairs, segs[1..], value);
|
||||
pairs.items[idx].value = .{ .doc = try sub_pairs.toOwnedSlice(arena) };
|
||||
return;
|
||||
// A non-numeric segment under an array is never a field to
|
||||
// create. This branch used to read "treat as non-array:
|
||||
// replace with a doc" and did exactly that -- `y.nope.b`
|
||||
// turned `y: [{b: 3}]` into `y: {nope: {b: 2}}`, discarding
|
||||
// every element and answering ok: 1. mongod refuses with
|
||||
// PathNotViable and leaves the document alone.
|
||||
//
|
||||
// The positional spellings took this same branch and are the
|
||||
// reason it was found; they are refused earlier, by
|
||||
// `reject_positional`. What reaches here is the rest of the
|
||||
// class: a plain field name, and a `$`-prefixed one that is
|
||||
// not positional.
|
||||
note(diag, path, segs[1]);
|
||||
return error.PathNotViable;
|
||||
};
|
||||
var items = try copy_to_list(bson.Value, arena, arr);
|
||||
defer items.deinit(arena);
|
||||
@@ -300,13 +383,13 @@ fn set_path(
|
||||
.doc => |sub| {
|
||||
var sub_pairs = try copy_to_list(bson.Pair, arena, sub);
|
||||
defer sub_pairs.deinit(arena);
|
||||
try set_path(arena, &sub_pairs, segs[2..], value);
|
||||
try set_path(arena, &sub_pairs, segs[2..], value, path, diag);
|
||||
items.items[index] = .{ .doc = try sub_pairs.toOwnedSlice(arena) };
|
||||
},
|
||||
else => {
|
||||
var sub_pairs: std.ArrayListUnmanaged(bson.Pair) = .empty;
|
||||
defer sub_pairs.deinit(arena);
|
||||
try set_path(arena, &sub_pairs, segs[2..], value);
|
||||
try set_path(arena, &sub_pairs, segs[2..], value, path, diag);
|
||||
items.items[index] = .{ .doc = try sub_pairs.toOwnedSlice(arena) };
|
||||
},
|
||||
}
|
||||
@@ -316,7 +399,7 @@ fn set_path(
|
||||
else => {
|
||||
var sub_pairs: std.ArrayListUnmanaged(bson.Pair) = .empty;
|
||||
defer sub_pairs.deinit(arena);
|
||||
try set_path(arena, &sub_pairs, segs[1..], value);
|
||||
try set_path(arena, &sub_pairs, segs[1..], value, path, diag);
|
||||
pairs.items[idx].value = .{ .doc = try sub_pairs.toOwnedSlice(arena) };
|
||||
},
|
||||
}
|
||||
@@ -425,7 +508,7 @@ test "$set, $inc, $unset, $rename" {
|
||||
.{ .key = "$inc", .value = .{ .doc = &.{.{ .key = "user.age", .value = .{ .int32 = 2 } }} } },
|
||||
.{ .key = "$unset", .value = .{ .doc = &.{.{ .key = "gone", .value = .{ .string = "" } }} } },
|
||||
.{ .key = "$rename", .value = .{ .doc = &.{.{ .key = "new", .value = .{ .string = "renamed" } }} } },
|
||||
}));
|
||||
}), null);
|
||||
|
||||
const user = bson.get_pair(doc.pairs, "user").?;
|
||||
try testing.expectEqualStrings("alice", user.doc[0].value.string);
|
||||
@@ -445,13 +528,13 @@ test "$push and $pull" {
|
||||
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$push", .value = .{ .doc = &.{.{ .key = "tags", .value = .{ .string = "c" } }} } },
|
||||
}));
|
||||
}), null);
|
||||
try testing.expectEqual(@as(usize, 3), bson.get_pair(doc.pairs, "tags").?.array.len);
|
||||
try testing.expectEqualStrings("c", bson.get_pair(doc.pairs, "tags").?.array[2].string);
|
||||
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$pull", .value = .{ .doc = &.{.{ .key = "tags", .value = .{ .string = "b" } }} } },
|
||||
}));
|
||||
}), null);
|
||||
const tags = bson.get_pair(doc.pairs, "tags").?.array;
|
||||
try testing.expectEqual(@as(usize, 2), tags.len);
|
||||
try testing.expectEqualStrings("a", tags[0].string);
|
||||
@@ -459,7 +542,7 @@ test "$push and $pull" {
|
||||
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$push", .value = .{ .doc = &.{.{ .key = "tags", .value = .{ .doc = &.{.{ .key = "$each", .value = .{ .array = &.{ .{ .string = "x" }, .{ .string = "y" } } } }} } }} } },
|
||||
}));
|
||||
}), null);
|
||||
try testing.expectEqual(@as(usize, 4), bson.get_pair(doc.pairs, "tags").?.array.len);
|
||||
}
|
||||
|
||||
@@ -475,7 +558,7 @@ test "$set nested creation and _id protection" {
|
||||
.{ .key = "a.b.c", .value = .{ .int32 = 42 } },
|
||||
.{ .key = "arr.1", .value = .{ .string = "x" } },
|
||||
} } },
|
||||
}));
|
||||
}), null);
|
||||
const a = bson.get_pair(doc.pairs, "a").?;
|
||||
const b = bson.get_pair(a.doc, "b").?;
|
||||
try testing.expectEqual(@as(i64, 42), bson.get_pair(b.doc, "c").?.int32);
|
||||
@@ -485,7 +568,7 @@ test "$set nested creation and _id protection" {
|
||||
|
||||
try testing.expectError(error.ImmutableId, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$set", .value = .{ .doc = &.{.{ .key = "_id", .value = .{ .int32 = 2 } }} } },
|
||||
})));
|
||||
}), null));
|
||||
}
|
||||
|
||||
test "a replacement keeps _id and drops every other field" {
|
||||
@@ -502,7 +585,7 @@ test "a replacement keeps _id and drops every other field" {
|
||||
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "fresh", .value = .{ .int32 = 42 } },
|
||||
}));
|
||||
}), null);
|
||||
|
||||
try testing.expectEqual(@as(usize, 2), doc.pairs.len);
|
||||
// _id survives, and stays at the front where it is stored.
|
||||
@@ -523,7 +606,7 @@ test "an empty replacement leaves a document holding only its _id" {
|
||||
|
||||
// Legal, and the reason `is_replacement` treats an empty document as one
|
||||
// rather than as a no-op set of operators.
|
||||
try apply(&doc, &doc_of(&.{}));
|
||||
try apply(&doc, &doc_of(&.{}), null);
|
||||
try testing.expectEqual(@as(usize, 1), doc.pairs.len);
|
||||
try testing.expectEqualStrings("_id", doc.pairs[0].key);
|
||||
}
|
||||
@@ -545,7 +628,7 @@ test "a replacement may repeat the _id it is replacing, but not change it" {
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "_id", .value = .{ .int32 = 7 } },
|
||||
.{ .key = "b", .value = .{ .int32 = 2 } },
|
||||
}));
|
||||
}), null);
|
||||
try testing.expectEqual(@as(i64, 7), doc.pairs[0].value.int32);
|
||||
try testing.expectEqual(@as(i64, 2), bson.get_pair(doc.pairs, "b").?.int32);
|
||||
// And only once, not twice.
|
||||
@@ -554,14 +637,14 @@ test "a replacement may repeat the _id it is replacing, but not change it" {
|
||||
// A different _id: refused.
|
||||
try testing.expectError(error.ImmutableId, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "_id", .value = .{ .int32 = 8 } },
|
||||
})));
|
||||
}), null));
|
||||
|
||||
// Equal across numeric types is the same _id, matching the canonical key
|
||||
// encoding the _id_ index descends on.
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "_id", .value = .{ .double = 7.0 } },
|
||||
.{ .key = "c", .value = .{ .int32 = 3 } },
|
||||
}));
|
||||
}), null);
|
||||
try testing.expectEqual(@as(i64, 3), bson.get_pair(doc.pairs, "c").?.int32);
|
||||
}
|
||||
|
||||
@@ -578,7 +661,7 @@ test "a replacement supplies the _id when the document has none" {
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "a", .value = .{ .int32 = 1 } },
|
||||
.{ .key = "_id", .value = .{ .int32 = 99 } },
|
||||
}));
|
||||
}), null);
|
||||
|
||||
try testing.expectEqual(@as(usize, 2), doc.pairs.len);
|
||||
try testing.expectEqualStrings("_id", doc.pairs[0].key);
|
||||
@@ -596,13 +679,13 @@ test "a mixed update document is refused from either side" {
|
||||
try testing.expectError(error.InvalidUpdate, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$set", .value = .{ .doc = &.{.{ .key = "a", .value = .{ .int32 = 1 } }} } },
|
||||
.{ .key = "plain", .value = .{ .int32 = 1 } },
|
||||
})));
|
||||
}), null));
|
||||
|
||||
// Starts with data, so it is a replacement and an operator has no meaning.
|
||||
try testing.expectError(error.InvalidUpdate, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "plain", .value = .{ .int32 = 1 } },
|
||||
.{ .key = "$set", .value = .{ .doc = &.{.{ .key = "a", .value = .{ .int32 = 1 } }} } },
|
||||
})));
|
||||
}), null));
|
||||
}
|
||||
|
||||
test "is_replacement decides on the first field only" {
|
||||
@@ -613,3 +696,153 @@ test "is_replacement decides on the first field only" {
|
||||
// replacement path then stores it, which is what MongoDB does with it.
|
||||
try testing.expect(is_replacement(&.{.{ .key = "", .value = .{ .int32 = 1 } }}));
|
||||
}
|
||||
|
||||
/// A document with one array field, rebuilt per case so a refusal can be
|
||||
/// checked against untouched bytes.
|
||||
fn array_doc(arena: std.mem.Allocator) !bson.Document {
|
||||
var doc = bson.Document{ .arena = std.heap.ArenaAllocator.init(arena), .pairs = &.{} };
|
||||
doc.pairs = try doc.arena.allocator().dupe(bson.Pair, &.{
|
||||
.{ .key = "y", .value = .{ .array = &.{
|
||||
.{ .doc = &.{.{ .key = "b", .value = .{ .int32 = 3 } }} },
|
||||
.{ .doc = &.{.{ .key = "b", .value = .{ .int32 = 1 } }} },
|
||||
} } },
|
||||
});
|
||||
return doc;
|
||||
}
|
||||
|
||||
fn expect_y_untouched(doc: *const bson.Document) !void {
|
||||
const y = doc.get("y") orelse return error.TestUnexpectedResult;
|
||||
// The array is still an array. Before this refusal existed it was a
|
||||
// document keyed by the path segment's literal text, and everything in it
|
||||
// was gone.
|
||||
try testing.expect(y == .array);
|
||||
try testing.expectEqual(@as(usize, 2), y.array.len);
|
||||
try testing.expectEqual(@as(i32, 3), y.array[0].doc[0].value.int32);
|
||||
try testing.expectEqual(@as(i32, 1), y.array[1].doc[0].value.int32);
|
||||
}
|
||||
|
||||
test "a positional path is refused and the array survives" {
|
||||
// The load-bearing test of the refusal. Each of these used to answer
|
||||
// success having replaced `y` with `{"<segment>": ...}` -- the array and
|
||||
// both its elements discarded, `ok: 1`, `modifiedCount: 1`.
|
||||
//
|
||||
// Mutation check: delete the `reject_positional` call in `apply` and every
|
||||
// case here goes red on `expect_y_untouched`, not on the error.
|
||||
const cases = [_][]const u8{
|
||||
"y.$[i].b", // filtered positional
|
||||
"y.$[].b", // all-positional
|
||||
"y.$.b", // positional
|
||||
"y.$[i]", // as the leaf
|
||||
"$", // in first position
|
||||
"y.$[i].c.$[j].d", // nested, two identifiers
|
||||
};
|
||||
for (cases) |path| {
|
||||
var doc = try array_doc(testing.allocator);
|
||||
defer doc.arena.deinit();
|
||||
var diag: Diagnostic = .{};
|
||||
try testing.expectError(error.PositionalUnsupported, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$set", .value = .{ .doc = &.{.{ .key = path, .value = .{ .int32 = 2 } }} } },
|
||||
}), &diag));
|
||||
try testing.expectEqualStrings(path, diag.path);
|
||||
try expect_y_untouched(&doc);
|
||||
}
|
||||
}
|
||||
|
||||
test "every operator refuses a positional path, not just $set" {
|
||||
// The destruction was below the operator, in the shared path walk: `$inc`
|
||||
// through `$[i]` stored its operand instead of incrementing. So the
|
||||
// refusal has to be below the operator too.
|
||||
const ops = [_]struct { op: []const u8, value: bson.Value }{
|
||||
.{ .op = "$set", .value = .{ .int32 = 2 } },
|
||||
.{ .op = "$inc", .value = .{ .int32 = 10 } },
|
||||
.{ .op = "$unset", .value = .{ .string = "" } },
|
||||
.{ .op = "$push", .value = .{ .int32 = 1 } },
|
||||
.{ .op = "$pull", .value = .{ .int32 = 1 } },
|
||||
};
|
||||
for (ops) |o| {
|
||||
var doc = try array_doc(testing.allocator);
|
||||
defer doc.arena.deinit();
|
||||
try testing.expectError(error.PositionalUnsupported, apply(&doc, &doc_of(&.{
|
||||
.{ .key = o.op, .value = .{ .doc = &.{.{ .key = "y.$[i].b", .value = o.value }} } },
|
||||
}), null));
|
||||
try expect_y_untouched(&doc);
|
||||
}
|
||||
}
|
||||
|
||||
test "$rename checks its destination, which is the value not the key" {
|
||||
var doc = try array_doc(testing.allocator);
|
||||
defer doc.arena.deinit();
|
||||
var diag: Diagnostic = .{};
|
||||
try testing.expectError(error.PositionalUnsupported, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$rename", .value = .{ .doc = &.{
|
||||
.{ .key = "y", .value = .{ .string = "z.$[i]" } },
|
||||
} } },
|
||||
}), &diag));
|
||||
try testing.expectEqualStrings("z.$[i]", diag.path);
|
||||
try expect_y_untouched(&doc);
|
||||
}
|
||||
|
||||
test "nothing in the update is applied when one of its paths is refused" {
|
||||
// The refusal is taken before the first write, so the good path in this
|
||||
// update does not land either. Mutation check: move `reject_positional`
|
||||
// inside the operator loop and `ok` appears on the document.
|
||||
var doc = try array_doc(testing.allocator);
|
||||
defer doc.arena.deinit();
|
||||
try testing.expectError(error.PositionalUnsupported, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$set", .value = .{ .doc = &.{
|
||||
.{ .key = "ok", .value = .{ .int32 = 1 } },
|
||||
.{ .key = "y.$[i].b", .value = .{ .int32 = 2 } },
|
||||
} } },
|
||||
}), null));
|
||||
try testing.expect(doc.get("ok") == null);
|
||||
try expect_y_untouched(&doc);
|
||||
}
|
||||
|
||||
test "a non-numeric segment under an array is PathNotViable, not a new field" {
|
||||
// The rest of the class the positional forms belonged to. `y.nope.b` is
|
||||
// not a positional operator and is refused for a different reason with a
|
||||
// different code -- measured on mongod 8.3.7, which answers 28 here and 2
|
||||
// for the positional forms.
|
||||
const cases = [_][]const u8{ "y.nope.b", "y.nope", "y.$x.b" };
|
||||
for (cases) |path| {
|
||||
var doc = try array_doc(testing.allocator);
|
||||
defer doc.arena.deinit();
|
||||
var diag: Diagnostic = .{};
|
||||
try testing.expectError(error.PathNotViable, apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$set", .value = .{ .doc = &.{.{ .key = path, .value = .{ .int32 = 2 } }} } },
|
||||
}), &diag));
|
||||
try expect_y_untouched(&doc);
|
||||
}
|
||||
}
|
||||
|
||||
test "a numeric segment still addresses an array element" {
|
||||
// The regression guard for the refusal above: indexed paths are the one
|
||||
// way into an array that does work, and they must keep working, including
|
||||
// the null padding past the end.
|
||||
var doc = try array_doc(testing.allocator);
|
||||
defer doc.arena.deinit();
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "$set", .value = .{ .doc = &.{
|
||||
.{ .key = "y.0.b", .value = .{ .int32 = 7 } },
|
||||
.{ .key = "y.3.b", .value = .{ .int32 = 8 } },
|
||||
} } },
|
||||
}), null);
|
||||
const y = doc.get("y").?;
|
||||
try testing.expectEqual(@as(usize, 4), y.array.len);
|
||||
try testing.expectEqual(@as(i32, 7), y.array[0].doc[0].value.int32);
|
||||
try testing.expect(y.array[2] == .null);
|
||||
try testing.expectEqual(@as(i32, 8), y.array[3].doc[0].value.int32);
|
||||
}
|
||||
|
||||
test "a replacement is not a path, so it is not refused" {
|
||||
// The one case out of seventeen where this server already agreed with
|
||||
// mongod: `arrayFilters` alongside a replacement is ignored by both, and
|
||||
// a replacement field named like a path is data, not a path.
|
||||
var doc = try array_doc(testing.allocator);
|
||||
defer doc.arena.deinit();
|
||||
try apply(&doc, &doc_of(&.{
|
||||
.{ .key = "z", .value = .{ .int32 = 1 } },
|
||||
}), null);
|
||||
try testing.expect(doc.get("y") == null);
|
||||
try testing.expectEqual(@as(i32, 1), doc.get("z").?.int32);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user