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