1 /*
 2  *  Copyright (c) 2020, Oracle and/or its affiliates. All rights reserved.
 3  *  DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 4  *
 5  *  This code is free software; you can redistribute it and/or modify it
 6  *  under the terms of the GNU General Public License version 2 only, as
 7  *  published by the Free Software Foundation.
 8  *
 9  *  This code is distributed in the hope that it will be useful, but WITHOUT
10  *  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  *  FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12  *  version 2 for more details (a copy is included in the LICENSE file that
13  *  accompanied this code).
14  *
15  *  You should have received a copy of the GNU General Public License version
16  *  2 along with this work; if not, write to the Free Software Foundation,
17  *  Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  *  Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20  *  or visit www.oracle.com if you need additional information or have any
21  *  questions.
22  */
23 
24 /*
25  * @test
26  * @run testng TestLayoutAttributes
27  */
28 
29 import jdk.incubator.foreign.MemoryLayout;
30 import jdk.incubator.foreign.MemoryLayouts;
31 import org.testng.annotations.Test;
32 
33 import java.util.List;
34 import java.util.stream.Collectors;
35 
36 import static org.testng.Assert.assertEquals;
37 import static org.testng.Assert.assertTrue;
38 
39 public class TestLayoutAttributes {
40 
41     @Test
42     public void testAttribute() {
43         MemoryLayout ml = MemoryLayouts.BITS_32_LE
44                 .withAttribute("MyAttribute", 10L);
45         assertEquals((long) ml.attribute("MyAttribute").orElseThrow(), 10L);
46     }
47 
48     @Test
49     public void testAttributeOverwrite() {
50         MemoryLayout ml = MemoryLayouts.BITS_32_LE
51                 .withAttribute("MyAttribute", 10L);
52         assertEquals((long) ml.attribute("MyAttribute").orElseThrow(), 10L);
53         ml = ml.withAttribute("MyAttribute", 11L);
54         assertEquals((long) ml.attribute("MyAttribute").orElseThrow(), 11L);
55     }
56 
57     @Test
58     public void testAttributeNonExistent() {
59         MemoryLayout ml = MemoryLayouts.BITS_32_LE
60                 .withAttribute("MyAttribute", 10L);
61         assertTrue(ml.attribute("Foo").isEmpty());
62     }
63 
64     @Test
65     public void testNameAttribute() {
66         MemoryLayout ml = MemoryLayouts.BITS_32_LE
67                 .withName("foo");
68         assertEquals(ml.name().orElseThrow(), "foo");
69         assertEquals(ml.attribute(MemoryLayout.LAYOUT_NAME).orElseThrow(), "foo");
70     }
71 
72     @Test
73     public void testAttributesStream() {
74         MemoryLayout ml = MemoryLayouts.BITS_32_LE
75                 .withName("foo")
76                 .withAttribute("MyAttribute", 10L);
77         List<String> attribs = ml.attributes().collect(Collectors.toList());
78         assertEquals(attribs.size(), 2);
79         assertTrue(attribs.contains("MyAttribute"));
80         assertTrue(attribs.contains(MemoryLayout.LAYOUT_NAME));
81     }
82 }