1 /*
   2  * Copyright (c) 2003, 2010, 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.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug     4530538
  27  * @summary Basic unit test of
  28  *          ThreadMXBean.setThreadContentionMonitoringEnabled()
  29  *          and ThreadMXBean.setThreadCpuTimeEnabled().
  30  * @author  Mandy Chung
  31  *
  32  * @run main EnableTest
  33  */
  34 
  35 import java.lang.management.*;
  36 
  37 public class EnableTest {
  38     private static ThreadMXBean tm = ManagementFactory.getThreadMXBean();
  39 
  40     private static void checkThreadContentionMonitoring(boolean expectedValue)
  41         throws Exception {
  42         boolean value = tm.isThreadContentionMonitoringEnabled();
  43         if (value != expectedValue) {
  44              throw new RuntimeException("TEST FAILED: " +
  45                  "isThreadContentionMonitoringEnabled() returns " + value +
  46                  " but expected to be " + expectedValue);
  47         }
  48 
  49     }
  50 
  51     private static void testThreadContentionMonitoring()
  52         throws Exception {
  53         if (!tm.isThreadContentionMonitoringSupported()) return;
  54 
  55         // Test setThreadContentionMonitoringEnabled()
  56         checkThreadContentionMonitoring(false);
  57 
  58         // Check enabled once
  59         tm.setThreadContentionMonitoringEnabled(true);
  60         checkThreadContentionMonitoring(true);
  61 
  62         // Enable it two more times
  63         tm.setThreadContentionMonitoringEnabled(true);
  64         checkThreadContentionMonitoring(true);
  65 
  66         tm.setThreadContentionMonitoringEnabled(true);
  67 
  68         // Disable once will globally disable the thread contention monitoring
  69         tm.setThreadContentionMonitoringEnabled(false);
  70         checkThreadContentionMonitoring(false);
  71 
  72         tm.setThreadContentionMonitoringEnabled(false);
  73         checkThreadContentionMonitoring(false);
  74 
  75         tm.setThreadContentionMonitoringEnabled(true);
  76         checkThreadContentionMonitoring(true);
  77     }
  78 
  79     private static void checkThreadCpuTime(boolean expectedValue)
  80         throws Exception {
  81         boolean value = tm.isThreadCpuTimeEnabled();
  82         if (value != expectedValue) {
  83              throw new RuntimeException("TEST FAILED: " +
  84                  "isThreadCpuTimeEnabled() returns " + value +
  85                  " but expected to be " + expectedValue);
  86         }
  87     }
  88 
  89     private static void testThreadCpuTime()
  90         throws Exception {
  91         if (!tm.isThreadCpuTimeSupported()) return;
  92 
  93         // Test setThreadCpuTimeEnabled()
  94         if (!tm.isThreadCpuTimeEnabled()) {
  95             tm.setThreadCpuTimeEnabled(true);
  96             checkThreadCpuTime(true);
  97         }
  98 
  99         tm.setThreadCpuTimeEnabled(false);
 100         checkThreadCpuTime(false);
 101 
 102         tm.setThreadCpuTimeEnabled(true);
 103         checkThreadCpuTime(true);
 104 
 105         tm.setThreadCpuTimeEnabled(true);
 106         checkThreadCpuTime(true);
 107 
 108         tm.setThreadCpuTimeEnabled(true);
 109 
 110         // disable globally
 111         tm.setThreadCpuTimeEnabled(false);
 112         checkThreadCpuTime(false);
 113 
 114         tm.setThreadCpuTimeEnabled(false);
 115         checkThreadCpuTime(false);
 116 
 117         tm.setThreadCpuTimeEnabled(true);
 118         checkThreadCpuTime(true);
 119     }
 120 
 121     public static void main(String args[]) throws Exception {
 122         try {
 123             testThreadContentionMonitoring();
 124             testThreadCpuTime();
 125         } finally {
 126             // restore the default
 127             if (tm.isThreadContentionMonitoringSupported()) {
 128                 tm.setThreadContentionMonitoringEnabled(false);
 129             }
 130             if (tm.isThreadCpuTimeSupported()) {
 131                 tm.setThreadCpuTimeEnabled(false);
 132             }
 133         }
 134 
 135 
 136         System.out.println("Test passed.");
 137     }
 138 }