Add test of handling MemoryError

Author: steve-s

graalpython/com.oracle.graal.python.test/src/tests/memoryerror.py

@@ -0,0 +1,57 @@
+# Copyright (c) 2024, 2024, Oracle and/or its affiliates. All rights reserved.
+# DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+#
+# The Universal Permissive License (UPL), Version 1.0
+#
+# Subject to the condition set forth below, permission is hereby granted to any
+# person obtaining a copy of this software, associated documentation and/or
+# data (collectively the "Software"), free of charge and under any and all
+# copyright rights in the Software, and any and all patent rights owned or
+# freely licensable by each licensor hereunder covering either (i) the
+# unmodified Software as contributed to or provided by such licensor, or (ii)
+# the Larger Works (as defined below), to deal in both
+#
+# (a) the Software, and
+#
+# (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
+# one is included with the Software each a "Larger Work" to which the Software
+# is contributed by such licensors),
+#
+# without restriction, including without limitation the rights to copy, create
+# derivative works of, display, perform, and distribute the Software and make,
+# use, sell, offer for sale, import, export, have made, and have sold the
+# Software and the Larger Work(s), and to sublicense the foregoing rights on
+# either these or other terms.
+#
+# This license is subject to the following condition:
+#
+# The above copyright notice and either this complete permission notice or at a
+# minimum a reference to the UPL must be included in all copies or substantial
+# portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+
+# Executed in a subprocess with small max heap size to exercise
+# real MemoryError caused by Java level OOM
+
+def alloc():
+    try: bytes(2140000000)
+    except MemoryError as e:
+        pass
+    else:
+        print("ERROR: MemoryError not caught!")
+        import sys
+        sys.exit(1)
+
+
+for i in range(1, 10):
+    alloc()
+
+print("DONE")

graalpython/com.oracle.graal.python.test/src/tests/test_exception.py

@@ -1,12 +1,14 @@
-# Copyright (c) 2018, 2023, Oracle and/or its affiliates.
+# Copyright (c) 2018, 2024, Oracle and/or its affiliates.
 # Copyright (C) 1996-2017 Python Software Foundation
 #
 # Licensed under the PYTHON SOFTWARE FOUNDATION LICENSE VERSION 2
-
+import os
 import unittest
 import sys
 import errno
 
+GRAALPYTHON = sys.implementation.name == "graalpy"
+
 def fun0(test_obj, expected_error):
     typ, val, tb = sys.exc_info()
     test_obj.assertEqual(typ, expected_error)
@@ -673,3 +675,16 @@ def __getattr__(self, i): raise AttributeError1
         assert list(Iter2()) == []
         sentinel = object()
         assert getattr(Obj1(), 'does_not_exist', sentinel) is sentinel
+
+
+# There is no simple way to restrict memory for CPython process
+@unittest.skipUnless(GRAALPYTHON)
+def test_memory_error():
+    import subprocess
+    file = os.path.join(os.path.dirname(__file__), 'memoryerror.py')
+    result = subprocess.check_output([sys.executable, '-S', '--experimental-options',
+                                      '--engine.MultiTier=false', '--engine.BackgroundCompilation=false',
+                                      '--engine.CompileImmediately', '--engine.CompileOnly=alloc',
+                                      '--vm.Xmx400m', file], text=True)
+    assert 'ERROR' not in result, result
+    assert 'DONE' in result, result