PYTHON-1470 - Fix test_create for 3.7+ mongos

behackett · behackett · commit 585d0fbd4f6d · 2018-02-20T13:46:48.000-08:00
diff --git a/test/test_collection.py b/test/test_collection.py
@@ -147,12 +147,19 @@ def test_create(self):
         # No Exception.
         db = client_context.client.pymongo_test
         db.create_test_no_wc.drop()
+        wait_until(lambda: 'create_test_no_wc' not in db.collection_names(),
+                 'drop create_test_no_wc collection')
         Collection(db, name='create_test_no_wc', create=True)
-        with self.assertRaises(OperationFailure):
-            Collection(
-                db, name='create-test-wc',
-                write_concern=IMPOSSIBLE_WRITE_CONCERN,
-                create=True)
+        wait_until(lambda: 'create_test_no_wc' in db.collection_names(),
+                 'create create_test_no_wc collection')
+        # SERVER-33317
+        if (not client_context.is_mongos or not
+                client_context.version.at_least(3, 7, 0)):
+            with self.assertRaises(OperationFailure):
+                Collection(
+                    db, name='create-test-wc',
+                    write_concern=IMPOSSIBLE_WRITE_CONCERN,
+                    create=True)
 
     def test_drop_nonexistent_collection(self):
         self.db.drop_collection('test')
