2 * Copyright (C) 2015, EfficiOS Inc., Alexandre Montplaisir <alexmonthy@efficios.com>
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.
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.
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.
19 package org
.lttng
.ust
.agent
.integration
.events
;
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
;
26 import java
.lang
.reflect
.Field
;
27 import java
.util
.Arrays
;
28 import java
.util
.Collections
;
29 import java
.util
.List
;
30 import java
.util
.logging
.Level
;
31 import java
.util
.logging
.LogManager
;
32 import java
.util
.logging
.Logger
;
34 import org
.junit
.After
;
35 import org
.junit
.AfterClass
;
36 import org
.junit
.Before
;
37 import org
.junit
.BeforeClass
;
38 import org
.junit
.Test
;
39 import org
.junit
.runner
.RunWith
;
40 import org
.lttng
.tools
.ILttngSession
;
41 import org
.lttng
.tools
.ILttngSession
.Domain
;
42 import org
.lttng
.ust
.agent
.ILttngHandler
;
43 import org
.lttng
.ust
.agent
.LTTngAgent
;
44 import org
.lttng
.ust
.agent
.utils
.JulTestUtils
;
45 import org
.lttng
.ust
.agent
.utils
.TestPrintRunner
;
48 * Enabled events test for the LTTng-UST JUL log handler, using the legacy API.
50 @RunWith(TestPrintRunner
.class)
51 @SuppressWarnings("deprecation")
52 public class JulLegacyApiIT
{
54 private static final Domain DOMAIN
= Domain
.JUL
;
56 private static final String EVENT_NAME_A
= "EventA";
57 private static final String EVENT_NAME_B
= "EventB";
59 private ILttngSession session
;
60 private LTTngAgent agent
;
62 private Logger loggerA
;
63 private Logger loggerB
;
69 public static void julClassSetup() {
70 JulTestUtils
.testClassSetup();
77 public static void julClassCleanup() {
78 JulTestUtils
.testClassCleanup();
86 /* Clear the JUL logger configuration */
87 LogManager
.getLogManager().reset();
90 loggerA
= Logger
.getLogger(EVENT_NAME_A
);
91 agent
= LTTngAgent
.getLTTngAgent();
92 loggerB
= Logger
.getLogger(EVENT_NAME_B
);
94 loggerA
.setLevel(Level
.ALL
);
95 loggerB
.setLevel(Level
.ALL
);
97 session
= ILttngSession
.createSession(null, DOMAIN
);
104 public void tearDown() {
114 * Test tracing with no events enabled in the tracing session.
117 public void testNoEvents() {
118 assertTrue(session
.start());
120 JulTestUtils
.send10EventsTo(loggerA
);
121 JulTestUtils
.send10EventsTo(loggerB
);
123 assertTrue(session
.stop());
125 List
<String
> output
= session
.view();
126 assertNotNull(output
);
127 assertTrue(output
.isEmpty());
129 ILttngHandler handler
= getAgentHandler();
130 assertEquals(0, handler
.getEventCount());
134 * Test tracing with all events enabled (-j -a) in the tracing session.
137 public void testAllEvents() {
138 assertTrue(session
.enableAllEvents());
139 assertTrue(session
.start());
141 JulTestUtils
.send10EventsTo(loggerA
);
142 JulTestUtils
.send10EventsTo(loggerB
);
144 assertTrue(session
.stop());
146 List
<String
> output
= session
.view();
147 assertNotNull(output
);
148 assertEquals(20, output
.size());
150 ILttngHandler handler
= getAgentHandler();
151 assertEquals(20, handler
.getEventCount());
155 * Test tracing with a subset of events enabled in the tracing session.
158 public void testSomeEvents() {
159 assertTrue(session
.enableEvents(EVENT_NAME_A
));
160 assertTrue(session
.start());
162 JulTestUtils
.send10EventsTo(loggerA
);
163 JulTestUtils
.send10EventsTo(loggerB
);
165 assertTrue(session
.stop());
167 List
<String
> output
= session
.view();
168 assertNotNull(output
);
169 assertEquals(10, output
.size());
171 ILttngHandler handler
= getAgentHandler();
172 assertEquals(10, handler
.getEventCount());
176 * Test that the "lttng list" commands lists the expected events.
179 public void testListEvents() {
180 List
<String
> enabledEvents
= session
.listEvents();
181 List
<String
> expectedEvents
= Arrays
.asList(EVENT_NAME_A
, EVENT_NAME_B
);
183 Collections
.sort(enabledEvents
);
184 Collections
.sort(expectedEvents
);
186 assertEquals(expectedEvents
, enabledEvents
);
190 * Get the singleton JUL Handler currently managed by the LTTngAgent. It is
191 * not public, so we need reflection to access it.
193 * @return The agent's JUL handler
195 private static ILttngHandler
getAgentHandler() {
197 Field julHandlerField
= LTTngAgent
.class.getDeclaredField("julHandler");
198 julHandlerField
.setAccessible(true);
199 return (ILttngHandler
) julHandlerField
.get(LTTngAgent
.getLTTngAgent());
200 } catch (ReflectiveOperationException
| SecurityException e
) {
201 fail(e
.getMessage());