lttng-ust-java-tests-jul/src/test/java/org/lttng/ust/agent/integration/events/JulLoggerHierarchyListIT.java

   1 /*
   2  * Copyright (C) 2016, EfficiOS Inc., Alexandre Montplaisir <alexmonthy@efficios.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 package org.lttng.ust.agent.integration.events;
  20
  21 import java.io.IOException;
  22 import java.util.logging.Handler;
  23 import java.util.logging.LogManager;
  24 import java.util.logging.Logger;
  25
  26 import org.junit.jupiter.api.AfterAll;
  27 import org.junit.jupiter.api.AfterEach;
  28 import org.junit.jupiter.api.BeforeAll;
  29 import org.junit.jupiter.api.BeforeEach;
  30 import org.lttng.tools.ILttngSession.Domain;
  31 import org.lttng.ust.agent.jul.LttngLogHandler;
  32 import org.lttng.ust.agent.utils.JulTestUtils;
  33
  34 /**
  35  * Implementation of {@link LoggerHierachyListITBase} for JUL log handlers.
  36  *
  37  * @author Alexandre Montplaisir
  38  */
  39 public class JulLoggerHierarchyListIT extends LoggerHierachyListITBase {
  40
  41     private Logger parentLogger;
  42     private Logger childLogger;
  43
  44     private Handler parentHandler;
  45     private Handler childHandler;
  46
  47     // ------------------------------------------------------------------------
  48     // Maintenance
  49     // ------------------------------------------------------------------------
  50
  51     /**
  52      * Class setup
  53      */
  54     @BeforeAll
  55     public static void julClassSetup() {
  56         JulTestUtils.testClassSetup();
  57     }
  58
  59     /**
  60      * Class cleanup
  61      */
  62     @AfterAll
  63     public static void julClassCleanup() {
  64         JulTestUtils.testClassCleanup();
  65     }
  66
  67     /**
  68      * Test setup
  69      */
  70     @SuppressWarnings("static-method")
  71     @BeforeEach
  72     public void setup() {
  73         /*
  74          * Kind of hackish, but it's the only way to ensure that loggers are
  75          * really removed in-between tests, since LogManager does not provide a
  76          * way to forcibly remove a logger, and it doesn't seem like it will any
  77          * time soon, see http://bugs.java.com/view_bug.do?bug_id=4811930
  78          */
  79         LogManager.getLogManager().reset();
  80         System.gc();
  81     }
  82
  83     /**
  84      * Test cleanup
  85      */
  86     @AfterEach
  87     public void cleanup() {
  88         if (parentLogger != null) {
  89             if (parentHandler != null) {
  90                 parentLogger.removeHandler(parentHandler);
  91                 parentHandler.close();
  92                 parentHandler = null;
  93             }
  94             parentLogger = null;
  95         }
  96
  97         if (childLogger != null) {
  98             if (childHandler != null) {
  99                 childLogger.removeHandler(childHandler);
 100                 childHandler.close();
 101                 childHandler = null;
 102             }
 103             childLogger = null;
 104         }
 105
 106         LogManager.getLogManager().reset();
 107         System.gc();
 108     }
 109
 110     // ------------------------------------------------------------------------
 111     // Abstract methods
 112     // ------------------------------------------------------------------------
 113
 114     @Override
 115     protected Domain getDomain() {
 116         return Domain.JUL;
 117     }
 118
 119     @Override
 120     protected void activateLoggers(boolean parentLoggerActive,
 121             boolean parentLoggerHasHandler,
 122             boolean childLoggerActive,
 123             boolean childLoggerHasHandler) throws IOException {
 124         if (parentLoggerActive) {
 125             parentLogger = Logger.getLogger(PARENT_LOGGER);
 126             if (parentLoggerHasHandler) {
 127                 parentHandler = new LttngLogHandler();
 128                 parentLogger.addHandler(parentHandler);
 129             }
 130         }
 131
 132         if (childLoggerActive) {
 133             childLogger = Logger.getLogger(CHILD_LOGGER);
 134             if (childLoggerHasHandler) {
 135                 childHandler = new LttngLogHandler();
 136                 childLogger.addHandler(childHandler);
 137             }
 138         }
 139     }
 140
 141 }