1 /*

     2  * Copyright (c) 2013, 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.jmx;

    27 

    28 import java.util.HashMap;

    29 import java.util.LinkedHashMap;

    30 import java.util.Map;

    31 

    32 import jdk.management.jfr.FlightRecorderMXBean;

    33 import jdk.test.lib.Asserts;

    34 

    35 /*

    36  * @test

    37  * @key jfr

    38  * @library /test/lib /test/jdk

    39  * @run main/othervm jdk.jfr.jmx.TestRecordingOptions

    40  */

    41 public class TestRecordingOptions {

    42     @SuppressWarnings({ "rawtypes", "unchecked" })

    43     public static void main(String[] args) throws Exception {

    44         Map<String, String> options = new HashMap<>();

    45         options.put("name", "myName");

    46         options.put("maxAge", "2 h");

    47         options.put("maxSize", "1234567890");

    48         options.put("dumpOnExit", "false");

    49         options.put("disk", "false");

    50         options.put("duration", "1 h"); // don't want recording to stop

    51 

    52         FlightRecorderMXBean bean = JmxHelper.getFlighteRecorderMXBean();

    53         long recId = bean.newRecording();

    54         Map<String, String> defaults = bean.getRecordingOptions(recId);

    55         bean.setRecordingOptions(recId, options);

    56 

    57         // Verify that all options have been set. We only check the option we

    58         // have set. Unknown options are ignored.

    59         Map<String, String> outOptions = bean.getRecordingOptions(recId);

    60         logMap("set options", options);

    61         logMap("get options", outOptions);

    62         for (String key : options.keySet()) {

    63             Asserts.assertTrue(outOptions.containsKey(key), "Missing key " + key);

    64             Asserts.assertEquals(options.get(key), outOptions.get(key), "Wrong value for key " + key);

    65         }

    66 

    67         // Verify options in RecordingInfo

    68         Asserts.assertEquals(outOptions.get("name"), "myName", "Wrong name");

    69         Asserts.assertEquals(outOptions.get("maxAge"), "2 h", "Wrong maxAge");

    70         Asserts.assertEquals(outOptions.get("maxSize"), "1234567890", "Wrong maxSize");

    71         Asserts.assertEquals(outOptions.get("dumpOnExit"), "false", "Wrong dumpOnExit");

    72         Asserts.assertEquals(outOptions.get("disk"), "false", "Wrong disk");

    73         Asserts.assertEquals(outOptions.get("duration"), "1 h", "Wrong duration");

    74 

    75         // try empty map

    76         bean.setRecordingOptions(recId, new HashMap<>());

    77 

    78         // try map that does not have string keys

    79         Map<Integer, String> invalidKeys = new HashMap<>();

    80         invalidKeys.put(4711, "value");

    81         try {

    82             bean.setRecordingOptions(recId, (Map) invalidKeys);

    83             throw new Error("Expected IllagalStateException for non String key");

    84         } catch (IllegalArgumentException iae) {

    85             // OK, as expected

    86         }

    87         // try map that does not have string values

    88         Map<String, Integer> invalidValues = new HashMap<>();

    89         invalidValues.put("duration", 4711);

    90         try {

    91             bean.setRecordingOptions(recId, (Map) invalidKeys);

    92             throw new Error("Expected IllagalStateException for non String value");

    93         } catch (IllegalArgumentException iae) {

    94             // OK, as expected

    95         }

    96 

    97         // Try one incorrect value, and make sure non

    98         // of the other values are set.

    99         Map<String, String> lastIncorrect = new LinkedHashMap<>();

   100         lastIncorrect.put("duration", "10 h");

   101         lastIncorrect.put("whatever", "4711");

   102         try {

   103             bean.setRecordingOptions(recId, lastIncorrect);

   104             throw new Error("Expected IllagalStateException for incorrect key");

   105         } catch (IllegalArgumentException iae) {

   106             // ok

   107             Asserts.assertEquals("1 h", bean.getRecordingOptions(recId).get("duration"));

   108         }

   109 

   110         // verify that defaults are set back, if we use null

   111         Map<String, String> nullMap = new HashMap<>();

   112         nullMap.put("name", null);

   113         nullMap.put("maxAge", null);

   114         nullMap.put("maxSize", null);

   115         nullMap.put("dumpOnExit", null);

   116         nullMap.put("disk", null);

   117         nullMap.put("duration", null);

   118         bean.setRecordingOptions(recId, nullMap);

   119         Asserts.assertEquals(bean.getRecordingOptions(recId), defaults);

   120 

   121         bean.closeRecording(recId);

   122     }

   123 

   124     private static void logMap(String name, Map<String, String> map) {

   125         for (String key : map.keySet()) {

   126             System.out.printf("%s: %s=%s%n", name, key, map.get(key));

   127         }

   128     }

   129 }