Fixed #657 -- handle OverflowErrors on large allocation requests

pyca · Jul 7, 2017 · ecc0325 · ecc0325
1 parent a421276
commit ecc0325
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_rand.py b/tests/test_rand.py
@@ -32,10 +32,10 @@ def test_bytes_wrong_args(self, args):
 
     def test_insufficient_memory(self):
         """
-        `OpenSSL.rand.bytes` raises `MemoryError` if more bytes are requested
-        than will fit in memory.
+        `OpenSSL.rand.bytes` raises `MemoryError` or `OverflowError` if more
+        bytes are requested than will fit in memory.
         """
-        with pytest.raises(MemoryError):
+        with pytest.raises((MemoryError, OverflowError)):
             rand.bytes(sys.maxsize)
 
     def test_bytes(self):