Add test

test/jdk/java/foreign/TestScope.java

@@ -0,0 +1,122 @@
+/*
+ * Copyright (c) 2023, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+/*
+ * @test
+ * @run testng/othervm --enable-native-access=ALL-UNNAMED TestScope
+ */
+
+import org.testng.annotations.*;
+
+import java.lang.foreign.Arena;
+import java.lang.foreign.MemorySegment;
+import java.lang.foreign.SymbolLookup;
+import java.nio.ByteBuffer;
+import java.nio.IntBuffer;
+
+import static org.testng.Assert.*;
+
+public class TestScope {
+
+    static {
+        System.loadLibrary("LookupTest");
+    }
+
+    @Test
+    public void testDifferentArrayScope() {
+        MemorySegment.Scope scope1 = MemorySegment.ofArray(new byte[10]).scope();
+        MemorySegment.Scope scope2 = MemorySegment.ofArray(new byte[10]).scope();
+        assertNotEquals(scope1, scope2);
+    }
+
+    @Test
+    public void testDifferentBufferScope() {
+        MemorySegment.Scope scope1 = MemorySegment.ofBuffer(ByteBuffer.allocateDirect(10)).scope();
+        MemorySegment.Scope scope2 = MemorySegment.ofBuffer(ByteBuffer.allocateDirect(10)).scope();
+        assertNotEquals(scope1, scope2);
+    }
+
+    @Test
+    public void testDifferentArenaScope() {
+        MemorySegment.Scope scope1 = Arena.ofAuto().allocate(10).scope();
+        MemorySegment.Scope scope2 = Arena.ofAuto().allocate(10).scope();
+        assertNotEquals(scope1, scope2);
+    }
+
+    @Test
+    public void testSameArrayScope() {
+        byte[] arr = new byte[10];
+        assertEquals(MemorySegment.ofArray(arr).scope(), MemorySegment.ofArray(arr).scope());
+        ByteBuffer buf = ByteBuffer.wrap(arr);
+        assertEquals(MemorySegment.ofArray(arr).scope(), MemorySegment.ofBuffer(buf).scope());
+        testDerivedBufferScope(MemorySegment.ofArray(arr));
+    }
+
+    @Test
+    public void testSameBufferScope() {
+        ByteBuffer buf = ByteBuffer.allocateDirect(10);
+        assertEquals(MemorySegment.ofBuffer(buf).scope(), MemorySegment.ofBuffer(buf).scope());
+        testDerivedBufferScope(MemorySegment.ofBuffer(buf));
+    }
+
+    @Test
+    public void testSameArenaScope() {
+        try (Arena arena = Arena.ofConfined()) {
+            MemorySegment segment1 = arena.allocate(10);
+            MemorySegment segment2 = arena.allocate(10);
+            assertEquals(segment1.scope(), segment2.scope());
+            testDerivedBufferScope(segment1);
+        }
+    }
+
+    @Test
+    public void testSameNativeScope() {
+        MemorySegment segment1 = MemorySegment.ofAddress(42);
+        MemorySegment segment2 = MemorySegment.ofAddress(43);
+        assertEquals(segment1.scope(), segment2.scope());
+        assertEquals(segment1.scope(), segment2.reinterpret(10).scope());
+        assertEquals(segment1.scope(), Arena.global().scope());
+        testDerivedBufferScope(segment1.reinterpret(10));
+    }
+
+    @Test
+    public void testSameLookupScope() {
+        SymbolLookup loaderLookup = SymbolLookup.loaderLookup();
+        MemorySegment segment1 = loaderLookup.find("f").get();
+        MemorySegment segment2 = loaderLookup.find("c").get();
+        assertEquals(segment1.scope(), segment2.scope());
+        testDerivedBufferScope(segment1.reinterpret(10));
+    }
+
+    void testDerivedBufferScope(MemorySegment segment) {
+        ByteBuffer buffer = segment.asByteBuffer();
+        MemorySegment.Scope expectedScope = segment.scope();
+        assertEquals(MemorySegment.ofBuffer(buffer).scope(), expectedScope);
+        // buffer slices should have same scope
+        ByteBuffer slice = buffer.slice(0, 2);
+        assertEquals(expectedScope, MemorySegment.ofBuffer(slice).scope());
+        // buffer views should have same scope
+        IntBuffer view = buffer.asIntBuffer();
+        assertEquals(expectedScope, MemorySegment.ofBuffer(view).scope());
+    }
+}