From cd5600018eb0dceb55f7e6ca2a8e785fbcd536fc Mon Sep 17 00:00:00 2001
From: Shane Harvey <shane.harvey@mongodb.com>
Date: Thu, 7 Feb 2019 16:01:32 -0800
Subject: [PATCH] PYTHON-1739 Skip failing change stream test on sharded
 clusters (#397)

(cherry picked from commit 9dcd828a0f910ed34a75c16b2234ec9d9278cb35)
---
 test/test_change_stream.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/test_change_stream.py b/test/test_change_stream.py
index b7719feaa..d07181a3a 100644
--- a/test/test_change_stream.py
+++ b/test/test_change_stream.py
@@ -296,13 +296,16 @@ class TestCollectionChangeStream(IntegrationTest):
                 change = next(change_stream)
                 self.assertEqual(change['_id'], change_stream._resume_token)
 
+    @client_context.require_no_mongos  # PYTHON-1739
     def test_raises_error_on_missing_id(self):
         """ChangeStream will raise an exception if the server response is
         missing the resume token.
         """
         with self.coll.watch([{'$project': {'_id': 0}}]) as change_stream:
             self.coll.insert_one({})
-            with self.assertRaises(InvalidOperation):
+            # Server returns an error after SERVER-37786, otherwise pymongo
+            # raises an error.
+            with self.assertRaises((InvalidOperation, OperationFailure)):
                 next(change_stream)
             # The cursor should now be closed.
             with self.assertRaises(StopIteration):