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

   1 /*
   2  * Copyright (C) 2015, 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 static org.junit.Assert.assertEquals;
  22 import static org.junit.Assert.assertNotNull;
  23 import static org.junit.Assert.assertTrue;
  24 import static org.junit.Assert.fail;
  25
  26 import java.lang.reflect.Field;
  27 import java.util.List;
  28 import java.util.logging.Level;
  29 import java.util.logging.Logger;
  30
  31 import org.junit.After;
  32 import org.junit.AfterClass;
  33 import org.junit.Before;
  34 import org.junit.BeforeClass;
  35 import org.junit.Test;
  36 import org.junit.runner.RunWith;
  37 import org.lttng.tools.ILttngSession;
  38 import org.lttng.tools.ILttngSession.Domain;
  39 import org.lttng.ust.agent.ILttngHandler;
  40 import org.lttng.ust.agent.LTTngAgent;
  41 import org.lttng.ust.agent.utils.JulTestUtils;
  42 import org.lttng.ust.agent.utils.TestPrintRunner;
  43
  44 /**
  45  * Enabled events test for the LTTng-UST JUL log handler, using the legacy API.
  46  */
  47 @RunWith(TestPrintRunner.class)
  48 @SuppressWarnings("deprecation")
  49 public class JulLegacyApiIT {
  50
  51     private static final Domain DOMAIN = Domain.JUL;
  52
  53     private static final String EVENT_NAME_A = "EventA";
  54     private static final String EVENT_NAME_B = "EventB";
  55
  56     private ILttngSession session;
  57
  58     private Logger loggerA;
  59     private Logger loggerB;
  60
  61     /**
  62      * Class setup
  63      */
  64     @BeforeClass
  65     public static void julClassSetup() {
  66         JulTestUtils.testClassSetup();
  67     }
  68
  69     /**
  70      * Class cleanup
  71      */
  72     @AfterClass
  73     public static void julClassCleanup() {
  74         JulTestUtils.testClassCleanup();
  75     }
  76
  77     /**
  78      * Test setup
  79      */
  80     @Before
  81     public void setup() {
  82         loggerA = Logger.getLogger(EVENT_NAME_A);
  83         LTTngAgent.getLTTngAgent();
  84         loggerB = Logger.getLogger(EVENT_NAME_B);
  85
  86         loggerA.setLevel(Level.ALL);
  87         loggerB.setLevel(Level.ALL);
  88
  89         session = ILttngSession.createSession(null, DOMAIN);
  90     }
  91
  92     /**
  93      * Test cleanup
  94      */
  95     @After
  96     public void tearDown() {
  97         session.close();
  98
  99         LTTngAgent.dispose();
 100
 101         loggerA = null;
 102         loggerB = null;
 103     }
 104
 105     /**
 106      * Test tracing with no events enabled in the tracing session.
 107      */
 108     @Test
 109     public void testNoEvents() {
 110         assertTrue(session.start());
 111
 112         JulTestUtils.send10EventsTo(loggerA);
 113         JulTestUtils.send10EventsTo(loggerB);
 114
 115         assertTrue(session.stop());
 116
 117         List<String> output = session.view();
 118         assertNotNull(output);
 119         assertTrue(output.isEmpty());
 120
 121         ILttngHandler handler = getAgentHandler();
 122         assertEquals(0, handler.getEventCount());
 123     }
 124
 125     /**
 126      * Test tracing with all events enabled (-j -a) in the tracing session.
 127      */
 128     @Test
 129     public void testAllEvents() {
 130         assertTrue(session.enableAllEvents());
 131         assertTrue(session.start());
 132
 133         JulTestUtils.send10EventsTo(loggerA);
 134         JulTestUtils.send10EventsTo(loggerB);
 135
 136         assertTrue(session.stop());
 137
 138         List<String> output = session.view();
 139         assertNotNull(output);
 140         assertEquals(20, output.size());
 141
 142         ILttngHandler handler = getAgentHandler();
 143         assertEquals(20, handler.getEventCount());
 144     }
 145
 146     /**
 147      * Test tracing with a subset of events enabled in the tracing session.
 148      */
 149     @Test
 150     public void testSomeEvents() {
 151         assertTrue(session.enableEvents(EVENT_NAME_A));
 152         assertTrue(session.start());
 153
 154         JulTestUtils.send10EventsTo(loggerA);
 155         JulTestUtils.send10EventsTo(loggerB);
 156
 157         assertTrue(session.stop());
 158
 159         List<String> output = session.view();
 160         assertNotNull(output);
 161         assertEquals(10, output.size());
 162
 163         ILttngHandler handler = getAgentHandler();
 164         assertEquals(10, handler.getEventCount());
 165     }
 166
 167     /**
 168      * Get the singleton JUL Handler currently managed by the LTTngAgent. It is
 169      * not public, so we need reflection to access it.
 170      *
 171      * @return The agent's JUL handler
 172      */
 173     private static ILttngHandler getAgentHandler() {
 174         try {
 175             Field julHandlerField = LTTngAgent.class.getDeclaredField("julHandler");
 176             julHandlerField.setAccessible(true);
 177             return (ILttngHandler) julHandlerField.get(LTTngAgent.getLTTngAgent());
 178         } catch (ReflectiveOperationException | SecurityException e) {
 179             fail(e.getMessage());
 180             return null;
 181         }
 182     }
 183
 184 }
 185