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 */ 48 public class TestEventFactoryRegistration { 49 50 public static void main(String[] args) throws Exception { 51 // Create an unregistered event 52 List<AnnotationElement> annotations = new ArrayList<>(); 53 annotations.add(new AnnotationElement(Registered.class, false)); 54 EventFactory factory = EventFactory.create(annotations, Collections.emptyList()); 55 56 try { 57 factory.getEventType(); 58 Asserts.fail("Should not be able to get event type from an unregistered event"); 59 } catch(IllegalStateException ise) { 60 // OK as expected 61 } 62 63 // Now, register the event 64 factory.register(); 65 EventType eventType = factory.getEventType(); 66 verifyRegistered(factory.getEventType()); 67 68 69 // Now, unregister the event 70 factory.unregister(); 71 72 verifyUnregistered(eventType); 73 74 // Create a registered event 75 factory = EventFactory.create(Collections.emptyList(), Collections.emptyList()); 76 77 eventType = factory.getEventType(); 78 Asserts.assertNotNull(eventType); 79 80 verifyRegistered(eventType); 81 82 } 83 84 private static void verifyUnregistered(EventType eventType) { 85 // Verify the event is not registered 86 for (EventType type : FlightRecorder.getFlightRecorder().getEventTypes()) { 87 if (eventType.getId() == type.getId()) { 88 Asserts.fail("Event is not unregistered"); 89 } 90 } 91 } 92 93 private static void verifyRegistered(EventType eventType) { 94 // Verify the event is registered 95 boolean found = false; 96 for (EventType type : FlightRecorder.getFlightRecorder().getEventTypes()) { 97 if (eventType.getId() == type.getId()) { 98 found = true; 99 } 100 } 101 if(!found) { 102 Asserts.fail("Event not registered"); 103 } 104 } 105 106 }