SERVER-52526 limit number of measurements in each bucket in a time-series collection

1 files changed, 21 insertions, 4 deletions

diff --git a/jstests/core/time_series/time_series_bucket_limit_count.js b/jstests/core/time_series/time_series_bucket_limit_count.js
index 8f347d4b7d1..feddbf4e200 100644
--- a/jstests/core/time_series/time_series_bucket_limit_count.js
+++ b/jstests/core/time_series/time_series_bucket_limit_count.js
@@ -53,11 +53,11 @@ for (let i = 0; i < numDocs; i++) {
 
 // Check bucket collection.
 const bucketDocs = bucketsColl.find().sort({_id: 1}).toArray();
-assert.eq(1, bucketDocs.length, bucketDocs);
+assert.eq(2, bucketDocs.length, bucketDocs);
 
 // Check both buckets.
 // First bucket should be full with 'bucketMaxCount' documents.
-assert.eq(numDocs,
+assert.eq(bucketMaxCount,
           bucketDocs[0].control.count,
           'invalid count in first bucket: ' + tojson(bucketDocs[0]));
 assert.eq(0,
@@ -66,10 +66,27 @@ assert.eq(0,
 assert.eq(0,
           bucketDocs[0].control.min.x,
           'invalid control.min for x in first bucket: ' + tojson(bucketDocs[0].control));
-assert.eq(numDocs - 1,
+assert.eq(bucketMaxCount - 1,
           bucketDocs[0].control.max._id,
           'invalid control.max for _id in first bucket: ' + tojson(bucketDocs[0].control));
-assert.eq(numDocs - 1,
+assert.eq(bucketMaxCount - 1,
           bucketDocs[0].control.max.x,
           'invalid control.max for x in first bucket: ' + tojson(bucketDocs[0].control));
+
+// Second bucket should contain the remaining documents.
+assert.eq(numDocs - bucketMaxCount,
+          bucketDocs[1].control.count,
+          'invalid count in second bucket: ' + tojson(bucketDocs[1]));
+assert.eq(bucketMaxCount,
+          bucketDocs[1].control.min._id,
+          'invalid control.min for _id in second bucket: ' + tojson(bucketDocs[1].control));
+assert.eq(bucketMaxCount,
+          bucketDocs[1].control.min.x,
+          'invalid control.min for x in second bucket: ' + tojson(bucketDocs[1].control));
+assert.eq(numDocs - 1,
+          bucketDocs[1].control.max._id,
+          'invalid control.max for _id in second bucket: ' + tojson(bucketDocs[1].control));
+assert.eq(numDocs - 1,
+          bucketDocs[1].control.max.x,
+          'invalid control.max for x in second bucket: ' + tojson(bucketDocs[1].control));
 })();