diff --git a/tests/collab/stress_test.rs b/tests/collab/stress_test.rs index ea802c31..5e33b01d 100644 --- a/tests/collab/stress_test.rs +++ b/tests/collab/stress_test.rs @@ -46,7 +46,7 @@ async fn run_multiple_text_edits() { // run test scenario let collab = writer.collabs.get(&object_id).unwrap().collab.clone(); - let expected = test_scenario.execute(collab).await; + let expected = test_scenario.execute(collab, 50_000).await; // wait for the writer to complete sync writer.wait_object_sync_complete(&object_id).await.unwrap(); diff --git a/tests/collab/util.rs b/tests/collab/util.rs index 50d25e04..7fc6daa5 100644 --- a/tests/collab/util.rs +++ b/tests/collab/util.rs @@ -133,13 +133,8 @@ impl TestScenario { data } - pub async fn execute(&self, collab: CollabRef) -> String { - let len = self.txns.len(); - let start = Instant::now(); - for (i, t) in self.txns.iter().enumerate() { - if i % 10_000 == 0 { - tracing::info!("step #{}/{} - {:?}", i + 1, len, start.elapsed()); - } + pub async fn execute(&self, collab: CollabRef, step_count: usize) -> String { + for t in self.txns.iter().take(step_count) { let mut lock = collab.write().await; let collab = lock.borrow_mut(); let mut txn = collab.context.transact_mut(); @@ -159,13 +154,11 @@ impl TestScenario { } // validate after applying all patches - let expected = self.end_content.as_str(); let lock = collab.read().await; let collab = lock.borrow(); let txn = collab.context.transact(); let txt: TextRef = collab.data.get_with_txn(&txn, "text-id").unwrap(); let actual = txt.get_string(&txn); - assert_eq!(actual, expected); actual } }