Add test for native storage cleanup

Author: msimacek

graalpython/com.oracle.graal.python.test/src/tests/cpyext/__init__.py

@@ -36,6 +36,7 @@
 # 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.
+import gc
 
 import sys
 import os
@@ -400,6 +401,7 @@ def test(self):
                 assert cresult == presult, ("%r == %r in %s(%s)" % (cresult, presult, self.name, pargs[i]))
             else:
                 assert self.cmpfunc(cresult, presult), ("%r == %r in %s(%s)" % (cresult, presult, self.name, pargs[i]))
+        gc.collect()
 
     def __get__(self, instance, typ=None):
         if typ is None:

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_bytes.py

@@ -132,6 +132,31 @@ def compile_module(self, name):
         resulttype="int"
     )
 
+    test_native_storage = CPyExtFunction(
+        lambda args: args[0].encode('utf-8')[-1],
+        lambda: (("hello",), ("world",)),
+        argspec="O",
+        arguments=["PyObject* arg"],
+        resultspec="i",
+        # The code is creating the bytes objects in such a roundabout way in order to make sure the native storage will
+        # get collected after the test
+        code="""
+        int wrap_test_native_storage(PyObject* str) {
+            PyObject* bytes = PyUnicode_AsUTF8String(str);
+            if (bytes == NULL)
+                return -1;
+            char* s;
+            Py_ssize_t sz;
+            if (PyBytes_AsStringAndSize(bytes, &s, &sz) < 0)
+                return -1;
+            int ret = s[sz - 1];
+            Py_DECREF(bytes);
+            return ret;
+        }
+        """,
+        callfunction='wrap_test_native_storage',
+    )
+
     # PyBytes_Size
     test_PyBytes_Size = CPyExtFunction(
         lambda b: len(b[0]),