From 614e22c46c57deca443a6ebfa2123d5ce383ef8f Mon Sep 17 00:00:00 2001 From: Shane Harvey Date: Wed, 19 Oct 2022 15:14:46 -0700 Subject: [PATCH] PYTHON-3478 Improve test_change_stream_can_resume_after_timeouts (#1083) --- test/test_csot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_csot.py b/test/test_csot.py index a9cf7a012..c2a62aa7f 100644 --- a/test/test_csot.py +++ b/test/test_csot.py @@ -76,10 +76,10 @@ class TestCSOT(IntegrationTest): @client_context.require_change_streams def test_change_stream_can_resume_after_timeouts(self): coll = self.db.test - with coll.watch(max_await_time_ms=150) as stream: + with coll.watch() as stream: with pymongo.timeout(0.1): with self.assertRaises(PyMongoError) as ctx: - stream.try_next() + stream.next() self.assertTrue(ctx.exception.timeout) self.assertTrue(stream.alive) with self.assertRaises(PyMongoError) as ctx: