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 }