PYTHON-5521 - Update TestBsonSizeBatches.test_06_insert_fails_over_16MiB error codes (mongodb#2515)

Author: NoahStapp

test/asynchronous/test_encryption.py

@@ -1268,7 +1268,7 @@ async def test_06_insert_fails_over_16MiB(self):
         with self.assertRaises(BulkWriteError) as ctx:
             await self.coll_encrypted.bulk_write([InsertOne(doc)])
         err = ctx.exception.details["writeErrors"][0]
-        self.assertEqual(2, err["code"])
+        self.assertIn(err["code"], [2, 10334])
         self.assertIn("object to insert too large", err["errmsg"])
 
 

test/test_encryption.py

@@ -1264,7 +1264,7 @@ def test_06_insert_fails_over_16MiB(self):
         with self.assertRaises(BulkWriteError) as ctx:
             self.coll_encrypted.bulk_write([InsertOne(doc)])
         err = ctx.exception.details["writeErrors"][0]
-        self.assertEqual(2, err["code"])
+        self.assertIn(err["code"], [2, 10334])
         self.assertIn("object to insert too large", err["errmsg"])