test/jdk/java/util/logging/Level/CustomLevel.java - platform/libcore - Git at Google

 /*
  * Copyright (c) 2013, 2016, Oracle and/or its affiliates. All rights reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * This code is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License version 2 only, as
  * published by the Free Software Foundation.
  *
  * This code is distributed in the hope that it will be useful, but WITHOUT
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  * version 2 for more details (a copy is included in the LICENSE file that
  * accompanied this code).
  *
  * You should have received a copy of the GNU General Public License version
  * 2 along with this work; if not, write to the Free Software Foundation,
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  *
  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  * or visit www.oracle.com if you need additional information or have any
  * questions.
  */

 import java.io.*;
 import java.lang.management.ManagementFactory;
 import java.lang.management.PlatformLoggingMXBean;
 import java.lang.ref.Reference;
 import java.lang.ref.ReferenceQueue;
 import java.lang.ref.WeakReference;
 import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 import java.net.MalformedURLException;
 import java.net.URL;
 import java.net.URLClassLoader;
 import java.util.*;
 import java.util.logging.*;

 /*
  * @test
  * @bug 8026027 6543126 8187073
  * @modules java.logging
   *         java.management
  * @summary Test Level.parse to look up custom levels by name and its
  *          localized name, as well as severity.
  *
  * @run main/othervm CustomLevel
  */

 public class CustomLevel extends Level {
     public CustomLevel(String name, int value, String resourceBundleName) {
         super(name, value, resourceBundleName);
     }

     private static final List<Level> levels = new ArrayList<>();
     private static final String RB_NAME = "myresource";
     private static final String OTHERRB_NAME = "myresource2";

     private static class CustomLevelReference extends WeakReference<Level> {
         final String name;
         final int value;
         final String resourceBundleName;
         public CustomLevelReference(Level level, ReferenceQueue<Level> queue) {
             super(level, queue);
             name = level.getName();
             value = level.intValue();
             resourceBundleName = level.getResourceBundleName();
         }

         @Override
         public String toString() {
             return "CustomLevelReference(\"" + name + "\", " + value + ", \""
                     + resourceBundleName + "\")";
         }

     }

     public static void main(String[] args) throws Exception {
         setupCustomLevels();
         setUpCustomLevelsOtherLoader();
         PlatformLoggingMXBean mxbean = ManagementFactory.getPlatformMXBean(PlatformLoggingMXBean.class);
         Logger logger = Logger.getLogger("foo.bar");

         // Level.parse will return the custom Level instance
         for (Level level : levels) {
             final ResourceBundle rb = getResourceBundle(level);
             String name = level.getName();
             Level l = Level.parse(name);
             if (!name.equals("WARNING") && !name.equals("INFO")
                  && !name.equals("SEVERE")) {
                 // custom level whose name doesn't conflict with any standard one
                 checkCustomLevel(l, level);
             } else if (l != Level.WARNING && l != Level.INFO && l != Level.SEVERE
                     || !name.equals(l.getName())) {
                 throw new RuntimeException("Unexpected level " + formatLevel(l));
             }
             System.out.println("Level.parse found expected level: "
                             + formatLevel(l));
             String localizedName = rb.getString(level.getName());
             l = Level.parse(localizedName);
             if (l != level) {
                 throw new RuntimeException("Unexpected level " + l + " "
                     + l.getClass() + " for " + localizedName
                     + " in " + rb.getBaseBundleName());
             }
             l = Level.parse(String.valueOf(level.intValue()));
             System.out.println("Level.parse(" + level.intValue() + ") returns " + l);
             if (l != level) {
                 if (l == null || l.intValue() != level.intValue()) {
                     throw new RuntimeException("Unexpected level " + l
                             + (l == null ? "" : (" " + l.getClass()))
                             + " for " + level.intValue());
                 }
             }
             mxbean.setLoggerLevel(logger.getName(), String.valueOf(level.intValue()));
             Level l2 = logger.getLevel();
             if (l2 != level) {
                 if (l2 == null || l2.intValue() != level.intValue()) {
                     throw new RuntimeException("Unexpected level " + l2
                             + (l2 == null ? "" : (" " + l2.getClass()))
                             + " for " + level.intValue());
                 }
             }
         }


         final long otherLevelCount = levels.stream()
             .filter(CustomLevel::isCustomLoader)
             .count();

         // Now verify that custom level instances are correctly
         // garbage collected when no longer referenced
         ReferenceQueue<Level> queue = new ReferenceQueue<>();
         List<CustomLevelReference> refs = new ArrayList<>();
         List<CustomLevelReference> customRefs = new ArrayList<>();
         int otherLevels = 0;
         while (!levels.isEmpty()) {
             Level l = levels.stream().findAny().get();
             boolean isCustomLoader = isCustomLoader(l);
             if (isCustomLoader) otherLevels++;

             CustomLevelReference ref = new CustomLevelReference(l, queue);
             if (isCustomLoader) {
                 customRefs.add(ref);
             } else {
                 refs.add(ref);
             }

             // remove strong references to l
             levels.remove(l);
             l = null;

             // Run gc and wait for garbage collection
             if (otherLevels == otherLevelCount) {
                 if (customRefs.size() != otherLevelCount) {
                     throw new RuntimeException("Test bug: customRefs.size() != "
                              + otherLevelCount);
                 }
                 waitForGC(customRefs, queue);
             }
         }
         if (otherLevelCount != otherLevels || otherLevelCount == 0) {
             throw new RuntimeException("Test bug: "
                 + "no or wrong count of levels loaded from custom loader");
         }
         if (!customRefs.isEmpty()) {
             throw new RuntimeException(
                 "Test bug: customRefs.size() should be empty!");
         }
         while (!refs.isEmpty()) {
             final Reference<?> ref = refs.remove(0);
             if (ref.get() == null) {
                 throw new RuntimeException("Unexpected garbage collection for "
                            + ref);
             }
         }
     }

     private static void waitForGC(List<CustomLevelReference> customRefs,
                                   ReferenceQueue<Level> queue)
          throws InterruptedException
     {
         while (!customRefs.isEmpty()) {
             Reference<? extends Level> ref2;
             do {
                 System.gc();
                 Thread.sleep(100);
             } while ((ref2 = queue.poll()) == null);

             // Check garbage collected reference
             if (!customRefs.contains(ref2)) {
                throw new RuntimeException("Unexpected reference: " + ref2);
             }
             CustomLevelReference ref = customRefs.remove(customRefs.indexOf(ref2));
             System.out.println(ref2 + " garbage collected");
             final String name = ref.name;
             Level l;
             try {
                 l = Level.parse(name);
                 if (!name.equals("SEVERE")
                     && !name.equals("INFO")
                     || !name.equals(l.getName())) {
                     throw new RuntimeException("Unexpected level "
                             + formatLevel(l));
                 } else {
                     if (l == Level.WARNING || l == Level.INFO
                             || l == Level.SEVERE) {
                         System.out.println("Level.parse found expected level: "
                                 + formatLevel(l));
                     } else {
                         throw new RuntimeException("Unexpected level "
                             + formatLevel(l));
                     }
                 }
             } catch (IllegalArgumentException iae) {
                 if (!name.equals("WARNING")
                     && !name.equals("INFO")
                     && !name.equals("SEVERE")) {
                     System.out.println("Level.parse fired expected exception: "
                         + iae);
                 } else {
                     throw iae;
                 }
             }
         }
     }

     private static boolean isCustomLoader(Level level) {
         final ClassLoader cl = level.getClass().getClassLoader();
         return cl != null
              && cl != ClassLoader.getPlatformClassLoader()
              && cl != ClassLoader.getSystemClassLoader();
     }

     static ResourceBundle getResourceBundle(Level level) {
         return isCustomLoader(level)
             ? ResourceBundle.getBundle(OTHERRB_NAME, Locale.getDefault(),
                                        level.getClass().getClassLoader())
             : ResourceBundle.getBundle(RB_NAME);
     }

     private static void setupCustomLevels() throws IOException {
         levels.add(new CustomLevel("EMERGENCY", 1090, RB_NAME));
         levels.add(new CustomLevel("ALERT", 1060, RB_NAME));
         levels.add(new CustomLevel("CRITICAL", 1030, RB_NAME));
         levels.add(new CustomLevel("WARNING", 1010, RB_NAME));
         levels.add(new CustomLevel("INFO", 1000, RB_NAME));
     }

     static void setUpCustomLevelsOtherLoader()
          throws MalformedURLException,
                ClassNotFoundException, NoSuchMethodException,
                IllegalAccessException, InvocationTargetException
     {
         final String classes = System.getProperty("test.classes",
                                                   "build/classes");
         final String sources = System.getProperty("test.src",
                                                   "src");
         final URL curl = new File(classes).toURI().toURL();
         final URL surl = new File(sources).toURI().toURL();
         URLClassLoader loader = new URLClassLoader(new URL[] {curl, surl},
                                      ClassLoader.getPlatformClassLoader());
         Class<?> customLevelClass = Class.forName("CustomLevel", false, loader);
         Method m = customLevelClass.getMethod("setUpCustomLevelsOtherLoader",
                                               List.class);
         m.invoke(null, levels);
     }

     public static void setUpCustomLevelsOtherLoader(List<Level> levels) {
         levels.add(new CustomLevel("OTHEREMERGENCY", 1091, OTHERRB_NAME));
         levels.add(new CustomLevel("OTHERALERT", 1061, OTHERRB_NAME));
         levels.add(new CustomLevel("OTHERCRITICAL", 1031, OTHERRB_NAME));
         levels.add(new CustomLevel("SEVERE", 1011, OTHERRB_NAME));
         levels.add(new CustomLevel("INFO", 1000, OTHERRB_NAME));
     }

     static void checkCustomLevel(Level level, Level expected) {
         // Level value must be the same
         if (!level.equals(expected)) {
             throw new RuntimeException(formatLevel(level) + " != "
                  + formatLevel(expected));
         }

         if (!level.getName().equals(expected.getName())) {
             throw new RuntimeException(formatLevel(level) + " != "
                  + formatLevel(expected));
         }

         // Level.parse is expected to return the custom Level
         if (level != expected) {
             throw new RuntimeException(formatLevel(level) + " != "
                  + formatLevel(expected));
         }

         final ResourceBundle rb = getResourceBundle(level);
         String name = rb.getString(level.getName());
         if (!level.getLocalizedName().equals(name)) {
             // must have the same localized name
             throw new RuntimeException(level.getLocalizedName() + " != " + name);
         }
     }

     static String formatLevel(Level l) {
         return l + ":" + l.intValue() + ":" + l.getClass().getName();
     }
 }
	/*
	* Copyright (c) 2013, 2016, Oracle and/or its affiliates. All rights reserved.
	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
	*
	* This code is free software; you can redistribute it and/or modify it
	* under the terms of the GNU General Public License version 2 only, as
	* published by the Free Software Foundation.
	*
	* This code is distributed in the hope that it will be useful, but WITHOUT
	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
	* version 2 for more details (a copy is included in the LICENSE file that
	* accompanied this code).
	*
	* You should have received a copy of the GNU General Public License version
	* 2 along with this work; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
	*
	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
	* or visit www.oracle.com if you need additional information or have any
	* questions.
	*/

	import java.io.*;
	import java.lang.management.ManagementFactory;
	import java.lang.management.PlatformLoggingMXBean;
	import java.lang.ref.Reference;
	import java.lang.ref.ReferenceQueue;
	import java.lang.ref.WeakReference;
	import java.lang.reflect.InvocationTargetException;
	import java.lang.reflect.Method;
	import java.net.MalformedURLException;
	import java.net.URL;
	import java.net.URLClassLoader;
	import java.util.*;
	import java.util.logging.*;

	/*
	* @test
	* @bug 8026027 6543126 8187073
	* @modules java.logging
	* java.management
	* @summary Test Level.parse to look up custom levels by name and its
	* localized name, as well as severity.
	*
	* @run main/othervm CustomLevel
	*/

	public class CustomLevel extends Level {
	public CustomLevel(String name, int value, String resourceBundleName) {
	super(name, value, resourceBundleName);
	}

	private static final List<Level> levels = new ArrayList<>();
	private static final String RB_NAME = "myresource";
	private static final String OTHERRB_NAME = "myresource2";

	private static class CustomLevelReference extends WeakReference<Level> {
	final String name;
	final int value;
	final String resourceBundleName;
	public CustomLevelReference(Level level, ReferenceQueue<Level> queue) {
	super(level, queue);
	name = level.getName();
	value = level.intValue();
	resourceBundleName = level.getResourceBundleName();
	}

	@Override
	public String toString() {
	return "CustomLevelReference(\"" + name + "\", " + value + ", \""
	+ resourceBundleName + "\")";
	}

	}

	public static void main(String[] args) throws Exception {
	setupCustomLevels();
	setUpCustomLevelsOtherLoader();
	PlatformLoggingMXBean mxbean = ManagementFactory.getPlatformMXBean(PlatformLoggingMXBean.class);
	Logger logger = Logger.getLogger("foo.bar");

	// Level.parse will return the custom Level instance
	for (Level level : levels) {
	final ResourceBundle rb = getResourceBundle(level);
	String name = level.getName();
	Level l = Level.parse(name);
	if (!name.equals("WARNING") && !name.equals("INFO")
	&& !name.equals("SEVERE")) {
	// custom level whose name doesn't conflict with any standard one
	checkCustomLevel(l, level);
	} else if (l != Level.WARNING && l != Level.INFO && l != Level.SEVERE
	\|\| !name.equals(l.getName())) {
	throw new RuntimeException("Unexpected level " + formatLevel(l));
	}
	System.out.println("Level.parse found expected level: "
	+ formatLevel(l));
	String localizedName = rb.getString(level.getName());
	l = Level.parse(localizedName);
	if (l != level) {
	throw new RuntimeException("Unexpected level " + l + " "
	+ l.getClass() + " for " + localizedName
	+ " in " + rb.getBaseBundleName());
	}
	l = Level.parse(String.valueOf(level.intValue()));
	System.out.println("Level.parse(" + level.intValue() + ") returns " + l);
	if (l != level) {
	if (l == null \|\| l.intValue() != level.intValue()) {
	throw new RuntimeException("Unexpected level " + l
	+ (l == null ? "" : (" " + l.getClass()))
	+ " for " + level.intValue());
	}
	}
	mxbean.setLoggerLevel(logger.getName(), String.valueOf(level.intValue()));
	Level l2 = logger.getLevel();
	if (l2 != level) {
	if (l2 == null \|\| l2.intValue() != level.intValue()) {
	throw new RuntimeException("Unexpected level " + l2
	+ (l2 == null ? "" : (" " + l2.getClass()))
	+ " for " + level.intValue());
	}
	}
	}


	final long otherLevelCount = levels.stream()
	.filter(CustomLevel::isCustomLoader)
	.count();

	// Now verify that custom level instances are correctly
	// garbage collected when no longer referenced
	ReferenceQueue<Level> queue = new ReferenceQueue<>();
	List<CustomLevelReference> refs = new ArrayList<>();
	List<CustomLevelReference> customRefs = new ArrayList<>();
	int otherLevels = 0;
	while (!levels.isEmpty()) {
	Level l = levels.stream().findAny().get();
	boolean isCustomLoader = isCustomLoader(l);
	if (isCustomLoader) otherLevels++;

	CustomLevelReference ref = new CustomLevelReference(l, queue);
	if (isCustomLoader) {
	customRefs.add(ref);
	} else {
	refs.add(ref);
	}

	// remove strong references to l
	levels.remove(l);
	l = null;

	// Run gc and wait for garbage collection
	if (otherLevels == otherLevelCount) {
	if (customRefs.size() != otherLevelCount) {
	throw new RuntimeException("Test bug: customRefs.size() != "
	+ otherLevelCount);
	}
	waitForGC(customRefs, queue);
	}
	}
	if (otherLevelCount != otherLevels \|\| otherLevelCount == 0) {
	throw new RuntimeException("Test bug: "
	+ "no or wrong count of levels loaded from custom loader");
	}
	if (!customRefs.isEmpty()) {
	throw new RuntimeException(
	"Test bug: customRefs.size() should be empty!");
	}
	while (!refs.isEmpty()) {
	final Reference<?> ref = refs.remove(0);
	if (ref.get() == null) {
	throw new RuntimeException("Unexpected garbage collection for "
	+ ref);
	}
	}
	}

	private static void waitForGC(List<CustomLevelReference> customRefs,
	ReferenceQueue<Level> queue)
	throws InterruptedException
	{
	while (!customRefs.isEmpty()) {
	Reference<? extends Level> ref2;
	do {
	System.gc();
	Thread.sleep(100);
	} while ((ref2 = queue.poll()) == null);

	// Check garbage collected reference
	if (!customRefs.contains(ref2)) {
	throw new RuntimeException("Unexpected reference: " + ref2);
	}
	CustomLevelReference ref = customRefs.remove(customRefs.indexOf(ref2));
	System.out.println(ref2 + " garbage collected");
	final String name = ref.name;
	Level l;
	try {
	l = Level.parse(name);
	if (!name.equals("SEVERE")
	&& !name.equals("INFO")
	\|\| !name.equals(l.getName())) {
	throw new RuntimeException("Unexpected level "
	+ formatLevel(l));
	} else {
	if (l == Level.WARNING \|\| l == Level.INFO
	\|\| l == Level.SEVERE) {
	System.out.println("Level.parse found expected level: "
	+ formatLevel(l));
	} else {
	throw new RuntimeException("Unexpected level "
	+ formatLevel(l));
	}
	}
	} catch (IllegalArgumentException iae) {
	if (!name.equals("WARNING")
	&& !name.equals("INFO")
	&& !name.equals("SEVERE")) {
	System.out.println("Level.parse fired expected exception: "
	+ iae);
	} else {
	throw iae;
	}
	}
	}
	}

	private static boolean isCustomLoader(Level level) {
	final ClassLoader cl = level.getClass().getClassLoader();
	return cl != null
	&& cl != ClassLoader.getPlatformClassLoader()
	&& cl != ClassLoader.getSystemClassLoader();
	}

	static ResourceBundle getResourceBundle(Level level) {
	return isCustomLoader(level)
	? ResourceBundle.getBundle(OTHERRB_NAME, Locale.getDefault(),
	level.getClass().getClassLoader())
	: ResourceBundle.getBundle(RB_NAME);
	}

	private static void setupCustomLevels() throws IOException {
	levels.add(new CustomLevel("EMERGENCY", 1090, RB_NAME));
	levels.add(new CustomLevel("ALERT", 1060, RB_NAME));
	levels.add(new CustomLevel("CRITICAL", 1030, RB_NAME));
	levels.add(new CustomLevel("WARNING", 1010, RB_NAME));
	levels.add(new CustomLevel("INFO", 1000, RB_NAME));
	}

	static void setUpCustomLevelsOtherLoader()
	throws MalformedURLException,
	ClassNotFoundException, NoSuchMethodException,
	IllegalAccessException, InvocationTargetException
	{
	final String classes = System.getProperty("test.classes",
	"build/classes");
	final String sources = System.getProperty("test.src",
	"src");
	final URL curl = new File(classes).toURI().toURL();
	final URL surl = new File(sources).toURI().toURL();
	URLClassLoader loader = new URLClassLoader(new URL[] {curl, surl},
	ClassLoader.getPlatformClassLoader());
	Class<?> customLevelClass = Class.forName("CustomLevel", false, loader);
	Method m = customLevelClass.getMethod("setUpCustomLevelsOtherLoader",
	List.class);
	m.invoke(null, levels);
	}

	public static void setUpCustomLevelsOtherLoader(List<Level> levels) {
	levels.add(new CustomLevel("OTHEREMERGENCY", 1091, OTHERRB_NAME));
	levels.add(new CustomLevel("OTHERALERT", 1061, OTHERRB_NAME));
	levels.add(new CustomLevel("OTHERCRITICAL", 1031, OTHERRB_NAME));
	levels.add(new CustomLevel("SEVERE", 1011, OTHERRB_NAME));
	levels.add(new CustomLevel("INFO", 1000, OTHERRB_NAME));
	}

	static void checkCustomLevel(Level level, Level expected) {
	// Level value must be the same
	if (!level.equals(expected)) {
	throw new RuntimeException(formatLevel(level) + " != "
	+ formatLevel(expected));
	}

	if (!level.getName().equals(expected.getName())) {
	throw new RuntimeException(formatLevel(level) + " != "
	+ formatLevel(expected));
	}

	// Level.parse is expected to return the custom Level
	if (level != expected) {
	throw new RuntimeException(formatLevel(level) + " != "
	+ formatLevel(expected));
	}

	final ResourceBundle rb = getResourceBundle(level);
	String name = rb.getString(level.getName());
	if (!level.getLocalizedName().equals(name)) {
	// must have the same localized name
	throw new RuntimeException(level.getLocalizedName() + " != " + name);
	}
	}

	static String formatLevel(Level l) {
	return l + ":" + l.intValue() + ":" + l.getClass().getName();
	}
	}