| 40 | |
| 41 | @flaky(reason="PYTHON-3522") |
| 42 | def test_timeout_nested(self): |
| 43 | coll = self.db.coll |
| 44 | self.assertEqual(_csot.get_timeout(), None) |
| 45 | self.assertEqual(_csot.get_deadline(), float("inf")) |
| 46 | self.assertEqual(_csot.get_rtt(), 0.0) |
| 47 | with pymongo.timeout(10): |
| 48 | coll.find_one() |
| 49 | self.assertEqual(_csot.get_timeout(), 10) |
| 50 | deadline_10 = _csot.get_deadline() |
| 51 | |
| 52 | # Capped at the original 10 deadline. |
| 53 | with pymongo.timeout(15): |
| 54 | coll.find_one() |
| 55 | self.assertEqual(_csot.get_timeout(), 15) |
| 56 | self.assertEqual(_csot.get_deadline(), deadline_10) |
| 57 | |
| 58 | # Should be reset to previous values |
| 59 | self.assertEqual(_csot.get_timeout(), 10) |
| 60 | self.assertEqual(_csot.get_deadline(), deadline_10) |
| 61 | coll.find_one() |
| 62 | |
| 63 | with pymongo.timeout(5): |
| 64 | coll.find_one() |
| 65 | self.assertEqual(_csot.get_timeout(), 5) |
| 66 | self.assertLess(_csot.get_deadline(), deadline_10) |
| 67 | |
| 68 | # Should be reset to previous values |
| 69 | self.assertEqual(_csot.get_timeout(), 10) |
| 70 | self.assertEqual(_csot.get_deadline(), deadline_10) |
| 71 | coll.find_one() |
| 72 | |
| 73 | # Should be reset to previous values |
| 74 | self.assertEqual(_csot.get_timeout(), None) |
| 75 | self.assertEqual(_csot.get_deadline(), float("inf")) |
| 76 | self.assertEqual(_csot.get_rtt(), 0.0) |
| 77 | |
| 78 | @client_context.require_change_streams |
| 79 | @flaky(reason="PYTHON-3522") |