Skip to content

Commit 5d2f876

Browse files
authored
RUST-1191 Reduce flakiness of change_stream::batch_mid_resume_token test (#655)
1 parent b555b17 commit 5d2f876

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

src/test/change_stream.rs

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,10 @@ use crate::{
88
options::ChangeStreamOptions,
99
ChangeStream,
1010
},
11+
coll::options::CollectionOptions,
1112
db::options::{ChangeStreamPreAndPostImages, CreateCollectionOptions},
1213
event::command::{CommandStartedEvent, CommandSucceededEvent},
14+
options::{Acknowledgment, WriteConcern},
1315
test::{CommandEvent, FailCommandOptions, FailPoint, FailPointMode},
1416
Collection,
1517
};
@@ -46,7 +48,12 @@ async fn init_stream(
4648
}
4749
let client = EventClient::with_options(options).await;
4850
let db = client.database("change_stream_tests");
49-
let coll = db.collection::<Document>(coll_name);
51+
let coll = db.collection_with_options::<Document>(
52+
coll_name,
53+
CollectionOptions::builder()
54+
.write_concern(WriteConcern::builder().w(Acknowledgment::Majority).build())
55+
.build(),
56+
);
5057
coll.drop(None).await?;
5158
let stream = coll.watch(None, None).await?;
5259
Ok(Some((client, coll, stream)))
@@ -448,8 +455,8 @@ async fn batch_mid_resume_token() -> Result<()> {
448455
None => return Ok(()),
449456
};
450457

451-
coll.insert_one(doc! {}, None).await?;
452-
coll.insert_one(doc! {}, None).await?;
458+
coll.insert_many((0..2).map(|i| doc! { "_id": i as i32 }), None)
459+
.await?;
453460

454461
let mid_id = stream.next().await.transpose()?.unwrap().id;
455462
assert_eq!(stream.resume_token(), Some(mid_id));

0 commit comments

Comments
 (0)