1 /*

     2  * Copyright (c) 2019, 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.consumer.recordingstream;

    27 

    28 import java.util.concurrent.CountDownLatch;

    29 import java.util.concurrent.atomic.AtomicBoolean;

    30 

    31 import jdk.jfr.Event;

    32 import jdk.jfr.consumer.RecordingStream;

    33 

    34 /**

    35  * @test

    36  * @summary Tests RecordingStream::onClose(...)

    37  * @key jfr

    38  * @requires vm.hasJFR

    39  * @library /test/lib

    40  * @run main/othervm jdk.jfr.api.consumer.recordingstream.TestMaxAge

    41  */

    42 public class TestOnClose {

    43 

    44     private static class CloseEvent extends Event {

    45     }

    46     public static void main(String... args) throws Exception {

    47         testOnCloseNull();

    48         testOnClosedUnstarted();

    49         testOnClosedStarted();

    50     }

    51 

    52     private static void testOnCloseNull() {

    53        try (RecordingStream rs = new RecordingStream()) {

    54           try {

    55               rs.onClose(null);

    56               throw new AssertionError("Expected NullPointerException from onClose(null");

    57           } catch (NullPointerException npe) {

    58               // OK; as expected

    59           }

    60        }

    61     }

    62 

    63     private static void testOnClosedStarted() throws InterruptedException {

    64         AtomicBoolean onClose = new AtomicBoolean(false);

    65         CountDownLatch event = new CountDownLatch(1);

    66         try (RecordingStream r = new RecordingStream()) {

    67             r.onEvent(e -> {

    68                 event.countDown();

    69             });

    70             r.onClose(() -> {

    71                 onClose.set(true);

    72             });

    73             r.startAsync();

    74             CloseEvent c = new CloseEvent();

    75             c.commit();

    76             event.await();

    77         }

    78         if (!onClose.get()) {

    79             throw new AssertionError("OnClose was not called");

    80         }

    81     }

    82 

    83     private static void testOnClosedUnstarted() {

    84         AtomicBoolean onClose = new AtomicBoolean(false);

    85         try (RecordingStream r = new RecordingStream()) {

    86             r.onClose(() -> {

    87                 onClose.set(true);

    88             });

    89         }

    90         if (!onClose.get()) {

    91             throw new AssertionError("OnClose was not called");

    92         }

    93     }

    94 }