Commit | Line | Data |
---|---|---|
2b408e85 AM |
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 | ||
da8308fe | 19 | package org.lttng.ust.agent.benchmarks.jul.handler.lttng.old; |
86316987 AM |
20 | |
21 | import static org.junit.Assert.assertTrue; | |
22 | import static org.junit.Assert.fail; | |
23 | ||
86316987 AM |
24 | import java.lang.reflect.Field; |
25 | ||
26 | import org.junit.After; | |
27 | import org.junit.Before; | |
28 | import org.lttng.ust.agent.LTTngAgent; | |
d4e2e87c | 29 | import org.lttng.ust.agent.benchmarks.jul.handler.JulHandlerBenchmarkBase; |
da8308fe | 30 | import org.lttng.ust.agent.jul.LttngLogHandler; |
8576633f AM |
31 | import org.lttng.ust.agent.utils.LttngSession; |
32 | import org.lttng.ust.agent.utils.LttngSession.Domain; | |
86316987 AM |
33 | |
34 | @SuppressWarnings("deprecation") | |
d4e2e87c | 35 | public class OldLttngJulHandlerTracingEnabledBenchmark extends JulHandlerBenchmarkBase { |
86316987 | 36 | |
8576633f | 37 | private LttngSession session; |
da8308fe | 38 | private LttngLogHandler agentHandler; |
86316987 AM |
39 | |
40 | @Before | |
8576633f | 41 | public void testSetup() { |
da8308fe | 42 | LTTngAgent agentInstance = LTTngAgent.getLTTngAgent(); |
86316987 AM |
43 | |
44 | /* | |
45 | * The "old API" works by attaching a handler managed by the agent to | |
da8308fe | 46 | * the root JUL logger. This causes problems here, because we use |
86316987 AM |
47 | * logger.setUserParentHandler(false), which does not trigger the |
48 | * handler as would be expected. | |
49 | * | |
50 | * Instead we will retrieve this handler through reflection, and attach | |
51 | * it to our own logger here for the duration of the test. | |
52 | */ | |
53 | try { | |
da8308fe AM |
54 | Field julHandlerField = LTTngAgent.class.getDeclaredField("julHandler"); |
55 | julHandlerField.setAccessible(true); | |
56 | agentHandler = (LttngLogHandler) julHandlerField.get(agentInstance); | |
86316987 | 57 | |
da8308fe | 58 | logger.addHandler(agentHandler); |
86316987 AM |
59 | |
60 | } catch (ReflectiveOperationException e) { | |
61 | fail(); | |
62 | } | |
63 | ||
8576633f AM |
64 | session = new LttngSession(null, Domain.JUL); |
65 | assertTrue(session.enableAllEvents()); | |
66 | assertTrue(session.start()); | |
86316987 AM |
67 | } |
68 | ||
69 | @After | |
70 | public void testTeardown() { | |
8576633f AM |
71 | assertTrue(session.stop()); |
72 | session.close(); | |
86316987 | 73 | |
da8308fe AM |
74 | logger.removeHandler(agentHandler); |
75 | LTTngAgent.dispose(); | |
86316987 AM |
76 | } |
77 | } |