Commit | Line | Data |
---|---|---|
eca1a136 MJ |
1 | /* |
2 | * Copyright (C) 2022, EfficiOS Inc. | |
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.utils; | |
20 | ||
21 | import java.net.URI; | |
22 | import java.net.URL; | |
23 | ||
24 | import org.apache.logging.log4j.LogManager; | |
25 | import org.apache.logging.log4j.core.LoggerContext; | |
26 | ||
27 | public class Log4j2TestContext { | |
28 | ||
29 | private final URI configFileUri; | |
30 | ||
31 | private LoggerContext loggerContext; | |
32 | ||
33 | public Log4j2TestContext(String configFile) { | |
34 | ||
35 | URL resource = getClass().getClassLoader().getResource(configFile); | |
36 | ||
37 | if (resource == null) { | |
38 | throw new IllegalArgumentException("Config file not found: " + configFile); | |
39 | } | |
40 | ||
41 | try { | |
42 | this.configFileUri = resource.toURI(); | |
43 | } catch (Exception e) { | |
44 | throw new IllegalArgumentException("Config file invalid URI: " + resource); | |
45 | } | |
46 | } | |
47 | ||
48 | public synchronized LoggerContext getLoggerContext() { | |
49 | return loggerContext; | |
50 | } | |
51 | ||
52 | public synchronized void beforeTest() { | |
53 | loggerContext = (LoggerContext) LogManager.getContext( | |
54 | ClassLoader.getSystemClassLoader(), false, configFileUri); | |
55 | } | |
56 | ||
57 | public synchronized void afterTest() { | |
58 | loggerContext.stop(); | |
59 | } | |
60 | } |