1 /* 2 * Copyright (c) 2018, 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. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 package jdk.jfr.api.event; 27 28 import java.util.ArrayList; 29 import java.util.Collections; 30 import java.util.List; 31 32 import jdk.jfr.AnnotationElement; 33 import jdk.jfr.EventFactory; 34 import jdk.jfr.EventType; 35 import jdk.jfr.FlightRecorder; 36 import jdk.jfr.Registered; 37 import jdk.test.lib.Asserts; 38 39 40 /** 41 * @test 42 * @summary EventFactory register/unregister API test 43 * @key jfr 44 * 45 * @library /lib / 46 * @run main/othervm jdk.jfr.api.event.TestEventFactoryRegistration 47 * @run main/othervm -XX:+AllowParallelDefineClass jdk.jfr.api.event.TestEventFactoryRegistration 48 */ 49 public class TestEventFactoryRegistration { 50 51 public static void main(String[] args) throws Exception { 52 // Create an unregistered event 53 List<AnnotationElement> annotations = new ArrayList<>(); 54 annotations.add(new AnnotationElement(Registered.class, false)); 55 EventFactory factory = EventFactory.create(annotations, Collections.emptyList()); 56 57 try { 58 factory.getEventType(); 59 Asserts.fail("Should not be able to get event type from an unregistered event"); 60 } catch(IllegalStateException ise) { 61 // OK as expected 62 } 63 64 // Now, register the event 65 factory.register(); 66 EventType eventType = factory.getEventType(); 67 verifyRegistered(factory.getEventType()); 68 69 70 // Now, unregister the event 71 factory.unregister(); 72 73 verifyUnregistered(eventType); 74 75 // Create a registered event 76 factory = EventFactory.create(Collections.emptyList(), Collections.emptyList()); 77 78 eventType = factory.getEventType(); 79 Asserts.assertNotNull(eventType); 80 81 verifyRegistered(eventType); 82 83 } 84 85 private static void verifyUnregistered(EventType eventType) { 86 // Verify the event is not registered 87 for (EventType type : FlightRecorder.getFlightRecorder().getEventTypes()) { 88 if (eventType.getId() == type.getId()) { 89 Asserts.fail("Event is not unregistered"); 90 } 91 } 92 } 93 94 private static void verifyRegistered(EventType eventType) { 95 // Verify the event is registered 96 boolean found = false; 97 for (EventType type : FlightRecorder.getFlightRecorder().getEventTypes()) { 98 if (eventType.getId() == type.getId()) { 99 found = true; 100 } 101 } 102 if(!found) { 103 Asserts.fail("Event not registered"); 104 } 105 } 106 107 }